vsc => vstruct

Author: lemon-mint

.goreleaser.yaml

@@ -9,7 +9,7 @@ before:
 builds:
   - env:
       - CGO_ENABLED=0
-    main: ./cli/vsc
+    main: ./cli/vstruct
     goos:
       - linux
       - windows
@@ -30,7 +30,7 @@ builds:
       - -X main.Version={{.Version}}
 archives:
   - files:
-    - ./cli/vsc/LICENSE.txt
+    - ./cli/vstruct/LICENSE.txt
     - ./README.md
 checksum:
   name_template: 'checksums.txt'

README.md

@@ -15,15 +15,21 @@ Code Generation Based High Speed Data Serialization Tool
 npm install -g vstruct
 ```
 
-## 2. From Source
+## 2. Go Get (Go Install) (also recommended)
+
+```bash
+go install github.com/lemon-mint/vstruct/cli/vstruct@latest
+```
+
+## 3. From Source
 
 ```
 git clone https://github.com/lemon-mint/vstruct.git
 cd vstruct
-go build -o vstruct ./cli/vsc
+go build -o vstruct ./cli/vstruct
 ```
 
-## 3. Pre-compiled binaries
+## 4. Pre-compiled binaries
 
 [https://github.com/lemon-mint/vstruct/releases/latest](https://github.com/lemon-mint/vstruct/releases/latest)
 

cli/vsc/main.go

@@ -0,0 +1,126 @@
+package main
+
+import (
+	"bytes"
+	"fmt"
+	"io"
+	"log"
+	"os"
+	"path/filepath"
+	"strings"
+
+	"github.com/lemon-mint/vstruct/compile/backend/dart"
+	"github.com/lemon-mint/vstruct/compile/backend/golang"
+	"github.com/lemon-mint/vstruct/compile/backend/python"
+	"github.com/lemon-mint/vstruct/compile/backend/rust"
+	"github.com/lemon-mint/vstruct/compile/frontend"
+	"github.com/lemon-mint/vstruct/ir"
+	"github.com/lemon-mint/vstruct/lexer"
+	"github.com/lemon-mint/vstruct/parser"
+)
+
+func ReadFileAsString(fileName string) string {
+	os.Open(fileName)
+	file, err := os.Open(fileName)
+	if err != nil {
+		panic(err)
+	}
+	defer file.Close()
+	data, err := io.ReadAll(file)
+	if err != nil {
+		panic(err)
+	}
+	return string(data)
+}
+
+func CreateDirectory(path string) error {
+	if _, err := os.Stat(path); os.IsNotExist(err) {
+		err := os.MkdirAll(path, 0755)
+		if err != nil {
+			return fmt.Errorf("error creating directory: %w", err)
+		}
+	}
+	return nil
+}
+
+func main() {
+	if len(os.Args) < 3 {
+		fmt.Println("Usage: vstruct <language: go, rust, dart, py> <file>...")
+		os.Exit(1)
+	}
+	generate := (func(w io.Writer, i *ir.IR, packageName string) error)(nil)
+	modelPath := filepath.Join("vstruct", "model")
+	ext := ""
+
+	language := os.Args[1]
+	switch language {
+	case "go":
+		generate = golang.Generate
+		modelPath = filepath.Join(".", modelPath)
+		ext = ".go"
+	case "rust":
+		generate = rust.Generate
+		modelPath = filepath.Join("src", modelPath)
+		ext = ".rs"
+	case "dart":
+		generate = dart.Generate
+		modelPath = filepath.Join("lib", modelPath)
+		ext = ".dart"
+	case "py":
+		generate = python.Generate
+		modelPath = filepath.Join(".", modelPath)
+		ext = ".py"
+	default:
+		fmt.Printf("Unknown language: %s\n", language)
+		os.Exit(1)
+	}
+
+	for _, path := range os.Args[2:] {
+		if !strings.HasSuffix(path, ".vstruct") {
+			path = path + ".vstruct"
+		}
+
+		filename, err := filepath.Abs(path)
+		if err != nil {
+			log.Println(err)
+			continue
+		}
+
+		input := ReadFileAsString(filename)
+		lex := lexer.NewLexer([]rune(input), path)
+		p := parser.New(lex)
+		file, err := p.Parse()
+		if err != nil {
+			panic(err)
+		}
+		front := frontend.New(file)
+		err = front.Compile()
+		if err != nil {
+			panic(err)
+		}
+		goir := front.Output()
+		goir.Options.UseUnsafe = true
+
+		packageName := filepath.Base(path)
+		packageName = strings.TrimSuffix(packageName, ".vstruct")
+
+		var buf bytes.Buffer
+		err = generate(&buf, goir, packageName)
+		if err != nil {
+			fmt.Println(buf.String())
+			panic(err)
+		}
+		out := buf.String()
+		fmt.Println(out)
+
+		if err := CreateDirectory(filepath.Join(modelPath, packageName)); err != nil {
+			panic(err)
+		}
+		f, err := os.Create(filepath.Join(modelPath, packageName, packageName+ext))
+		if err != nil {
+			panic(err)
+		}
+		f.WriteString(out)
+		f.Close()
+	}
+}