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