Add custom code

Author: tkrajina

typescriptify/typescriptify.go

@@ -89,6 +89,8 @@ type TypeScriptify struct {
 	CreateInterface   bool
 	CustomJsonTag     string
 	customImports     []string
+	customCodeBefore  []string
+	customCodeAfter   []string
 	silent            bool
 
 	structTypes []StructType
@@ -352,6 +354,14 @@ func (t *TypeScriptify) Convert(customCode map[string]string) (string, error) {
 		}
 	}
 
+	if len(t.customCodeBefore) > 0 {
+		result += "\n"
+		for _, code := range t.customCodeBefore {
+			result += "\n" + code + "\n"
+		}
+		result += "\n"
+	}
+
 	for _, enumTyp := range t.enumTypes {
 		elements := t.enums[enumTyp.Type]
 		typeScriptCode, err := t.convertEnum(depth, enumTyp.Type, elements)
@@ -368,6 +378,15 @@ func (t *TypeScriptify) Convert(customCode map[string]string) (string, error) {
 		}
 		result += "\n" + strings.Trim(typeScriptCode, " "+t.Indent+"\r\n")
 	}
+
+	if len(t.customCodeAfter) > 0 {
+		result += "\n"
+		for _, code := range t.customCodeAfter {
+			result += "\n" + code + "\n"
+		}
+		result += "\n"
+	}
+
 	return result, nil
 }
 
@@ -747,6 +766,14 @@ func (t *TypeScriptify) AddImport(i string) {
 	t.customImports = append(t.customImports, i)
 }
 
+func (t *TypeScriptify) WithCustomCodeBefore(i string) {
+	t.customCodeBefore = append(t.customCodeBefore, i)
+}
+
+func (t *TypeScriptify) WithCustomCodeAfter(i string) {
+	t.customCodeAfter = append(t.customCodeAfter, i)
+}
+
 type typeScriptClassBuilder struct {
 	types                map[reflect.Kind]string
 	indent               string

typescriptify/typescriptify_test.go

@@ -67,6 +67,48 @@ export class Person {
 	testConverter(t, converter, false, desiredResult, nil)
 }
 
+func TestTypescriptifyWithCustomCode(t *testing.T) {
+	t.Parallel()
+	converter := New()
+
+	converter.WithCustomCodeBefore(`"a"
+"b"
+"b"`)
+	converter.WithCustomCodeAfter(`"d"
+"e"
+"f"`)
+	converter.AddType(reflect.TypeOf(Person{}))
+	converter.CreateConstructor = false
+	converter.BackupDir = ""
+
+	desiredResult := `"a"
+"b"
+"b"
+
+
+export class Dummy {
+    something: string;
+}
+export class Address {
+    duration: number;
+    text?: string;
+}
+export class Person {
+    name: string;
+    nicknames: string[];
+    addresses: Address[];
+    address?: Address;
+    metadata: {[key:string]:string};
+    friends: Person[];
+    a: Dummy;
+}
+
+"d"
+"e"
+"f"`
+	testConverter(t, converter, false, desiredResult, nil)
+}
+
 func TestTypescriptifyWithCustomImports(t *testing.T) {
 	t.Parallel()
 	converter := New()