bootstrapping kotlin

Author: mightyguava

internal/cmd/generate.go

@@ -110,7 +110,18 @@ func Generate(dir string, stderr io.Writer) (map[string]string, error) {
 
 		}
 
-		files, err := dinosql.Generate(result, combo)
+		var files map[string]string
+		switch pkg.Language {
+		case dinosql.LanguageGo:
+			files, err = dinosql.Generate(result, combo)
+		case dinosql.LanguageKotlin:
+			ktRes, ok := result.(dinosql.KtGenerateable)
+			if !ok {
+				err = fmt.Errorf("Kotlin not supported")
+				break
+			}
+			files, err = dinosql.KtGenerate(ktRes, combo)
+		}
 		if err != nil {
 			fmt.Fprintf(stderr, "# package %s\n", name)
 			fmt.Fprintf(stderr, "error generating code: %s\n", err)

internal/dinosql/config.go

@@ -41,9 +41,17 @@ const (
 	EnginePostgreSQL Engine = "postgresql"
 )
 
+type Language string
+
+const (
+	LanguageGo     Language = "go"
+	LanguageKotlin Language = "kotlin"
+)
+
 type PackageSettings struct {
 	Name                string     `json:"name"`
 	Engine              Engine     `json:"engine,omitempty"`
+	Language            Language   `json:"language,omitempty"`
 	Path                string     `json:"path"`
 	Schema              string     `json:"schema"`
 	Queries             string     `json:"queries"`
@@ -196,6 +204,9 @@ func ParseConfig(rd io.Reader) (GenerateSettings, error) {
 		if config.Packages[j].Engine == "" {
 			config.Packages[j].Engine = EnginePostgreSQL
 		}
+		if config.Packages[j].Language == "" {
+			config.Packages[j].Language = LanguageGo
+		}
 	}
 	return config, nil
 }