template / files / internal / core / app.go
app.go
37160b89 2026-05-01 Pavel Safronov | 1 | package core |
f819f801 2022-09-21 Pavel Safronov | 2 | |
| 3 | import ( | |
bbd011d0 2022-12-10 Pavel Safronov | 4 | "bufio" |
f819f801 2022-09-21 Pavel Safronov | 5 | "fmt" |
6962cd0b 2023-09-14 Pavel Safronov | 6 | "io/fs" |
f819f801 2022-09-21 Pavel Safronov | 7 | "os" |
05f9e8ef 2022-10-06 Pavel Safronov | 8 | "os/user" |
6962cd0b 2023-09-14 Pavel Safronov | 9 | "path" |
| 10 | "path/filepath" | |
9573ca02 2022-11-16 Pavel Safronov | 11 | "strconv" |
4a7759db 2022-09-24 Pavel Safronov | 12 | "strings" |
f819f801 2022-09-21 Pavel Safronov | 13 | |
886f7fa2 2022-11-02 Pavel Safronov | 14 | log "github.com/sirupsen/logrus" |
37160b89 2026-05-01 Pavel Safronov | 15 | "github.com/urfave/cli/v2" |
4a7759db 2022-09-24 Pavel Safronov | 16 | "gopkg.in/yaml.v3" |
f819f801 2022-09-21 Pavel Safronov | 17 | ) |
| 18 | ||
ae89a42a 2022-11-02 Pavel Safronov | 19 | const defaultUserConfigPath = "~/.template.user.yaml" |
| 20 | const defaultConfigPath = "~/.template.yaml" | |
886f7fa2 2022-11-02 Pavel Safronov | 21 | const defaultLogLevel = "info" |
ae89a42a 2022-11-02 Pavel Safronov | 22 | |
37160b89 2026-05-01 Pavel Safronov | 23 | type argumentValues struct { |
ae89a42a 2022-11-02 Pavel Safronov | 24 | profile string |
| 25 | projectName string | |
8fc7c452 2023-03-29 Pavel Safronov | 26 | fileTemplate string |
| 27 | dirTemplate string | |
fd0a4446 2023-04-03 Pavel Safronov | 28 | scriptTemplate string |
ae89a42a 2022-11-02 Pavel Safronov | 29 | configFilePath string |
| 30 | userConfigPath string | |
| 31 | extrasFilePath string | |
886f7fa2 2022-11-02 Pavel Safronov | 32 | logLevel string |
9573ca02 2022-11-16 Pavel Safronov | 33 | } |
37160b89 2026-05-01 Pavel Safronov | 34 | type envVarValues struct { |
9573ca02 2022-11-16 Pavel Safronov | 35 | useDefaultConfig bool |
b1ded4b4 2022-10-30 Pavel Safronov | 36 | } |
37160b89 2026-05-01 Pavel Safronov | 37 | type settings struct { |
| 38 | argumentValues | |
| 39 | envVarValues | |
bbd011d0 2022-12-10 Pavel Safronov | 40 | extras map[string]string |
| 41 | fullConfigPath string | |
| 42 | config Config | |
| 43 | usr *user.User | |
b1ded4b4 2022-10-30 Pavel Safronov | 44 | } |
| 45 | ||
37160b89 2026-05-01 Pavel Safronov | 46 | // Run is the entry point for the CLI application. |
| 47 | func Run(args []string, version string) error { | |
| 48 | av := argumentValues{} | |
f819f801 2022-09-21 Pavel Safronov | 49 | |
| 50 | app := &cli.App{ | |
| 51 | Name: "template", | |
| 52 | Usage: "create file or directory from a template", | |
| 53 | EnableBashCompletion: true, | |
185f54e8 2022-10-05 Pavel Safronov | 54 | Suggest: true, |
f819f801 2022-09-21 Pavel Safronov | 55 | Flags: []cli.Flag{ |
| 56 | &cli.StringFlag{ | |
ea896f47 2022-09-23 Pavel Safronov | 57 | Name: "profile", |
| 58 | Aliases: []string{"p"}, | |
| 59 | Value: "", | |
f819f801 2022-09-21 Pavel Safronov | 60 | Usage: "`PROFILE` to use for the template", |
37160b89 2026-05-01 Pavel Safronov | 61 | Destination: &av.profile, |
f819f801 2022-09-21 Pavel Safronov | 62 | }, |
8fc7c452 2023-03-29 Pavel Safronov | 63 | &cli.StringFlag{ |
| 64 | Name: "file", | |
| 65 | Aliases: []string{"f"}, | |
| 66 | Value: "", | |
| 67 | Usage: "file template to use to create a file", | |
37160b89 2026-05-01 Pavel Safronov | 68 | Destination: &av.fileTemplate, |
8fc7c452 2023-03-29 Pavel Safronov | 69 | }, |
| 70 | &cli.StringFlag{ | |
| 71 | Name: "dir", | |
| 72 | Aliases: []string{"d"}, | |
| 73 | Value: "", | |
| 74 | Usage: "folder pattern to use to create a directory", | |
37160b89 2026-05-01 Pavel Safronov | 75 | Destination: &av.dirTemplate, |
8fc7c452 2023-03-29 Pavel Safronov | 76 | }, |
fd0a4446 2023-04-03 Pavel Safronov | 77 | &cli.StringFlag{ |
| 78 | Name: "script", | |
| 79 | Aliases: []string{"s"}, | |
| 80 | Value: "", | |
| 81 | Usage: "templated script to run in the shell", | |
37160b89 2026-05-01 Pavel Safronov | 82 | Destination: &av.scriptTemplate, |
fd0a4446 2023-04-03 Pavel Safronov | 83 | }, |
d026838a 2022-09-22 Pavel Safronov | 84 | &cli.StringFlag{ |
| 85 | Name: "name", | |
| 86 | Aliases: []string{"n"}, | |
| 87 | Usage: "name of the project to create", | |
| 88 | Value: "", | |
37160b89 2026-05-01 Pavel Safronov | 89 | Destination: &av.projectName, |
d026838a 2022-09-22 Pavel Safronov | 90 | }, |
185f54e8 2022-10-05 Pavel Safronov | 91 | &cli.PathFlag{ |
| 92 | Name: "config-yaml", | |
| 93 | Aliases: []string{"c"}, | |
| 94 | Usage: "path to the config yaml file", | |
ae89a42a 2022-11-02 Pavel Safronov | 95 | Value: defaultConfigPath, |
37160b89 2026-05-01 Pavel Safronov | 96 | Destination: &av.configFilePath, |
185f54e8 2022-10-05 Pavel Safronov | 97 | }, |
ae89a42a 2022-11-02 Pavel Safronov | 98 | &cli.PathFlag{ |
| 99 | Name: "user-config-yaml", | |
| 100 | Aliases: []string{"u"}, | |
| 101 | Usage: "path to the user config yaml file", | |
| 102 | Value: defaultUserConfigPath, | |
37160b89 2026-05-01 Pavel Safronov | 103 | Destination: &av.userConfigPath, |
ae89a42a 2022-11-02 Pavel Safronov | 104 | }, |
185f54e8 2022-10-05 Pavel Safronov | 105 | &cli.PathFlag{ |
55c36f6a 2022-10-05 Pavel Safronov | 106 | Name: "extras-yaml", |
| 107 | Aliases: []string{"e"}, | |
| 108 | Usage: "path to the extras yaml file", | |
| 109 | Value: "~/.template-extras.yaml", | |
37160b89 2026-05-01 Pavel Safronov | 110 | Destination: &av.extrasFilePath, |
185f54e8 2022-10-05 Pavel Safronov | 111 | }, |
886f7fa2 2022-11-02 Pavel Safronov | 112 | &cli.StringFlag{ |
| 113 | Name: "log-level", | |
| 114 | Aliases: []string{"l"}, | |
| 115 | Usage: "log level", | |
| 116 | Value: defaultLogLevel, | |
37160b89 2026-05-01 Pavel Safronov | 117 | Destination: &av.logLevel, |
886f7fa2 2022-11-02 Pavel Safronov | 118 | }, |
f819f801 2022-09-21 Pavel Safronov | 119 | }, |
8c15a714 2022-11-04 Pavel Safronov | 120 | Commands: []*cli.Command{ |
| 121 | { | |
| 122 | Name: "list", | |
| 123 | Usage: "list available profiles", | |
| 124 | Action: func(c *cli.Context) error { | |
37160b89 2026-05-01 Pavel Safronov | 125 | s, err := createSettings(av, c) |
8c15a714 2022-11-04 Pavel Safronov | 126 | if err != nil { |
| 127 | return fmt.Errorf("error creating settings: %v", err) | |
| 128 | } | |
| 129 | ||
37160b89 2026-05-01 Pavel Safronov | 130 | cfg := s.config |
8c15a714 2022-11-04 Pavel Safronov | 131 | fmt.Println("Available profiles:") |
37160b89 2026-05-01 Pavel Safronov | 132 | for i, p := range cfg.Profiles { |
7add4991 2026-01-31 Pavel Safronov | 133 | fmt.Printf(" %d: %v\n", i+1, p.Name) |
83f12389 2026-01-15 Pavel Safronov | 134 | fmt.Printf(" %v\n", p.Description) |
8c15a714 2022-11-04 Pavel Safronov | 135 | } |
f58f2e35 2023-01-30 Pavel Safronov | 136 | fmt.Println() |
8c15a714 2022-11-04 Pavel Safronov | 137 | return nil |
| 138 | }, | |
| 139 | }, | |
6962cd0b 2023-09-14 Pavel Safronov | 140 | { |
| 141 | Name: "create", | |
| 142 | Usage: "create a fully-hardcoded template profile from a file or folder", | |
| 143 | Flags: []cli.Flag{ | |
| 144 | &cli.StringFlag{ | |
| 145 | Name: "dir", | |
| 146 | Aliases: []string{"d"}, | |
| 147 | Value: "", | |
| 148 | Usage: "folder from which to create the template", | |
37160b89 2026-05-01 Pavel Safronov | 149 | Destination: &av.dirTemplate, |
6962cd0b 2023-09-14 Pavel Safronov | 150 | }, |
| 151 | &cli.StringFlag{ | |
| 152 | Name: "file", | |
| 153 | Aliases: []string{"f"}, | |
| 154 | Value: "", | |
| 155 | Usage: "file from which to create the template", | |
37160b89 2026-05-01 Pavel Safronov | 156 | Destination: &av.fileTemplate, |
6962cd0b 2023-09-14 Pavel Safronov | 157 | }, |
| 158 | }, | |
| 159 | Action: func(c *cli.Context) error { | |
| 160 | c.Args() | |
| 161 | ||
37160b89 2026-05-01 Pavel Safronov | 162 | filePath := av.fileTemplate |
| 163 | dirPath := av.dirTemplate | |
6962cd0b 2023-09-14 Pavel Safronov | 164 | if filePath == "" && dirPath == "" { |
| 165 | return fmt.Errorf("file or dir not specified, use `file` or `dir` flag") | |
| 166 | } | |
| 167 | ||
37160b89 2026-05-01 Pavel Safronov | 168 | profileName := av.profile |
6962cd0b 2023-09-14 Pavel Safronov | 169 | if profileName == "" { |
| 170 | profileName = "temp" | |
| 171 | } | |
| 172 | ||
| 173 | sourcePath := filePath | |
| 174 | profileType := FileType | |
| 175 | if dirPath != "" { | |
| 176 | profileType = FolderType | |
| 177 | sourcePath = dirPath | |
| 178 | } | |
| 179 | profile := Profile{ | |
| 180 | Name: profileName, | |
| 181 | Description: fmt.Sprintf("temp %s profile generated from %s", profileType, sourcePath), | |
| 182 | Type: profileType, | |
| 183 | TargetDirectory: ".", | |
| 184 | } | |
| 185 | ||
| 186 | dir, file := path.Split(sourcePath) | |
| 187 | profile.TargetDirectory = dir | |
| 188 | profile.NamePatterns = []ParameterizedText{ | |
| 189 | ParameterizedText(file), | |
| 190 | } | |
| 191 | profile.PostScripts = []ParameterizedText{ | |
| 192 | ParameterizedText(fmt.Sprintf("code ./%s", file)), | |
| 193 | } | |
| 194 | ||
| 195 | if filePath != "" { | |
37160b89 2026-05-01 Pavel Safronov | 196 | fileContent, err := readFile(filePath) |
6962cd0b 2023-09-14 Pavel Safronov | 197 | if err != nil { |
| 198 | return fmt.Errorf("error reading file %s: %w", filePath, err) | |
| 199 | } | |
37160b89 2026-05-01 Pavel Safronov | 200 | profile.FileTemplate = ParameterizedText(fileContent) |
6962cd0b 2023-09-14 Pavel Safronov | 201 | } else { |
37160b89 2026-05-01 Pavel Safronov | 202 | var files []FileInFolder |
| 203 | err := filepath.WalkDir(dirPath, func(p string, d fs.DirEntry, err error) error { | |
6962cd0b 2023-09-14 Pavel Safronov | 204 | if err != nil { |
37160b89 2026-05-01 Pavel Safronov | 205 | fmt.Printf("prevent panic by handling failure accessing a path %q: %v\n", p, err) |
6962cd0b 2023-09-14 Pavel Safronov | 206 | return err |
| 207 | } | |
| 208 | ||
37160b89 2026-05-01 Pavel Safronov | 209 | isRoot := p == dirPath |
6962cd0b 2023-09-14 Pavel Safronov | 210 | isDir := d.IsDir() |
37160b89 2026-05-01 Pavel Safronov | 211 | if isRoot || isDir { |
6962cd0b 2023-09-14 Pavel Safronov | 212 | return nil |
| 213 | } | |
37160b89 2026-05-01 Pavel Safronov | 214 | relPath, err := filepath.Rel(dirPath, p) |
6962cd0b 2023-09-14 Pavel Safronov | 215 | if err != nil { |
37160b89 2026-05-01 Pavel Safronov | 216 | return fmt.Errorf("error getting relative path for %s: %w", p, err) |
6962cd0b 2023-09-14 Pavel Safronov | 217 | } |
37160b89 2026-05-01 Pavel Safronov | 218 | contents, err := readFile(p) |
6962cd0b 2023-09-14 Pavel Safronov | 219 | if err != nil { |
37160b89 2026-05-01 Pavel Safronov | 220 | return fmt.Errorf("error reading file %s: %w", p, err) |
6962cd0b 2023-09-14 Pavel Safronov | 221 | } |
37160b89 2026-05-01 Pavel Safronov | 222 | files = append(files, FileInFolder{ |
6962cd0b 2023-09-14 Pavel Safronov | 223 | Name: ParameterizedText(relPath), |
| 224 | FileTemplate: ParameterizedText(contents), | |
37160b89 2026-05-01 Pavel Safronov | 225 | }) |
6962cd0b 2023-09-14 Pavel Safronov | 226 | return nil |
| 227 | }) | |
| 228 | if err != nil { | |
| 229 | return fmt.Errorf("error walking dir %s: %w", dirPath, err) | |
| 230 | } | |
| 231 | profile.Files = files | |
| 232 | } | |
| 233 | ||
37160b89 2026-05-01 Pavel Safronov | 234 | cfg := Config{Profiles: []Profile{profile}} |
6962cd0b 2023-09-14 Pavel Safronov | 235 | writer := strings.Builder{} |
| 236 | encoder := yaml.NewEncoder(&writer) | |
37160b89 2026-05-01 Pavel Safronov | 237 | if err := encoder.Encode(cfg); err != nil { |
6962cd0b 2023-09-14 Pavel Safronov | 238 | return fmt.Errorf("error encoding config: %v", err) |
| 239 | } | |
37160b89 2026-05-01 Pavel Safronov | 240 | fmt.Println(writer.String()) |
6962cd0b 2023-09-14 Pavel Safronov | 241 | return nil |
| 242 | }, | |
| 243 | }, | |
8c15a714 2022-11-04 Pavel Safronov | 244 | { |
| 245 | Name: "get", | |
| 246 | Usage: "get a profile", | |
| 247 | Action: func(c *cli.Context) error { | |
37160b89 2026-05-01 Pavel Safronov | 248 | s, err := createSettings(av, c) |
8c15a714 2022-11-04 Pavel Safronov | 249 | if err != nil { |
| 250 | return fmt.Errorf("error creating settings: %v", err) | |
| 251 | } | |
37160b89 2026-05-01 Pavel Safronov | 252 | cfg := s.config |
8c15a714 2022-11-04 Pavel Safronov | 253 | |
37160b89 2026-05-01 Pavel Safronov | 254 | index, err := findProfileIndex(cfg.Profiles, s.profile) |
8c15a714 2022-11-04 Pavel Safronov | 255 | if err != nil { |
37160b89 2026-05-01 Pavel Safronov | 256 | return fmt.Errorf("profile '%v' not found", s.profile) |
8c15a714 2022-11-04 Pavel Safronov | 257 | } |
37160b89 2026-05-01 Pavel Safronov | 258 | profile := cfg.Profiles[index] |
8c15a714 2022-11-04 Pavel Safronov | 259 | |
| 260 | writer := strings.Builder{} | |
| 261 | encoder := yaml.NewEncoder(&writer) | |
37160b89 2026-05-01 Pavel Safronov | 262 | if err = encoder.Encode(profile); err != nil { |
8c15a714 2022-11-04 Pavel Safronov | 263 | return fmt.Errorf("error encoding profile: %v", err) |
| 264 | } | |
| 265 | ||
37160b89 2026-05-01 Pavel Safronov | 266 | fmt.Printf("Profile '%v':\n", s.profile) |
| 267 | fmt.Println(writer.String()) | |
f58f2e35 2023-01-30 Pavel Safronov | 268 | fmt.Println() |
8c15a714 2022-11-04 Pavel Safronov | 269 | return nil |
| 270 | }, | |
| 271 | }, | |
e5793361 2026-01-16 Pavel Safronov | 272 | { |
| 273 | Name: "init", | |
| 274 | Usage: "if global template config is not present, initializes the global template config with default profiles", | |
| 275 | Action: func(c *cli.Context) error { | |
| 276 | usr, err := user.Current() | |
| 277 | if err != nil { | |
| 278 | log.Fatal(err) | |
| 279 | } | |
37160b89 2026-05-01 Pavel Safronov | 280 | fullConfigPath := getFullPath(av.configFilePath, *usr) |
e5793361 2026-01-16 Pavel Safronov | 281 | |
| 282 | _, err = os.Stat(fullConfigPath) | |
| 283 | if err == nil { | |
| 284 | fmt.Printf("Config file already exists at '%v', skipping init\n", fullConfigPath) | |
| 285 | fmt.Println() | |
| 286 | return nil | |
| 287 | } | |
| 288 | ||
| 289 | fmt.Printf("Setting config at '%v' to default\n", fullConfigPath) | |
37160b89 2026-05-01 Pavel Safronov | 290 | if err = saveConfig(fullConfigPath, defaultConfig()); err != nil { |
e5793361 2026-01-16 Pavel Safronov | 291 | return fmt.Errorf("error saving default config: %v", err) |
| 292 | } | |
| 293 | fmt.Println() | |
| 294 | return nil | |
| 295 | }, | |
| 296 | }, | |
8c15a714 2022-11-04 Pavel Safronov | 297 | { |
| 298 | Name: "reset", | |
| 299 | Usage: "reset the global template config to default profiles", | |
| 300 | Action: func(c *cli.Context) error { | |
873846b9 2022-12-09 Pavel Safronov | 301 | usr, err := user.Current() |
8c15a714 2022-11-04 Pavel Safronov | 302 | if err != nil { |
873846b9 2022-12-09 Pavel Safronov | 303 | log.Fatal(err) |
8c15a714 2022-11-04 Pavel Safronov | 304 | } |
37160b89 2026-05-01 Pavel Safronov | 305 | fullConfigPath := getFullPath(av.configFilePath, *usr) |
8c15a714 2022-11-04 Pavel Safronov | 306 | |
| 307 | fmt.Printf("Resetting config at '%v' to default\n", fullConfigPath) | |
37160b89 2026-05-01 Pavel Safronov | 308 | if err = saveConfig(fullConfigPath, defaultConfig()); err != nil { |
8c15a714 2022-11-04 Pavel Safronov | 309 | return fmt.Errorf("error saving default config: %v", err) |
| 310 | } | |
f58f2e35 2023-01-30 Pavel Safronov | 311 | fmt.Println() |
8c15a714 2022-11-04 Pavel Safronov | 312 | return nil |
| 313 | }, | |
| 314 | }, | |
2543ce03 2023-01-29 Pavel Safronov | 315 | { |
| 316 | Name: "version", | |
| 317 | Usage: "print version", | |
| 318 | Action: func(c *cli.Context) error { | |
37160b89 2026-05-01 Pavel Safronov | 319 | fmt.Println(version) |
2543ce03 2023-01-29 Pavel Safronov | 320 | return nil |
| 321 | }, | |
| 322 | }, | |
8c15a714 2022-11-04 Pavel Safronov | 323 | }, |
37160b89 2026-05-01 Pavel Safronov | 324 | Action: func(c *cli.Context) error { |
| 325 | s, err := createSettings(av, c) | |
05f9e8ef 2022-10-06 Pavel Safronov | 326 | if err != nil { |
b1ded4b4 2022-10-30 Pavel Safronov | 327 | return fmt.Errorf("error parsing arguments: %w", err) |
05f9e8ef 2022-10-06 Pavel Safronov | 328 | } |
37160b89 2026-05-01 Pavel Safronov | 329 | if err = runMain(s); err != nil { |
b1ded4b4 2022-10-30 Pavel Safronov | 330 | return fmt.Errorf("error in main logic: %w", err) |
55d96f9c 2022-09-23 Pavel Safronov | 331 | } |
f58f2e35 2023-01-30 Pavel Safronov | 332 | fmt.Println() |
b1ded4b4 2022-10-30 Pavel Safronov | 333 | return nil |
| 334 | }, | |
| 335 | } | |
55d96f9c 2022-09-23 Pavel Safronov | 336 | |
37160b89 2026-05-01 Pavel Safronov | 337 | return app.Run(args) |
b1ded4b4 2022-10-30 Pavel Safronov | 338 | } |
319b8b3f 2022-09-22 Pavel Safronov | 339 | |
37160b89 2026-05-01 Pavel Safronov | 340 | func createSettings(av argumentValues, context *cli.Context) (settings, error) { |
| 341 | ev, err := readEnvVars() | |
9573ca02 2022-11-16 Pavel Safronov | 342 | if err != nil { |
37160b89 2026-05-01 Pavel Safronov | 343 | return settings{}, fmt.Errorf("error getting env var values: %w", err) |
9573ca02 2022-11-16 Pavel Safronov | 344 | } |
873846b9 2022-12-09 Pavel Safronov | 345 | |
37160b89 2026-05-01 Pavel Safronov | 346 | s := settings{ |
| 347 | argumentValues: av, | |
| 348 | envVarValues: ev, | |
b1ded4b4 2022-10-30 Pavel Safronov | 349 | } |
319b8b3f 2022-09-22 Pavel Safronov | 350 | |
b1ded4b4 2022-10-30 Pavel Safronov | 351 | args := context.Args() |
886f7fa2 2022-11-02 Pavel Safronov | 352 | |
8fc7c452 2023-03-29 Pavel Safronov | 353 | var level log.Level |
37160b89 2026-05-01 Pavel Safronov | 354 | switch strings.ToLower(s.logLevel) { |
886f7fa2 2022-11-02 Pavel Safronov | 355 | case "debug": |
8fc7c452 2023-03-29 Pavel Safronov | 356 | level = log.DebugLevel |
886f7fa2 2022-11-02 Pavel Safronov | 357 | case "info": |
8fc7c452 2023-03-29 Pavel Safronov | 358 | level = log.InfoLevel |
886f7fa2 2022-11-02 Pavel Safronov | 359 | case "warn": |
8fc7c452 2023-03-29 Pavel Safronov | 360 | level = log.WarnLevel |
886f7fa2 2022-11-02 Pavel Safronov | 361 | case "error": |
8fc7c452 2023-03-29 Pavel Safronov | 362 | level = log.ErrorLevel |
886f7fa2 2022-11-02 Pavel Safronov | 363 | case "fatal": |
8fc7c452 2023-03-29 Pavel Safronov | 364 | level = log.FatalLevel |
886f7fa2 2022-11-02 Pavel Safronov | 365 | case "panic": |
8fc7c452 2023-03-29 Pavel Safronov | 366 | level = log.PanicLevel |
886f7fa2 2022-11-02 Pavel Safronov | 367 | default: |
37160b89 2026-05-01 Pavel Safronov | 368 | return settings{}, fmt.Errorf("invalid log level specified %v", s.logLevel) |
886f7fa2 2022-11-02 Pavel Safronov | 369 | } |
8fc7c452 2023-03-29 Pavel Safronov | 370 | log.SetLevel(level) |
886f7fa2 2022-11-02 Pavel Safronov | 371 | log.SetOutput(os.Stdout) |
| 372 | ||
| 373 | log.WithFields(log.Fields{"args": args}).Debug("Args parsed") | |
319b8b3f 2022-09-22 Pavel Safronov | 374 | |
37160b89 2026-05-01 Pavel Safronov | 375 | skipProfile := len(s.fileTemplate) > 0 || len(s.dirTemplate) > 0 || len(s.scriptTemplate) > 0 |
8fc7c452 2023-03-29 Pavel Safronov | 376 | currentParamIndex := 0 |
| 377 | ||
| 378 | if !skipProfile { | |
37160b89 2026-05-01 Pavel Safronov | 379 | if s.profile == "" && args.Len() > currentParamIndex { |
| 380 | s.profile = args.Get(currentParamIndex) | |
8fc7c452 2023-03-29 Pavel Safronov | 381 | } |
37160b89 2026-05-01 Pavel Safronov | 382 | if s.profile != "" { |
| 383 | fmt.Println("Profile:", s.profile) | |
7add4991 2026-01-31 Pavel Safronov | 384 | } |
8fc7c452 2023-03-29 Pavel Safronov | 385 | currentParamIndex++ |
b1ded4b4 2022-10-30 Pavel Safronov | 386 | } |
4a7759db 2022-09-24 Pavel Safronov | 387 | |
37160b89 2026-05-01 Pavel Safronov | 388 | if s.projectName == "" && args.Len() > currentParamIndex { |
| 389 | s.projectName = args.Get(currentParamIndex) | |
b1ded4b4 2022-10-30 Pavel Safronov | 390 | } |
37160b89 2026-05-01 Pavel Safronov | 391 | if s.projectName != "" { |
| 392 | fmt.Println("Project name:", s.projectName) | |
7add4991 2026-01-31 Pavel Safronov | 393 | } |
8179bb1c 2022-10-26 Pavel Safronov | 394 | |
37160b89 2026-05-01 Pavel Safronov | 395 | _, err = os.Stat(s.extrasFilePath) |
ae89a42a 2022-11-02 Pavel Safronov | 396 | if err == nil { |
37160b89 2026-05-01 Pavel Safronov | 397 | extras, err := loadExtras(s.extrasFilePath) |
ae89a42a 2022-11-02 Pavel Safronov | 398 | if err != nil { |
37160b89 2026-05-01 Pavel Safronov | 399 | return s, fmt.Errorf("error loading extras from %v: %w", s.extrasFilePath, err) |
ae89a42a 2022-11-02 Pavel Safronov | 400 | } |
37160b89 2026-05-01 Pavel Safronov | 401 | s.extras = extras |
b1ded4b4 2022-10-30 Pavel Safronov | 402 | } |
4a7759db 2022-09-24 Pavel Safronov | 403 | |
b1ded4b4 2022-10-30 Pavel Safronov | 404 | usr, err := user.Current() |
| 405 | if err != nil { | |
| 406 | log.Fatal(err) | |
| 407 | } | |
37160b89 2026-05-01 Pavel Safronov | 408 | s.usr = usr |
4a7361b0 2022-09-21 Pavel Safronov | 409 | |
37160b89 2026-05-01 Pavel Safronov | 410 | fullConfigPath := getFullPath(s.configFilePath, *usr) |
| 411 | s.fullConfigPath = fullConfigPath | |
8c15a714 2022-11-04 Pavel Safronov | 412 | |
f58f2e35 2023-01-30 Pavel Safronov | 413 | var fullConfig Config |
37160b89 2026-05-01 Pavel Safronov | 414 | if s.useDefaultConfig { |
f58f2e35 2023-01-30 Pavel Safronov | 415 | fullConfig = defaultConfig() |
b1ded4b4 2022-10-30 Pavel Safronov | 416 | } else { |
f58f2e35 2023-01-30 Pavel Safronov | 417 | config, err := loadConfig(fullConfigPath) |
b1ded4b4 2022-10-30 Pavel Safronov | 418 | if err != nil { |
37160b89 2026-05-01 Pavel Safronov | 419 | return s, fmt.Errorf("error loading config from %v: %w\nConsider running `template reset` to recreate the config file, but keep in mind that `reset` will remove existing config settings", fullConfigPath, err) |
5e424f1b 2022-11-12 Pavel Safronov | 420 | } |
bbd011d0 2022-12-10 Pavel Safronov | 421 | fmt.Printf("Loaded config from %v\n", fullConfigPath) |
37160b89 2026-05-01 Pavel Safronov | 422 | fullConfig = loadAndMergeUserConfig(s.userConfigPath, usr, config) |
b1ded4b4 2022-10-30 Pavel Safronov | 423 | } |
37160b89 2026-05-01 Pavel Safronov | 424 | s.config = fullConfig |
5e424f1b 2022-11-12 Pavel Safronov | 425 | |
37160b89 2026-05-01 Pavel Safronov | 426 | return s, nil |
8c15a714 2022-11-04 Pavel Safronov | 427 | } |
b1ded4b4 2022-10-30 Pavel Safronov | 428 | |
f58f2e35 2023-01-30 Pavel Safronov | 429 | func loadAndMergeUserConfig(userConfigPath string, usr *user.User, globalConfig Config) Config { |
| 430 | fullUserConfigPath := getFullPath(userConfigPath, *usr) | |
| 431 | _, err := os.Stat(fullUserConfigPath) | |
| 432 | if err != nil { | |
| 433 | fmt.Printf("User config file not found at %v, skipping\n", fullUserConfigPath) | |
| 434 | return globalConfig | |
| 435 | } | |
| 436 | ||
| 437 | userConfig, err := loadConfig(fullUserConfigPath) | |
| 438 | if err != nil { | |
| 439 | fmt.Printf("Error loading user config from %v: %v\n", fullUserConfigPath, err) | |
| 440 | return globalConfig | |
| 441 | } | |
| 442 | ||
| 443 | fmt.Printf("Loaded user config from %v, combining with global config\n", fullUserConfigPath) | |
| 444 | fullConfig, err := mergeConfigs(globalConfig, userConfig) | |
| 445 | if err != nil { | |
| 446 | fmt.Printf("Error merging configs: %v\n", err) | |
| 447 | } | |
| 448 | ||
| 449 | return fullConfig | |
| 450 | } | |
| 451 | ||
37160b89 2026-05-01 Pavel Safronov | 452 | func readEnvVars() (envVarValues, error) { |
| 453 | ev := envVarValues{} | |
9573ca02 2022-11-16 Pavel Safronov | 454 | |
| 455 | useDefaultConfigText := os.Getenv("TEMPLATE_USE_DEFAULT_CONFIG") | |
| 456 | if useDefaultConfigText != "" { | |
| 457 | useDefaultConfig, err := strconv.ParseBool(useDefaultConfigText) | |
| 458 | if err != nil { | |
37160b89 2026-05-01 Pavel Safronov | 459 | return ev, fmt.Errorf("error parsing env var TEMPLATE_USE_DEFAULT_CONFIG: \"%s\" %w", useDefaultConfigText, err) |
9573ca02 2022-11-16 Pavel Safronov | 460 | } |
37160b89 2026-05-01 Pavel Safronov | 461 | ev.useDefaultConfig = useDefaultConfig |
9573ca02 2022-11-16 Pavel Safronov | 462 | } |
| 463 | ||
37160b89 2026-05-01 Pavel Safronov | 464 | return ev, nil |
9573ca02 2022-11-16 Pavel Safronov | 465 | } |
| 466 | ||
37160b89 2026-05-01 Pavel Safronov | 467 | func findProfile(s settings) (Profile, error) { |
| 468 | var profile Profile | |
| 469 | cfg := s.config | |
| 470 | createTempProfile := len(s.fileTemplate) > 0 || len(s.dirTemplate) > 0 || len(s.scriptTemplate) > 0 | |
8fc7c452 2023-03-29 Pavel Safronov | 471 | |
| 472 | if createTempProfile { | |
37160b89 2026-05-01 Pavel Safronov | 473 | if len(s.scriptTemplate) > 0 { |
fd0a4446 2023-04-03 Pavel Safronov | 474 | profile = Profile{ |
| 475 | Name: "temp", | |
| 476 | Description: "temp-script-profile", | |
| 477 | Type: ScriptType, | |
37160b89 2026-05-01 Pavel Safronov | 478 | PostScripts: []ParameterizedText{ParameterizedText(s.scriptTemplate)}, |
fd0a4446 2023-04-03 Pavel Safronov | 479 | } |
| 480 | } else { | |
| 481 | profileType := FileType | |
37160b89 2026-05-01 Pavel Safronov | 482 | if len(s.dirTemplate) > 0 { |
fd0a4446 2023-04-03 Pavel Safronov | 483 | profileType = FolderType |
| 484 | } | |
37160b89 2026-05-01 Pavel Safronov | 485 | var patterns []ParameterizedText |
| 486 | if len(s.fileTemplate) > 0 { | |
| 487 | patterns = append(patterns, ParameterizedText(s.fileTemplate)) | |
| 488 | } else if len(s.dirTemplate) > 0 { | |
| 489 | patterns = append(patterns, ParameterizedText(s.dirTemplate)) | |
fd0a4446 2023-04-03 Pavel Safronov | 490 | } |
| 491 | profile = Profile{ | |
| 492 | Name: "temp", | |
| 493 | Description: fmt.Sprintf("temp-%s-profile", profileType), | |
| 494 | Type: profileType, | |
| 495 | NamePatterns: patterns, | |
| 496 | } | |
ae89a42a 2022-11-02 Pavel Safronov | 497 | } |
| 498 | } else { | |
37160b89 2026-05-01 Pavel Safronov | 499 | if s.profile == "" { |
| 500 | var err error | |
| 501 | profile, err = findDefaultProfile(cfg.Profiles) | |
8fc7c452 2023-03-29 Pavel Safronov | 502 | if err != nil { |
| 503 | return profile, fmt.Errorf("no profile specified and no default profile found") | |
| 504 | } | |
| 505 | } else { | |
37160b89 2026-05-01 Pavel Safronov | 506 | index, err := findProfileIndex(cfg.Profiles, s.profile) |
8fc7c452 2023-03-29 Pavel Safronov | 507 | if err != nil { |
| 508 | return profile, fmt.Errorf("error finding profile: %v", err) | |
| 509 | } | |
37160b89 2026-05-01 Pavel Safronov | 510 | profile = cfg.Profiles[index] |
ae89a42a 2022-11-02 Pavel Safronov | 511 | } |
b1ded4b4 2022-10-30 Pavel Safronov | 512 | } |
8c15a714 2022-11-04 Pavel Safronov | 513 | return profile, nil |
| 514 | } | |
| 515 | ||
37160b89 2026-05-01 Pavel Safronov | 516 | func runMain(s settings) error { |
| 517 | profile, err := findProfile(s) | |
8c15a714 2022-11-04 Pavel Safronov | 518 | if err != nil { |
| 519 | return fmt.Errorf("error finding profile: %v", err) | |
| 520 | } | |
b1ded4b4 2022-10-30 Pavel Safronov | 521 | |
bbd011d0 2022-12-10 Pavel Safronov | 522 | stat, err := os.Stdin.Stat() |
| 523 | if err != nil { | |
| 524 | return fmt.Errorf("error getting stdin stat: %w", err) | |
| 525 | } | |
37160b89 2026-05-01 Pavel Safronov | 526 | var pipeInputScanner *bufio.Scanner |
dc7227fa 2022-12-10 Pavel Safronov | 527 | incomingPipeData := (stat.Mode() & os.ModeCharDevice) == 0 |
| 528 | if incomingPipeData { | |
bbd011d0 2022-12-10 Pavel Safronov | 529 | pipeInputScanner = bufio.NewScanner(os.Stdin) |
dc7227fa 2022-12-10 Pavel Safronov | 530 | fmt.Printf("Reading from stdin\n") |
bbd011d0 2022-12-10 Pavel Safronov | 531 | } |
| 532 | ||
37160b89 2026-05-01 Pavel Safronov | 533 | errors := processProfile(profile, s.projectName, s.extras, *s.usr, pipeInputScanner) |
b1ded4b4 2022-10-30 Pavel Safronov | 534 | if len(errors) > 0 { |
| 535 | return fmt.Errorf("error processing profile: %v", errors) | |
| 536 | } | |
| 537 | ||
| 538 | return nil | |
f819f801 2022-09-21 Pavel Safronov | 539 | } |