|  | 
|  | 1 | +package main | 
|  | 2 | + | 
|  | 3 | +import ( | 
|  | 4 | +	"encoding/json" | 
|  | 5 | +	"flag" | 
|  | 6 | +	"fmt" | 
|  | 7 | +	"os" | 
|  | 8 | +	"strings" | 
|  | 9 | + | 
|  | 10 | +	_ "embed" | 
|  | 11 | + | 
|  | 12 | +	"gopkg.in/yaml.v3" | 
|  | 13 | +) | 
|  | 14 | + | 
|  | 15 | +type YamlSnippet struct { | 
|  | 16 | +	Name        string | 
|  | 17 | +	Description string | 
|  | 18 | +	Prefix      string | 
|  | 19 | +	Scope       string | 
|  | 20 | +	Body        string | 
|  | 21 | +} | 
|  | 22 | + | 
|  | 23 | +//go:embed version.txt | 
|  | 24 | +var version []byte | 
|  | 25 | + | 
|  | 26 | +//go:embed help.md | 
|  | 27 | +var help []byte | 
|  | 28 | + | 
|  | 29 | +func readYamlFile(yamlFilePath string) (map[string]YamlSnippet, error) { | 
|  | 30 | + | 
|  | 31 | +	yamlFile, err := os.ReadFile(yamlFilePath) | 
|  | 32 | + | 
|  | 33 | +	if err != nil { | 
|  | 34 | +		return nil, err | 
|  | 35 | +	} | 
|  | 36 | + | 
|  | 37 | +	data := make(map[string]YamlSnippet) | 
|  | 38 | + | 
|  | 39 | +	err = yaml.Unmarshal(yamlFile, &data) | 
|  | 40 | + | 
|  | 41 | +	if err != nil { | 
|  | 42 | +		return nil, err | 
|  | 43 | +	} | 
|  | 44 | +	return data, nil | 
|  | 45 | +} | 
|  | 46 | + | 
|  | 47 | +func generateJsonSnippets(yamlSnippets map[string]YamlSnippet) ([]byte, error) { | 
|  | 48 | +	VSCodeSnippets := make(map[string]interface{}) | 
|  | 49 | + | 
|  | 50 | +	for _, snippet := range yamlSnippets { | 
|  | 51 | +		body := strings.Split(snippet.Body, "\n") | 
|  | 52 | + | 
|  | 53 | +		VSCodeSnippets[snippet.Name] = map[string]interface{}{ | 
|  | 54 | +			"prefix":      snippet.Prefix, | 
|  | 55 | +			"description": snippet.Description, | 
|  | 56 | +			"scope":       snippet.Scope, | 
|  | 57 | +			"body":        body, | 
|  | 58 | +		} | 
|  | 59 | + | 
|  | 60 | +	} | 
|  | 61 | + | 
|  | 62 | +	bSnippets, err := json.MarshalIndent(VSCodeSnippets, "", "  ") | 
|  | 63 | + | 
|  | 64 | +	if err != nil { | 
|  | 65 | +		return nil, err | 
|  | 66 | +	} | 
|  | 67 | +	return bSnippets, nil | 
|  | 68 | +} | 
|  | 69 | + | 
|  | 70 | +func writeJsonFile(jsonFilePath string, jsonData []byte) error { | 
|  | 71 | + | 
|  | 72 | +	f, err := os.Create(jsonFilePath) | 
|  | 73 | + | 
|  | 74 | +	if err != nil { | 
|  | 75 | +		return err | 
|  | 76 | +	} | 
|  | 77 | + | 
|  | 78 | +	defer f.Close() | 
|  | 79 | + | 
|  | 80 | +	_, err = f.WriteString(string(jsonData)) | 
|  | 81 | + | 
|  | 82 | +	if err != nil { | 
|  | 83 | +		return err | 
|  | 84 | +	} | 
|  | 85 | +	return nil | 
|  | 86 | +} | 
|  | 87 | + | 
|  | 88 | +func parse(command string, args []string) error { | 
|  | 89 | +	switch command { | 
|  | 90 | + | 
|  | 91 | +	case "generate", "gen": | 
|  | 92 | + | 
|  | 93 | +		flagSet := flag.NewFlagSet("generate", flag.ExitOnError) | 
|  | 94 | + | 
|  | 95 | +		input := flagSet.String("input", "", "input yaml file path") | 
|  | 96 | +		output := flagSet.String("output", "", "output json file path") | 
|  | 97 | + | 
|  | 98 | +		flagSet.Parse(args[0:]) | 
|  | 99 | + | 
|  | 100 | +		yamlData, err := readYamlFile(*input) | 
|  | 101 | +		if err != nil { | 
|  | 102 | +			fmt.Println("π‘", err.Error()) | 
|  | 103 | +			os.Exit(1) | 
|  | 104 | +		} | 
|  | 105 | +		jsonData, err := generateJsonSnippets(yamlData) | 
|  | 106 | +		if err != nil { | 
|  | 107 | +			fmt.Println("π‘", err.Error()) | 
|  | 108 | +			os.Exit(1) | 
|  | 109 | +		} | 
|  | 110 | + | 
|  | 111 | +		err = writeJsonFile(*output, jsonData) | 
|  | 112 | + | 
|  | 113 | +		if err != nil { | 
|  | 114 | +			fmt.Println("π‘", err.Error()) | 
|  | 115 | +			os.Exit(1) | 
|  | 116 | +		} | 
|  | 117 | +		fmt.Println("π", *output, "generated") | 
|  | 118 | +		//os.Exit(0) | 
|  | 119 | +		return nil | 
|  | 120 | + | 
|  | 121 | +	case "version": | 
|  | 122 | +		fmt.Println(string(version)) | 
|  | 123 | +		//os.Exit(0) | 
|  | 124 | +		return nil | 
|  | 125 | + | 
|  | 126 | +	case "help": | 
|  | 127 | +		fmt.Println(string(help)) | 
|  | 128 | +		//os.Exit(0) | 
|  | 129 | +		return nil | 
|  | 130 | + | 
|  | 131 | +	default: | 
|  | 132 | +		return fmt.Errorf("π΄ invalid command") | 
|  | 133 | +	} | 
|  | 134 | +} | 
|  | 135 | + | 
|  | 136 | +func main() { | 
|  | 137 | + | 
|  | 138 | +	flag.Parse() | 
|  | 139 | + | 
|  | 140 | +	if len(flag.Args()) < 1 { | 
|  | 141 | +		fmt.Println("π΄ invalid command") | 
|  | 142 | +		os.Exit(0) | 
|  | 143 | +	} | 
|  | 144 | + | 
|  | 145 | +	command := flag.Args()[0] | 
|  | 146 | + | 
|  | 147 | +	parse(command, flag.Args()[1:]) | 
|  | 148 | + | 
|  | 149 | +} | 
0 commit comments