diff options
Diffstat (limited to 'cmd')
| -rw-r--r-- | cmd/vartan-go/generate.go | 153 | ||||
| -rw-r--r-- | cmd/vartan-go/main.go | 12 | ||||
| -rw-r--r-- | cmd/vartan/parse.go | 20 |
3 files changed, 176 insertions, 9 deletions
diff --git a/cmd/vartan-go/generate.go b/cmd/vartan-go/generate.go new file mode 100644 index 0000000..27f7236 --- /dev/null +++ b/cmd/vartan-go/generate.go @@ -0,0 +1,153 @@ +package main + +import ( + "encoding/json" + "fmt" + "io/ioutil" + "os" + "runtime/debug" + + mldriver "github.com/nihei9/maleeni/driver" + "github.com/nihei9/vartan/driver" + "github.com/nihei9/vartan/spec" + "github.com/spf13/cobra" +) + +func Execute() error { + err := generateCmd.Execute() + if err != nil { + fmt.Fprintf(os.Stderr, "%v\n", err) + return err + } + + return nil +} + +var generateFlags = struct { + pkgName *string +}{} + +var generateCmd = &cobra.Command{ + Use: "vartan-go", + Short: "Generate a parser for Go", + Long: `vartan-go generates a parser for Go.`, + Example: ` vartan-go grammar.json`, + Args: cobra.ExactArgs(1), + RunE: runGenerate, + SilenceErrors: true, + SilenceUsage: true, +} + +func init() { + generateFlags.pkgName = generateCmd.Flags().StringP("package", "p", "main", "package name") +} + +func runGenerate(cmd *cobra.Command, args []string) (retErr error) { + defer func() { + panicked := false + v := recover() + if v != nil { + err, ok := v.(error) + if !ok { + retErr = fmt.Errorf("an unexpected error occurred: %v", v) + fmt.Fprintf(os.Stderr, "%v:\n%v", retErr, string(debug.Stack())) + return + } + + retErr = err + panicked = true + } + + if retErr != nil { + if panicked { + fmt.Fprintf(os.Stderr, "%v:\n%v", retErr, string(debug.Stack())) + } else { + fmt.Fprintf(os.Stderr, "%v\n", retErr) + } + } + }() + + cgram, err := readCompiledGrammar(args[0]) + if err != nil { + return fmt.Errorf("Cannot read a compiled grammar: %w", err) + } + + { + b, err := mldriver.GenLexer(cgram.LexicalSpecification.Maleeni.Spec, *generateFlags.pkgName) + if err != nil { + return fmt.Errorf("Failed to generate a lexer: %w", err) + } + + filePath := fmt.Sprintf("%v_lexer.go", cgram.Name) + + f, err := os.OpenFile(filePath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0644) + if err != nil { + return fmt.Errorf("Failed to create an output file: %v", err) + } + defer f.Close() + + _, err = f.Write(b) + if err != nil { + return fmt.Errorf("Failed to write lexer source code: %v", err) + } + } + + { + b, err := driver.GenParser(cgram, *generateFlags.pkgName) + if err != nil { + return fmt.Errorf("Failed to generate a parser: %w", err) + } + + filePath := fmt.Sprintf("%v_parser.go", cgram.Name) + + f, err := os.OpenFile(filePath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0644) + if err != nil { + return fmt.Errorf("Failed to create an output file: %v", err) + } + defer f.Close() + + _, err = f.Write(b) + if err != nil { + return fmt.Errorf("Failed to write parser source code: %v", err) + } + } + + { + b, err := driver.GenSemanticAction(*generateFlags.pkgName) + if err != nil { + return fmt.Errorf("Failed to generate a semantic action set: %w", err) + } + + filePath := fmt.Sprintf("%v_semantic_action.go", cgram.Name) + + f, err := os.OpenFile(filePath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0644) + if err != nil { + return fmt.Errorf("Failed to create an output file: %v", err) + } + defer f.Close() + + _, err = f.Write(b) + if err != nil { + return fmt.Errorf("Failed to write semantic action source code: %v", err) + } + } + + return nil +} + +func readCompiledGrammar(path string) (*spec.CompiledGrammar, error) { + f, err := os.Open(path) + if err != nil { + return nil, err + } + data, err := ioutil.ReadAll(f) + if err != nil { + return nil, err + } + cgram := &spec.CompiledGrammar{} + err = json.Unmarshal(data, cgram) + if err != nil { + return nil, err + } + return cgram, nil +} diff --git a/cmd/vartan-go/main.go b/cmd/vartan-go/main.go new file mode 100644 index 0000000..701f02f --- /dev/null +++ b/cmd/vartan-go/main.go @@ -0,0 +1,12 @@ +package main + +import ( + "os" +) + +func main() { + err := Execute() + if err != nil { + os.Exit(1) + } +} diff --git a/cmd/vartan/parse.go b/cmd/vartan/parse.go index caf1cbd..9b15251 100644 --- a/cmd/vartan/parse.go +++ b/cmd/vartan/parse.go @@ -63,7 +63,7 @@ func runParse(cmd *cobra.Command, args []string) (retErr error) { return fmt.Errorf("You cannot enable --only-parse and --cst at the same time") } - cgram, err := readCompiledGrammar(args[0]) + cg, err := readCompiledGrammar(args[0]) if err != nil { return fmt.Errorf("Cannot read a compiled grammar: %w", err) } @@ -81,13 +81,15 @@ func runParse(cmd *cobra.Command, args []string) (retErr error) { src = f } + gram := driver.NewGrammar(cg) + var opts []driver.ParserOption { switch { case *parseFlags.cst: - treeAct = driver.NewSyntaxTreeActionSet(cgram, false, true) + treeAct = driver.NewSyntaxTreeActionSet(gram, false, true) case !*parseFlags.onlyParse: - treeAct = driver.NewSyntaxTreeActionSet(cgram, true, false) + treeAct = driver.NewSyntaxTreeActionSet(gram, true, false) } if treeAct != nil { opts = append(opts, driver.SemanticAction(treeAct)) @@ -98,12 +100,12 @@ func runParse(cmd *cobra.Command, args []string) (retErr error) { } } - toks, err := driver.NewTokenStream(cgram, src) + toks, err := driver.NewTokenStream(cg, src) if err != nil { return err } - p, err = driver.NewParser(toks, driver.NewGrammar(cgram), opts...) + p, err = driver.NewParser(toks, gram, opts...) if err != nil { return err } @@ -125,7 +127,7 @@ func runParse(cmd *cobra.Command, args []string) (retErr error) { case tok.Invalid(): msg = fmt.Sprintf("'%v' (<invalid>)", string(tok.Lexeme())) default: - t := cgram.ParsingTable.Terminals[tok.TerminalID()] + t := cg.ParsingTable.Terminals[tok.TerminalID()] msg = fmt.Sprintf("'%v' (%v)", string(tok.Lexeme()), t) } @@ -161,10 +163,10 @@ func readCompiledGrammar(path string) (*spec.CompiledGrammar, error) { if err != nil { return nil, err } - cgram := &spec.CompiledGrammar{} - err = json.Unmarshal(data, cgram) + cg := &spec.CompiledGrammar{} + err = json.Unmarshal(data, cg) if err != nil { return nil, err } - return cgram, nil + return cg, nil } |