123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225
import { DenoDoc } from "./deno-doc.model.ts";import * as oa from "../../builder/openapi-models.ts";import { JsDocObject, JsDocParse, PropertyJsDocParse } from "./js-doc-parser.ts";import { getOpenApiMetadataArgsStorage } from "../../metadata/openapi-metadata.storage.ts";import TsType = DenoDoc.TsType;
export interface ParsedNamesDocMap {  classes: Map<string, DenoDoc.RootDef>;  interfaces: Map<string, DenoDoc.RootDef>;  controllers: Map<string, DenoDoc.RootDef>;  enums: Map<string, DenoDoc.RootDef>;}
export function getParsedNames(doc: DenoDoc.RootDef[]): ParsedNamesDocMap {  const classes = new Map<string, DenoDoc.RootDef>();  const interfaces = new Map<string, DenoDoc.RootDef>();  const controllers = new Map<string, DenoDoc.RootDef>();  const enums = new Map<string, DenoDoc.RootDef>();
  function findRootDefByDoc(doc: DenoDoc.RootDef[]) {    for (let i = 0; i < doc.length; i++) {      const currentDoc = doc[i];
      if (currentDoc.kind === "import") {        currentDoc.importDef.def && findRootDefByDoc(currentDoc.importDef.def);      } else {        if (currentDoc.kind === "class") {          // TODO set all without equivalent          // if(classes.has(currentDoc.name)) {          //     console.log(currentDoc.name)          // }
          if (            currentDoc.classDef?.decorators &&            currentDoc.classDef.decorators.some((d) => d.name === "Controller")          ) {            controllers.set(currentDoc.name, currentDoc);          } else {            classes.set(currentDoc.name, currentDoc);          }        }
        if (currentDoc.kind === "interface") {          interfaces.set(currentDoc.name, currentDoc);        }
        if (currentDoc.kind === "enum") {          enums.set(currentDoc.name, currentDoc);        }      }    }  }
  findRootDefByDoc(doc);
  return { classes, controllers, interfaces, enums };}
export function getSchemeByDef(def: DenoDoc.RootDef): oa.SchemaObject {  let result: oa.SchemaObject = {};
  const jsDoc = def.jsDoc && JsDocParse(def.jsDoc);
  if (jsDoc) {    result = getSchemeFromJsDoc(jsDoc);  }
  // Parse properties  let properties = def.classDef ? def.classDef.properties : def.interfaceDef.properties;
  if (properties) {    result.properties = {};    properties.filter((p) => p.accessibility !== "private").forEach(      (property) => {        let propertyResult: oa.SchemaObject = {};
        const stdJsDoc = property.jsDoc && JsDocParse(property.jsDoc);
        if (stdJsDoc) {          propertyResult = getSchemeFromJsDoc(stdJsDoc);          const propertyJsDoc = property.jsDoc &&            PropertyJsDocParse(property.jsDoc);
          propertyResult = { ...propertyResult, ...propertyJsDoc as any };        }
        const propertyScheme = getSchemeFromProperty(property);        propertyResult = { ...propertyResult, ...propertyScheme };        result.properties![property.name] = propertyResult;      },    );  }
  return result;}
export function getShemeByEnumDef(def: DenoDoc.RootDef): oa.SchemaObject {  let result: oa.SchemaObject = {};
  const jsDocParsed = def.jsDoc && JsDocParse(def.jsDoc);
  if (jsDocParsed) {    result = getSchemeFromJsDoc(jsDocParsed);  }
  result.type = "string";
  if (def.enumDef) {    result.enum = [];    def.enumDef.members.forEach((member) => {      result.enum!.push(member.name);    });  }
  return result;}
function getSchemeFromJsDoc(jsDocParsed: JsDocObject) {  const result: oa.SchemaObject = {};
  result.description = jsDocParsed.description;  result.example = jsDocParsed.example;  result.deprecated = jsDocParsed.deprecated;  result.default = jsDocParsed.default;  // @ts-ignore  result.required = jsDocParsed.required;
  return result;}
function getSchemeFromProperty(prop: DenoDoc.Property): oa.SchemaObject {  const result: oa.SchemaObject = {};
  if (!prop.tsType) {    return result;  }
  if (prop.tsType.kind === "keyword") {    result.type = getKeywordType(prop.tsType.repr);  }
  if (prop.tsType.kind === "array") {    result.type = "array";
    // TODO add other array keywords, enums for example    if (prop.tsType.array.kind === "keyword") {      result.items = {        type: getKeywordType(prop.tsType.array.repr),      };    }
    if (prop.tsType.array.kind === "typeRef") {      if (isStdTsType(prop.tsType)) {        result.items = {          type: "string",        };      } else {        result.items = {          $ref: GetShemeLinkAndRegister(prop.tsType.array.typeRef),        };      }    }  }
  if (prop.tsType.kind === "typeRef") {    if (isStdTsType(prop.tsType)) {      result.type = "string";    } else {      result["$ref"] = GetShemeLinkAndRegister(prop.tsType.typeRef);    }  }  return result;}
function isStdTsType(ref: TsType): boolean {  switch (ref.repr) {    case "Object":    case "Date":    case "Symbol":    case "Map":    case "JSON":    case "RegExp":    case "String":    case "ArrayBuffer":    case "DataView":    case "Int8Array":    case "Uint8Array":    case "Uint8ClampedArray":    case "Int16Array":    case "Uint16Array":    case "Int32Array":    case "Uint32Array":    case "Float32Array":    case "Float64Array":      return true;  }
  return false;}
function getKeywordType(repr: string): string {  switch (repr) {    case "number":    case "bigint":      return "integer";    case "any":      return "object";    default:      return repr;  }}
/** * Gets right scheme link and register as uses * @param name */function GetShemeLinkAndRegister(typeRef: DenoDoc.TypeRef): string {  if (typeRef.typeName) {    getOpenApiMetadataArgsStorage().usableClassNamesSet.add(typeRef.typeName);    return typeRef.typeName && "#/components/schemas/" + typeRef.typeName;  }
  return "";}