write Models.pm

internal/gen.go

@@ -2,28 +2,94 @@ package perl
 
 import (
 	//"bufio"
-	//"bytes"
+	"bufio"
+	"bytes"
 	"context"
+
 	//"errors"
 	//"fmt"
 	//"go/format"
-	//"strings"
-	//"text/template"
+	"strings"
+	"text/template"
+
+	"github.com/kfly8/sqlc-gen-perl/internal/opts"
+	"github.com/sqlc-dev/plugin-sdk-go/sdk"
 
-	//"github.com/sqlc-dev/plugin-sdk-go/sdk"
 	//"github.com/sqlc-dev/plugin-sdk-go/metadata"
 	"github.com/sqlc-dev/plugin-sdk-go/plugin"
 )
 
+
+type tmplCtx struct {
+	Q string
+	SqlcVersion string
+	SourceName string
+}
+
 func Generate(ctx context.Context, req *plugin.GenerateRequest) (*plugin.GenerateResponse, error) {
-	return generate(req)
+	options, err := opts.Parse(req)
+	if err != nil {
+		return nil, err
+	}
+
+	if err := opts.ValidateOpts(options); err != nil {
+		return nil, err
+	}
+
+	return generate(req, options)
 }
 
 
-func generate(req *plugin.GenerateRequest) (*plugin.GenerateResponse, error) {
+func generate(req *plugin.GenerateRequest, options *opts.Options) (*plugin.GenerateResponse, error) {
+
+	tctx := tmplCtx{
+		Q: "`",
+		SqlcVersion: req.SqlcVersion,
+	}
+
+	funcMap := template.FuncMap{
+		"lowerTitle": sdk.LowerTitle,
+		"comment":    sdk.DoubleSlashComment,
+		"escape":     sdk.EscapeBacktick,
+		"hasPrefix":  strings.HasPrefix,
+	}
+
+	tmpl := template.Must(
+		template.New("table").
+			Funcs(funcMap).
+			ParseFS(
+				templates,
+				"templates/*.tmpl",
+			),
+	)
 
 	output := map[string]string{}
-	output["Query.pm"] = "package Query";
+
+	execute := func(name, templateName string) error {
+		var b bytes.Buffer
+		w := bufio.NewWriter(&b)
+		tctx.SourceName = name
+		err := tmpl.ExecuteTemplate(w, templateName, &tctx)
+		w.Flush()
+		if err != nil {
+			return err
+		}
+
+		if !strings.HasSuffix(name, ".pm") {
+			name += ".pm"
+		}
+
+		output[name] = b.String()
+		return nil
+	}
+
+	modelsFileName := "Models.pm"
+	if options.OutputModelsFileName != "" {
+		modelsFileName = options.OutputModelsFileName
+	}
+	if err := execute(modelsFileName, "modelsFile"); err != nil {
+		return nil, err
+	}
 
 	resp := plugin.GenerateResponse{}
 

internal/opts/options.go

@@ -0,0 +1,42 @@
+package opts
+
+import (
+	"encoding/json"
+	"fmt"
+	//"maps"
+	///"path/filepath"
+
+	"github.com/sqlc-dev/plugin-sdk-go/plugin"
+)
+
+type Options struct {
+	OutputModelsFileName        string            `json:"output_models_file_name,omitempty" yaml:"output_models_file_name"`
+}
+
+func Parse(req *plugin.GenerateRequest) (*Options, error) {
+	options, err := parseOpts(req)
+	if err != nil {
+		return nil, err
+	}
+
+	return options, nil
+}
+
+func parseOpts(req *plugin.GenerateRequest) (*Options, error) {
+	var options Options
+	if len(req.PluginOptions) == 0 {
+		return &options, nil
+	}
+	if err := json.Unmarshal(req.PluginOptions, &options); err != nil {
+		return nil, fmt.Errorf("unmarshalling plugin options: %w", err)
+	}
+
+	return &options, nil
+}
+
+func ValidateOpts(opts *Options) error {
+	// TODO: validate options
+
+	return nil
+}
+

internal/template.go

@@ -0,0 +1,6 @@
+package perl
+
+import "embed"
+
+//go:embed templates/*
+var templates embed.FS

internal/templates/template.tmpl

@@ -0,0 +1,17 @@
+{{define "modelsFile"}}
+# Code generated by sqlc. DO NOT EDIT.
+# versions:
+#   sqlc: {{.SqlcVersion}}
+
+package Models;
+use strict;
+use warnings;
+
+{{template "modelsCode" .}}
+
+1;
+{{end}}
+
+{{define "modelsCode"}}
+Hello
+{{end}}

template.go

@@ -0,0 +1,7 @@
+package perl
+
+import "embed"
+
+//go:embed templates/*
+//go:embed templates/*/*
+var templates embed.FS