fix(cli): generate

Author: cp-20

cli/generator.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"fmt"
+	"os"
 
 	"github.com/spf13/cobra"
 	h24w17 "github.com/traP-jp/h24w-17"
@@ -23,7 +24,16 @@ var generateCmd = &cobra.Command{
 		}
 		distPath := args[0]
 
-		g := h24w17.NewGenerator(plan, schema)
+		planContent, err := readFile(plan)
+		if err != nil {
+			return fmt.Errorf("error reading plan file: %v", err)
+		}
+		schemaContent, err := readFile(schema)
+		if err != nil {
+			return fmt.Errorf("error reading schema file: %v", err)
+		}
+
+		g := h24w17.NewGenerator(planContent, schemaContent)
 		g.Generate(distPath)
 
 		return nil
@@ -35,3 +45,11 @@ func init() {
 	generateCmd.Flags().StringP("schema", "s", "schema.sql", "File containing the table schema")
 	rootCmd.AddCommand(generateCmd)
 }
+
+func readFile(path string) (string, error) {
+	data, err := os.ReadFile(path)
+	if err != nil {
+		return "", fmt.Errorf("error reading file: %v", err)
+	}
+	return string(data), nil
+}