import * as _ from 'lodash'

export default class JsonSchemaHelper {
  static getBaseType(schema: JsonSchema): string {
    if (schema.type == undefined) {
      return "";
    } else if (typeof schema.type === "string") {
      return schema.type as string;
    } else if (schema.type.indexOf("null") != -1) {
      return schema.type[0];
    }
  }

  static isNullable(schema: JsonSchema): boolean {
    return (schema.type as InternalTypeOfSchema[]).indexOf("null") != -1;
  }

  static parseSchema(schema: string): JsonSchema {
    var data = JSON.parse(schema) as JsonSchema;

    if (data.definitions == undefined) {
      return data;
    } else {
      return this.flat(data, data.definitions, {});
    }
  }

  static flat(
    data: JsonSchema,
    originalDefinitions: { [index: string]: JsonSchema },
    flattenedDefinitions: { [index: string]: JsonSchema }
  ): JsonSchema {
    if (originalDefinitions == undefined) {
      return data;
    }

    for (var index in data) {
      if (index === "definitions" || data.hasOwnProperty(index) == false) {
        continue;
      }

      if (data[index] != null && data[index]["$ref"] !== undefined) {
        data[index] = {...data[index], ...this.getFromDefinition(
          data[index]["$ref"],
          originalDefinitions,
          flattenedDefinitions
        )};
        delete data[index]["$ref"];
      } else if (typeof data[index] !== "string") {
        this.flat(data[index], originalDefinitions, flattenedDefinitions);
      }
    }

    return data;
  }

  static getFromDefinition(
    id: string,
    originalDefinitions: { [index: string]: JsonSchema },
    flattenedDefinitions: { [index: string]: JsonSchema }
  ) {
    var isNotAlreadyFlattened = flattenedDefinitions[id] == undefined;
    if (isNotAlreadyFlattened) {
      var pathToDefinition = id.replace("#/definitions/", "").replace(/\//g, ".");
      flattenedDefinitions[id] = _.get(originalDefinitions, pathToDefinition);
      
      this.flat(
        flattenedDefinitions[id],
        originalDefinitions,
        flattenedDefinitions
      );
    }

    return flattenedDefinitions[id];
  }
}