common.js

'use strict';

var util = require('util');
var recurse = require('openapi_optimise/common.js').recurse;
var circular = require('openapi_optimise/circular.js');
var jptr = require('jgexml/jpath.js');

const MAX_SCHEMA_DEPTH=100;

/* originally from https://github.com/for-GET/know-your-http-well/blob/master/json/status-codes.json */
/* "Unlicensed", public domain */
var statusCodes = require('./statusCodes.json');

// could change these to be regexes...
var xmlContentTypes = ['application/xml', 'text/xml', 'image/svg+xml', 'application/rss+xml', 'application/rdf+xml', 'application/atom+xml', 'application/mathml+xml', 'application/hal+xml'];
var jsonContentTypes = ['application/json', 'text/json', 'application/hal+json', 'application/ld+json', 'application/json-patch+json'];
var yamlContentTypes = ['application/x-yaml', 'text/x-yaml'];
var formContentTypes = ['multipart/form-data', 'application/x-www-form-urlencoded', 'application/octet-stream'];

function clone(obj) {
    return JSON.parse(JSON.stringify(obj));
}

function nop(obj) {
    return obj;
}

function dereference(obj, circles, api, cloneFunc) {
    if (!cloneFunc) cloneFunc = nop;
    while (obj && obj["$ref"] && !circular.isCircular(circles, obj.$ref)) {
        var oRef = obj.$ref;
        obj = cloneFunc(jptr.jptr(api, obj["$ref"]));
        obj["x-old-ref"] = oRef;
    }
    var changes = 1;
    while (changes > 0) {
        changes = 0;
        recurse(obj, {}, function (obj, state) {
            if ((state.key === '$ref') && (typeof obj === 'string') && (!circular.isCircular(circles, obj))) {
                state.parents[state.parents.length - 2][state.keys[state.keys.length - 2]] = cloneFunc(jptr.jptr(api, obj));
                state.parents[state.parents.length - 2][state.keys[state.keys.length - 2]]["x-old-ref"] = obj;
                delete state.parent["$ref"]; // just in case
                changes++;
            }
        });
    }
    return obj;
}

function doContentType(types, targets) {
    for (var type in types) {
        for (var target of targets) {
            if (types[type] === target) return true;
        }
    }
    return false;
}

function languageCheck(language, language_tabs, mutate) {
    var lcLang = language.toLowerCase();
    if (lcLang === 'c#') lcLang = 'csharp';
    if (lcLang === 'c++') lcLang = 'cpp';
    for (var l in language_tabs) {
        var target = language_tabs[l];
        if (typeof target === 'object') {
            if (Object.keys(target)[0] === lcLang) {
                return lcLang;
            }
        }
        else {
            if (target === lcLang) return lcLang;
        }
    }
    if (mutate) {
        var newLang = {};
        newLang[lcLang] = language;
        language_tabs.push(newLang);
        return lcLang;
    }
    return false;
}

function gfmLink(text) {
    text = text.trim().toLowerCase();
    text = text.split("'").join('');
    text = text.split('"').join('');
    text = text.split('.').join('');
    text = text.split('`').join('');
    text = text.split(':').join('');
    text = text.split('/').join('');
    text = text.split('<').join('');
    text = text.split('>').join('');
    text = text.split('<').join('');
    text = text.split('>').join('');
    text = text.split(' ').join('-');
    return text;
}

function extract(o,parent,seen,depth,callback){
    JSON.stringify(o,function(k,v){
        if (v && v.properties) {
            for (let p in v.properties) {
                var already = seen.indexOf(v.properties[p])>=0;
                if (v.properties[p] && v.properties[p].type === 'array') already = true; // array processing
                if (!already) {
                    let required = false;
                    if (v.required && Array.isArray(v.required)) {
                        required = v.required.indexOf(p)>=0;
                    }
                    let oldRef = '';
                    if (v.properties[p]) {
                        oldRef = v.properties[p]["x-old-ref"]||v.properties[p].$ref||'';
                    }
                    let newProp = {};
                    newProp[p] = v.properties[p];
                    callback(newProp,depth,required,oldRef);
                    if (depth<MAX_SCHEMA_DEPTH) {
                        extract(v.properties[p],p,seen,depth+1,callback);
                    }
                    else {
                        throw new Error('Max schema depth exceeded');
                    }
                    seen.push(v.properties[p]);
                 }
            }
        }
        if (v && v.items) { // array processing
            var name = k||'anonymous';
            var dummy = {};
            dummy.properties = {};
            dummy.properties[name] = v.items;
            dummy.properties[name].description = v.description;
            dummy.properties[name]["x-isArray"] = true;
            extract(dummy,k,seen,depth,callback);
        }
        return v;
    });
}

function schemaToArray(schema,depth,lines,trim) {

    let seen = [];
    extract(schema,'',seen,depth,function(obj,depth,required,oldRef){
        let prefix = '»'.repeat(depth);
        for (let p in obj) {
            if (obj[p]) {
                var prop = {};
                prop.name = (prefix+' '+p).trim();
                prop.in = 'body';
                prop.type = obj[p].type||'Unknown';
                if (obj[p].format) prop.type = prop.type+'('+obj[p].format+')';

                if (((prop.type === 'object') || (prop.type === 'Unknown')) && oldRef) {
                    oldRef = oldRef.split('/').pop();
                    prop.type = '['+oldRef+'](#schema'+gfmLink(oldRef)+')';
                }

                if (obj[p]["x-isArray"]) {
                    prop.type = '['+prop.type+']';
                }

                prop.required = required;
                prop.description = (obj[p].description && obj[p].description !== 'undefined') ? obj[p].description : 'No description'; // the actual string 'undefined'
                if (trim && typeof prop.description === 'string') prop.description = prop.description.split('\n').join(' ');
                prop.depth = depth;
                if (obj[p].enum) prop.schema = {enum:obj[p].enum};
                lines.push(prop);
            }
        }
    });
    if (!schema.properties && !schema.items) {
        let prop = {};
        prop.name = schema.title;
        if (!prop.name && schema.type && schema.type !== 'object') prop.name = 'simple';
        if (!prop.name && schema.additionalProperties) prop.name = 'additionalProperties';
        if (!prop.name && schema.patternProperties) prop.name = 'patternProperties';
        prop.description = schema.description||'No description';
        if (trim) prop.description = prop.description.split('\n').join(' ');
        prop.type = schema.type||'Unknown';
        prop.required = false;
        prop.in = 'body';
        if (schema.format) prop.type = prop.type+'('+schema.format+')';
        prop.depth = 0;
        lines.unshift(prop);
    }
}

module.exports = {
    statusCodes : statusCodes,
    xmlContentTypes : xmlContentTypes,
    jsonContentTypes : jsonContentTypes,
    yamlContentTypes : yamlContentTypes,
    formContentTypes : formContentTypes,
    dereference : dereference,
    doContentType : doContentType,
    languageCheck : languageCheck,
    clone : clone,
    gfmLink : gfmLink,
    schemaToArray : schemaToArray
};