unit Sample.SwagDoc.DelphiRESTClient;

interface

uses
  classes,
  system.json,
  System.SysUtils,
  System.Generics.Collections,
  System.Generics.Defaults,
  Swag.Doc,
  Swag.Common.Types,
  Swag.Doc.Path.Operation,
  Swag.Doc.Path.Operation.Response,
  DelphiUnit
  ;

type


  TSwagDocToDelphiRESTClientBuilder = class(TObject)
  private
    FSwagDoc : TSwagDoc;
    function CapitalizeFirstLetter(const typeName: string): string;
    function RewriteUriToSwaggerWay(const uri:string): string;
    function OperationIdToFunctionName(inOperation: TSwagPathOperation): string;
    procedure SortTypeDefinitions(delphiUnit: TDelphiUnit);
    function GenerateUnitText(delphiUnit: TDelphiUnit): string;
    procedure ConvertSwaggerDefinitionsToTypeDefinitions(delphiUnit: TDelphiUnit);
    function ConvertSwaggerTypeToDelphiType(inSwaggerType: TSwagTypeParameter): string;
    function ConvertRefToType(inRef: String): string;
    function ConvertRefToVarName(inRef: String): string;
    procedure ChildType(DelphiUnit : TDelphiUnit; json: TJSONPair);
  public
    constructor Create(SwagDoc: TSwagDoc);
    function Generate: string;
  end;

implementation

uses
  Json.Common.Helpers
  , Winapi.Windows
  , System.IOUtils
  ;

{ TSwagDocToDelphiMVCFrameworkBuilder }

function TSwagDocToDelphiRESTClientBuilder.OperationIdToFunctionName(inOperation: TSwagPathOperation):string;
begin
  Result := inOperation.OperationId.Replace('{','').Replace('}','').Replace('-','');
  if not CharInSet(Result[1], ['a'..'z','A'..'Z']) then
    Result := 'F' + Result;
end;

function TSwagDocToDelphiRESTClientBuilder.RewriteUriToSwaggerWay(const uri:string):string;
begin
  Result := uri.Replace('{','($').Replace('}',')');
end;

function TSwagDocToDelphiRESTClientBuilder.CapitalizeFirstLetter(const typeName: string): string;
begin
  if typeName.Length > 2 then
    Result := Copy(typeName, 1, 1).ToUpper + Copy(typeName, 2, typeName.Length - 1)
  else
    Result := typeName;
end;


constructor TSwagDocToDelphiRESTClientBuilder.Create(SwagDoc: TSwagDoc);
begin
  FSwagDoc := SwagDoc;
end;

function TSwagDocToDelphiRESTClientBuilder.ConvertRefToType(inRef:String):string;
begin
  Result := Copy(inRef, inRef.LastIndexOf('/') + 2);
  Result := Copy(Result,1,1).ToUpper + Copy(Result,2);
  Result := 'T' + Result;
end;

function TSwagDocToDelphiRESTClientBuilder.ConvertRefToVarName(inRef:String):string;
begin
  Result := Copy(inRef, inRef.LastIndexOf('/') + 2);
end;

function TSwagDocToDelphiRESTClientBuilder.Generate: string;
var
  i: Integer;
  j: Integer;
  k: Integer;
  LDelphiUnit : TDelphiUnit;
  LMVCControllerClient : TUnitTypeDefinition;
  LMethod : TUnitMethod;
  LParam : TUnitParameter;
  LParamType : TUnitTypeDefinition;
  LResponse : TPair<string, TSwagResponse>;
  LSchemaObj : TJsonObject;
  LResultParam : TUnitParameter;
  LField : TUnitFieldDefinition;
  LRef : String;
begin
  LDelphiUnit := nil;
  try
    LDelphiUnit := TDelphiUnit.Create;
    LDelphiUnit.UnitFile := 'mvccontrollerclient';
    LDelphiUnit.AddInterfaceUnit('IPPeerClient');
    LDelphiUnit.AddInterfaceUnit('REST.Client');
    LDelphiUnit.AddInterfaceUnit('REST.Authenticator.OAuth');
    LDelphiUnit.AddInterfaceUnit('REST.Types');
    LDelphiUnit.AddInterfaceUnit('MVCFramework');
    LDelphiUnit.AddInterfaceUnit('MVCFramework.Commons');
    LDelphiUnit.AddImplementationUnit('Swag.Doc');

    LMVCControllerClient := TUnitTypeDefinition.Create;
    LMVCControllerClient.TypeName := 'TMyMVCControllerClient';
    LMVCControllerClient.TypeInherited := 'TObject';
    LMVCControllerClient.AddAttribute('  [MVCPath(''' + RewriteUriToSwaggerWay(fSwagDoc.BasePath) + ''')]');

    LField := TUnitFieldDefinition.Create;
    LField.FieldName := 'RESTClient';
    LField.FieldType := 'TRESTClient';
    LMVCControllerClient.Fields.Add(LField);

    LField := TUnitFieldDefinition.Create;
    LField.FieldName := 'RESTRequest';
    LField.FieldType := 'TRESTRequest';
    LMVCControllerClient.Fields.Add(LField);

    LField := TUnitFieldDefinition.Create;
    LField.FieldName := 'RESTResponse';
    LField.FieldType := 'TRESTResponse';
    LMVCControllerClient.Fields.Add(LField);

    LDelphiUnit.AddType(LMVCControllerClient);
    ConvertSwaggerDefinitionsToTypeDefinitions(LDelphiUnit);

    for i := 0 to fSwagDoc.Paths.Count - 1 do
    begin
      for j := 0 to fSwagDoc.Paths[i].Operations.Count - 1 do
      begin
        LMethod := TUnitMethod.Create;
        LMethod.AddAttribute('    [MVCDoc(' + QuotedStr(fSwagDoc.Paths[i].Operations[j].Description) + ')]');
        LMethod.AddAttribute('    [MVCPath(''' + fSwagDoc.Paths[i].Uri + ''')]');
        LMethod.AddAttribute('    [MVCHTTPMethod([http' + fSwagDoc.Paths[i].Operations[j].OperationToString + '])]');
        LMethod.Name := OperationIdToFunctionName(fSwagDoc.Paths[i].Operations[j]);


        for LResponse in FSwagDoc.Paths[i].Operations[j].Responses do
        begin
//        MVCResponse(200, 'success', TEmployee)
          LSchemaObj := LResponse.Value.Schema.JsonSchema;
          if LSchemaObj = nil then
            continue;
          if LSchemaObj.TryGetValue('$ref', LRef) then
          begin
            LMethod.AddAttribute('    [MVCResponse(' + LResponse.Key + ', ' +
                                                   QuotedStr(LResponse.Value.Description) + ', ' + ConvertRefToType(LRef) + ')]');
            LResultParam := TUnitParameter.Create;
            LResultParam.ParamName := ConvertRefToVarName(LRef);
            LResultParam.ParamType := TUnitTypeDefinition.Create;
            LResultParam.ParamType.TypeName := ConvertRefToType(LRef);
            LMethod.AddLocalVariable(LResultParam);
            LMethod.Content.Add('  ' + ConvertRefToVarName(LRef) + ' := ' + ConvertRefToType(LRef) + '.Create;');
              for k := 0 to FSwagDoc.Paths[i].Operations[j].Parameters.Count - 1 do
              begin
//              if fSwagDoc.Paths[i].Operations[j].Parameters[k].InLocation <> rpiPath then
                begin
                  LResultParam := TUnitParameter.Create;
                  LResultParam.ParamName := 'param' + CapitalizeFirstLetter(fSwagDoc.Paths[i].Operations[j].Parameters[k].Name);
                  LResultParam.ParamType := TUnitTypeDefinition.Create;
                  LResultParam.ParamType.TypeName := ConvertSwaggerTypeToDelphiType(fSwagDoc.Paths[i].Operations[j].Parameters[k].TypeParameter);
                  LMethod.AddParameter(LResultParam);
//                  LMethod.AddLocalVariable(LResultParam);
//                  LMethod.Content.Add('  param' + TidyUpTypeName(fSwagDoc.Paths[i].Operations[j].Parameters[k].Name) + ' := Context.Request.Params[' + QuotedStr(fSwagDoc.Paths[i].Operations[j].Parameters[k].Name) + '];');
                end;
              end;
//            method.Content.Add('  Render(' + response.Key + ', ' + ConvertRefToVarName(ref) + ');');
          end
          else
          begin
            if not LSchemaObj.TryGetValue('properties', LSchemaObj) then
              continue;
            if not LSchemaObj.TryGetValue('employees', LSchemaObj) then
              continue;
            if not LSchemaObj.TryGetValue('items', LSchemaObj) then
              continue;
            if LSchemaObj.TryGetValue('$ref', LRef) then
            begin
              LMethod.AddAttribute('    [MVCResponseList(' + LResponse.Key + ', ' +
                                                     QuotedStr(LResponse.Value.Description) + ', ' + ConvertRefToType(LRef) + ')]');
              LResultParam := TUnitParameter.Create;
              LResultParam.ParamName := ConvertRefToVarName(LRef);
              LResultParam.ParamType := TUnitTypeDefinition.Create;
              LResultParam.ParamType.TypeName := 'TObjectList<' + ConvertRefToType(LRef) + '>';
              LMethod.AddLocalVariable(LResultParam);
              LDelphiUnit.AddInterfaceUnit('Generics.Collections');
              LMethod.Content.Add('  ' + ConvertRefToVarName(LRef) + ' := TObjectList<' + ConvertRefToType(LRef) + '>.Create;');

              for k := 0 to FSwagDoc.Paths[i].Operations[j].Parameters.Count - 1 do
              begin
                if fSwagDoc.Paths[i].Operations[j].Parameters[k].InLocation <> rpiPath then
                begin
                  LResultParam := TUnitParameter.Create;
                  LResultParam.ParamName := 'param' + fSwagDoc.Paths[i].Operations[j].Parameters[k].Name;
                  LResultParam.ParamType := TUnitTypeDefinition.Create;
                  LResultParam.ParamType.TypeName := ConvertSwaggerTypeToDelphiType(fSwagDoc.Paths[i].Operations[j].Parameters[k].TypeParameter);
                  LMethod.AddLocalVariable(LResultParam);
                  LMethod.Content.Add('  ' + fSwagDoc.Paths[i].Operations[j].Parameters[k].Name + ' := Context.Request.Params[' + QuotedStr(fSwagDoc.Paths[i].Operations[j].Parameters[k].Name) + '];');
                end;
              end;



//              method.Content.Add('  Render(' + response.Key + ', ' + ConvertRefToVarName(ref) + ');');
            end
            else
            begin
              for k := 0 to FSwagDoc.Paths[i].Operations[j].Parameters.Count - 1 do
              begin
                if fSwagDoc.Paths[i].Operations[j].Parameters[k].InLocation <> rpiPath then
                begin
                  LResultParam := TUnitParameter.Create;
                  LResultParam.ParamName := 'param' + fSwagDoc.Paths[i].Operations[j].Parameters[k].Name;
                  LResultParam.ParamType := TUnitTypeDefinition.Create;
                  LResultParam.ParamType.TypeName := ConvertSwaggerTypeToDelphiType(fSwagDoc.Paths[i].Operations[j].Parameters[k].TypeParameter);
                  LMethod.AddLocalVariable(LResultParam);
                  LMethod.Content.Add('  ' + fSwagDoc.Paths[i].Operations[j].Parameters[k].Name + ' := Context.Request.Params[' + QuotedStr(fSwagDoc.Paths[i].Operations[j].Parameters[k].Name) + '];');
                end;
              end;
              LMethod.AddAttribute('    [MVCResponse(' + LResponse.Key + ', ' +
                                                   QuotedStr(LResponse.Value.Description) + ')]');
            end;
          end;
        end;

        LMVCControllerClient.FMethods.Add(LMethod);
      end;
    end;

    SortTypeDefinitions(LDelphiUnit);

    Result := GenerateUnitText(LDelphiUnit);
  finally
    fSwagDoc.Free;
    LDelphiUnit.Free;
  end;
end;


procedure TSwagDocToDelphiRESTClientBuilder.ChildType(DelphiUnit : TDelphiUnit; json: TJSONPair);
var
  LTypeInfo: TUnitTypeDefinition;
  LJsonProps: TJSONObject;
  LFieldInfo: TUnitFieldDefinition;
  LTypeObj: TJSONObject;
  j: Integer;
  LValue : string;
begin
  OutputDebugString(PChar('Child: ' + json.ToJSON));
  LTypeInfo := TUnitTypeDefinition.Create;
  LTypeInfo.TypeName := 'T' + CapitalizeFirstLetter(json.JSONString.Value);

  LJsonProps := (json.JSONValue as TJSONObject).Values['properties'] as TJSONObject;
  for j := 0 to LJsonProps.Count - 1 do
  begin
    OutputDebugString(PChar(LJsonProps.Pairs[j].ToJSON));
    LFieldInfo := TUnitFieldDefinition.Create;
    LFieldInfo.FieldName := LJsonProps.Pairs[j].JsonString.Value;
    LTypeObj := LJsonProps.Pairs[j].JsonValue as TJSONObject;
    LFieldInfo.FieldType := LTypeObj.Values['type'].Value;
    if LFieldInfo.FieldType = 'number' then
      LFieldInfo.FieldType := 'Double'
    else if LFieldInfo.FieldType = 'object' then
    begin
      LFieldInfo.FieldType := 'T' + CapitalizeFirstLetter(LJsonProps.Pairs[j].JsonString.Value);
      ChildType(DelphiUnit, LJsonProps.Pairs[j]);
    end;
    if LTypeObj.TryGetValue('description', LValue) then
      LFieldInfo.AddAttribute('[MVCDoc(' + QuotedStr(LValue) + ')]');
    if LTypeObj.TryGetValue('format', LValue) then
      LFieldInfo.AddAttribute('[MVCFormat(' + QuotedStr(LValue) + ')]');
    if LTypeObj.TryGetValue('maxLength', LValue) then
      LFieldInfo.AddAttribute('[MVCMaxLength(' + LValue + ')]');
    LTypeInfo.Fields.Add(LFieldInfo);
  end;
  delphiUnit.AddType(LTypeInfo);
end;

procedure TSwagDocToDelphiRESTClientBuilder.ConvertSwaggerDefinitionsToTypeDefinitions(delphiUnit: TDelphiUnit);
var
  LTypeInfo: TUnitTypeDefinition;
  LJsonProps: TJSONObject;
  LFieldInfo: TUnitFieldDefinition;
  LTypeObj: TJSONObject;
  i: Integer;
  j: Integer;
  LValue : string;
begin
  for i := 0 to fSwagDoc.Definitions.Count - 1 do
  begin
    LTypeInfo := TUnitTypeDefinition.Create;
    LTypeInfo.TypeName := 'T' + CapitalizeFirstLetter(fSwagDoc.Definitions[i].Name);
    LJsonProps := fSwagDoc.Definitions[i].JsonSchema.Values['properties'] as TJSONObject;
    for j := 0 to LJsonProps.Count - 1 do
    begin
      OutputDebugString(PChar(LJsonProps.Pairs[j].ToJSON));
      LFieldInfo := TUnitFieldDefinition.Create;
      LFieldInfo.FieldName := LJsonProps.Pairs[j].JsonString.Value;
      LTypeObj := LJsonProps.Pairs[j].JsonValue as TJSONObject;
      LFieldInfo.FieldType := LTypeObj.Values['type'].Value;
      if LFieldInfo.FieldType = 'number' then
        LFieldInfo.FieldType := 'Double'
      else if LFieldInfo.FieldType = 'object' then
      begin
        LFieldInfo.FieldType := 'T' + CapitalizeFirstLetter(LJsonProps.Pairs[j].JsonString.Value);
        ChildType(DelphiUnit, LJsonProps.Pairs[j]);
      end;
      if LTypeObj.TryGetValue('description', LValue) then
        LFieldInfo.AddAttribute('[MVCDoc(' + QuotedStr(LValue) + ')]');
      if LTypeObj.TryGetValue('format', LValue) then
        LFieldInfo.AddAttribute('[MVCFormat(' + QuotedStr(LValue) + ')]');
      if LTypeObj.TryGetValue('maxLength', LValue) then
        LFieldInfo.AddAttribute('[MVCMaxLength(' + LValue + ')]');
      if LTypeObj.TryGetValue('minimum', LValue) then
        LFieldInfo.AddAttribute('[MVCMinimum(' + LValue + ')]');
      if LTypeObj.TryGetValue('maximum', LValue) then
        LFieldInfo.AddAttribute('[MVCMaximum(' + LValue + ')]');
      LTypeInfo.Fields.Add(LFieldInfo);
    end;
    delphiUnit.AddType(LTypeInfo);
  end;
end;

function TSwagDocToDelphiRESTClientBuilder.ConvertSwaggerTypeToDelphiType(inSwaggerType: TSwagTypeParameter): string;
begin
  case inSwaggerType of
    stpNotDefined: Result := 'notdefined';
    stpString: Result := 'String';
    stpNumber: Result := 'Double';
    stpInteger: Result := 'Integer';
    stpBoolean: Result := 'Boolean';
    stpArray: Result := 'Array of';
    stpFile: Result := 'err File';
  end;

end;

function TSwagDocToDelphiRESTClientBuilder.GenerateUnitText(delphiUnit: TDelphiUnit): string;
var
  i: Integer;
  j: Integer;
  LMethod: TUnitMethod;
  LMvcFile: TStringList;
begin
  LMvcFile := TStringList.Create;
  try
    LMvcFile.Add(delphiUnit.GenerateInterfaceSectionStart);
    LMvcFile.Add(delphiUnit.GenerateInterfaceUses);
    LMvcFile.Add('(*');
    LMvcFile.Add('Title: ' + fSwagDoc.Info.Title);
    LMvcFile.Add('Description: ' + fSwagDoc.Info.Description);
    LMvcFile.Add('License: ' + fSwagDoc.Info.License.Name);
    LMvcFile.Add('*)');
    LMvcFile.Add('');
    LMvcFile.Add('type');

    SortTypeDefinitions(delphiUnit);

    for i := 0 to delphiUnit.TypeDefinitions.Count - 1 do
    begin
      LMvcFile.Add(delphiUnit.TypeDefinitions[i].GenerateInterface);
    end;
    LMvcFile.Add(delphiUnit.GenerateImplementationSectionStart);
    LMvcFile.Add(delphiUnit.GenerateImplementationUses);
    LMvcFile.Add('');
    for j := 0 to delphiUnit.TypeDefinitions.Count - 1 do
    begin
      for LMethod in delphiUnit.TypeDefinitions[j].GetMethods do
      begin
        LMvcFile.Add(LMethod.GenerateImplementation(delphiUnit.TypeDefinitions[j]));
      end;
    end;
    LMvcFile.Add('end.');
    Result := LMvcFile.Text;
    LMvcFile.SaveToFile(TPath.Combine(TPath.GetDirectoryName(ParamStr(0)), '..\..\mvccontrollerclient.pas'));
  finally
    FreeAndNil(LMvcFile);
  end;
end;


procedure TSwagDocToDelphiRESTClientBuilder.SortTypeDefinitions(delphiUnit: TDelphiUnit);
begin
  { TODO : Make this much more advanced to handle dependency ordering of declarations }

  delphiUnit.TypeDefinitions.Sort(TComparer<TUnitTypeDefinition>.Construct(function (const L, R: TUnitTypeDefinition): integer
  begin
    if L.TypeInherited = 'TMyMVCController' then
      Result := -1
    else if R.TypeInherited = 'TMyMVCController' then
      Result := 1
    else if L.TypeName = R.TypeName then
      Result := 0
    else if L.TypeName < R.TypeName then
      Result := -1
    else
      Result := 1;
  end));
end;

end.