Merge pull request #3 from tomtwinkle/fix/include-go-files

Fixes to embed Go files

Author: tomtwinkle

main.go

@@ -2,12 +2,16 @@ package main
 
 import (
 	"context"
+	"embed"
 	"fmt"
 
 	"github.com/sqlc-dev/plugin-sdk-go/codegen"
 	"github.com/sqlc-dev/plugin-sdk-go/plugin"
 )
 
+//go:embed templates/*
+var templates embed.FS
+
 const (
 	generateFileName = "bulk.sql.go"
 

parser.go

@@ -5,13 +5,12 @@ import (
 	"go/ast"
 	"go/parser"
 	"go/token"
-	"os"
 )
 
 // parseGoCode parses a Go source file and extracts the code of a specific function by its name.
 // It returns the function code as a byte slice or an error if the function is not found.
 func parseGoCode(sourceFile string, targetFuncName string) ([]byte, error) {
-	srcBytes, err := os.ReadFile(sourceFile)
+	srcBytes, err := templates.ReadFile(sourceFile)
 	if err != nil {
 		return nil, err
 	}

template.go

@@ -4,7 +4,6 @@ import (
 	"bufio"
 	"bytes"
 	"context"
-	"embed"
 	"fmt"
 	"go/format"
 	"strconv"
@@ -14,9 +13,6 @@ import (
 	"github.com/sqlc-dev/plugin-sdk-go/sdk"
 )
 
-//go:embed templates/*
-var templates embed.FS
-
 func executeTemplate(
 	_ context.Context, templateName string, data any,
 ) ([]byte, error) {