Add option for type-keyword imports

raileasyuk · Nov 6, 2023 · 668bb46 · 668bb46
1 parent bceb542
commit 668bb46
Show file tree

Hide file tree

Showing 7 changed files with 24 additions and 15 deletions.
diff --git a/TypeScript.ContractGenerator/Internals/TypeScriptImportDefaultFromPathStatement.cs b/TypeScript.ContractGenerator/Internals/TypeScriptImportDefaultFromPathStatement.cs
@@ -6,11 +6,13 @@ internal class TypeScriptImportDefaultFromPathStatement : TypeScriptImportStatem
     {
         public override string GenerateCode(ICodeGenerationContext context)
         {
-            return $"import {TypeName} from '{context.GetReferenceFromUnitToAnother(CurrentUnit.Path, PathToUnit)}';";
+            var typeKeyword = UseTypeKeyword ? "type " : "";
+            return $"import {typeKeyword}{TypeName} from '{context.GetReferenceFromUnitToAnother(CurrentUnit.Path, PathToUnit)}';";
         }
 
         public string TypeName { get; set; }
         public TypeScriptUnit CurrentUnit { get; set; }
         public string PathToUnit { get; set; }
+        public bool UseTypeKeyword { get; set; }
     }
-}
+}
diff --git a/TypeScript.ContractGenerator/Internals/TypeScriptImportFromPathStatement.cs b/TypeScript.ContractGenerator/Internals/TypeScriptImportFromPathStatement.cs
@@ -6,11 +6,13 @@ internal class TypeScriptImportFromPathStatement : TypeScriptImportStatement
     {
         public override string GenerateCode(ICodeGenerationContext context)
         {
-            return $"import {{ {TypeName} }} from '{context.GetReferenceFromUnitToAnother(CurrentUnit.Path, PathToUnit)}';";
+            var typeKeyword = UseTypeKeyword ? "type " : "";
+            return $"import {typeKeyword}{{ {TypeName} }} from '{context.GetReferenceFromUnitToAnother(CurrentUnit.Path, PathToUnit)}';";
         }
 
         public string TypeName { get; set; }
         public TypeScriptUnit CurrentUnit { get; set; }
         public string PathToUnit { get; set; }
+        public bool UseTypeKeyword { get; set; }
     }
-}
+}
diff --git a/TypeScript.ContractGenerator/Internals/TypeScriptImportFromUnitStatement.cs b/TypeScript.ContractGenerator/Internals/TypeScriptImportFromUnitStatement.cs
@@ -6,11 +6,13 @@ internal class TypeScriptImportFromUnitStatement : TypeScriptImportStatement
     {
         public override string GenerateCode(ICodeGenerationContext context)
         {
-            return $"import {{ {TypeName} }} from '{context.GetReferenceFromUnitToAnother(CurrentUnit.Path, TargetUnit.Path)}';";
+            var typeKeyword = UseTypeKeyword ? "type " : "";
+            return $"import {typeKeyword}{{ {TypeName} }} from '{context.GetReferenceFromUnitToAnother(CurrentUnit.Path, TargetUnit.Path)}';";
         }
 
         public string TypeName { get; set; }
         public TypeScriptUnit TargetUnit { get; set; }
         public TypeScriptUnit CurrentUnit { get; set; }
+        public bool UseTypeKeyword { get; set; }
     }
-}
+}
diff --git a/TypeScript.ContractGenerator/TypeBuilders/RedirectToTypeBuildingContext.cs b/TypeScript.ContractGenerator/TypeBuilders/RedirectToTypeBuildingContext.cs
@@ -14,10 +14,10 @@ public RedirectToTypeBuildingContext(string typeName, string path, ITypeInfo typ
 
         protected override TypeScriptType ReferenceFromInternal(ITypeInfo type, TypeScriptUnit targetUnit, ITypeGenerator typeGenerator)
         {
-            return targetUnit.AddTypeImport(type, new TypeScriptInterfaceDeclaration {Name = typeName}, new TypeScriptUnit {Path = path});
+            return targetUnit.AddTypeImport(type, new TypeScriptInterfaceDeclaration {Name = typeName}, new TypeScriptUnit {Path = path}, typeGenerator.Options.UseTypeImports);
         }
 
         private readonly string typeName;
         private readonly string path;
     }
-}
+}
diff --git a/TypeScript.ContractGenerator/TypeBuilders/TypeBuildingContext.cs b/TypeScript.ContractGenerator/TypeBuilders/TypeBuildingContext.cs
@@ -22,7 +22,7 @@ public override void Initialize(ITypeGenerator typeGenerator)
 
         protected override TypeScriptType ReferenceFromInternal(ITypeInfo type, TypeScriptUnit targetUnit, ITypeGenerator typeGenerator)
         {
-            return targetUnit.AddTypeImport(type, Declaration, Unit);
+            return targetUnit.AddTypeImport(type, Declaration, Unit, typeGenerator.Options.UseTypeImports);
         }
     }
-}
+}
diff --git a/TypeScript.ContractGenerator/TypeScriptGenerationOptions.cs b/TypeScript.ContractGenerator/TypeScriptGenerationOptions.cs
@@ -7,7 +7,7 @@ public class TypeScriptGenerationOptions
         public bool EnableOptionalProperties { get; set; } = true;
         public bool UseGlobalNullable { get; set; }
         public string? CustomContentMarker { get; set; }
-
+        public bool UseTypeImports { get; set; } = false;
         public static TypeScriptGenerationOptions Default => new TypeScriptGenerationOptions();
     }
-}
+}
diff --git a/TypeScript.ContractGenerator/TypeScriptUnit.cs b/TypeScript.ContractGenerator/TypeScriptUnit.cs
@@ -15,7 +15,7 @@ public class TypeScriptUnit
 
         public List<TypeScriptStatement> Body { get; } = new List<TypeScriptStatement>();
 
-        public TypeScriptTypeReference AddTypeImport(ITypeInfo sourceType, TypeScriptTypeDeclaration typeDeclaration, TypeScriptUnit sourceUnit)
+        public TypeScriptTypeReference AddTypeImport(ITypeInfo sourceType, TypeScriptTypeDeclaration typeDeclaration, TypeScriptUnit sourceUnit, bool useTypeKeyword=false)
         {
             if (sourceUnit != this && !imports.ContainsKey(sourceType))
             {
@@ -24,6 +24,7 @@ public TypeScriptTypeReference AddTypeImport(ITypeInfo sourceType, TypeScriptTyp
                         TypeName = typeDeclaration.Name,
                         CurrentUnit = this,
                         TargetUnit = sourceUnit,
+                        UseTypeKeyword = useTypeKeyword
                     });
             }
             return new TypeScriptTypeReference(typeDeclaration.Name);
@@ -39,12 +40,13 @@ public TypeScriptVariableReference AddSymbolImport(string symbolName, string pat
                         TypeName = symbolName,
                         CurrentUnit = this,
                         PathToUnit = path,
+                        UseTypeKeyword = false,
                     });
             }
             return new TypeScriptVariableReference(symbolName);
         }
 
-        public TypeScriptVariableReference AddDefaultSymbolImport(string localName, string path)
+        public TypeScriptVariableReference AddDefaultSymbolImport(string localName, string path, bool useTypeKeyword=false)
         {
             var importedSymbol = new ImportedSymbol("default", localName, path);
             if (!symbolImports.ContainsKey(importedSymbol))
@@ -54,6 +56,7 @@ public TypeScriptVariableReference AddDefaultSymbolImport(string localName, stri
                         TypeName = localName,
                         CurrentUnit = this,
                         PathToUnit = path,
+                        UseTypeKeyword = useTypeKeyword,
                     });
             }
             return new TypeScriptVariableReference(localName);
@@ -84,4 +87,4 @@ public string GenerateCode(DefaultCodeGenerationContext context)
         private readonly Dictionary<ITypeInfo, TypeScriptImportStatement> imports = new Dictionary<ITypeInfo, TypeScriptImportStatement>();
         private readonly Dictionary<ImportedSymbol, TypeScriptImportStatement> symbolImports = new Dictionary<ImportedSymbol, TypeScriptImportStatement>();
     }
-}
+}