Wrote function to copy embedded config files to a config directory
This commit is contained in:
		
							
								
								
									
										63
									
								
								config.go
									
									
									
									
									
								
							
							
						
						
									
										63
									
								
								config.go
									
									
									
									
									
								
							@@ -2,13 +2,76 @@ package main
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
import (
 | 
					import (
 | 
				
			||||||
	"ccat/stack"
 | 
						"ccat/stack"
 | 
				
			||||||
 | 
						"embed"
 | 
				
			||||||
 | 
						"errors"
 | 
				
			||||||
 | 
						"io/fs"
 | 
				
			||||||
	"os"
 | 
						"os"
 | 
				
			||||||
 | 
						"os/user"
 | 
				
			||||||
 | 
						"path/filepath"
 | 
				
			||||||
	"regexp"
 | 
						"regexp"
 | 
				
			||||||
 | 
						"runtime"
 | 
				
			||||||
	"strings"
 | 
						"strings"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	"gopkg.in/yaml.v2"
 | 
						"gopkg.in/yaml.v2"
 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					//go:embed config
 | 
				
			||||||
 | 
					var storedConfigs embed.FS // Embed the folder containing config files
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func runningOnWindows() bool {
 | 
				
			||||||
 | 
						return runtime.GOOS == "windows"
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// generateDefaultConfigs is used to generate a folder of default config files
 | 
				
			||||||
 | 
					// for common languages. These default config files are embedded into the program, and will
 | 
				
			||||||
 | 
					// be outputted into a directory.
 | 
				
			||||||
 | 
					//
 | 
				
			||||||
 | 
					// If there is an error encountered, the error is returned.
 | 
				
			||||||
 | 
					func generateDefaultConfigs() error {
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						var configOutputPath string // Location of config files, depends on OS
 | 
				
			||||||
 | 
						if runningOnWindows() {
 | 
				
			||||||
 | 
							configOutputPath = "%APPDATA%\\ccat"
 | 
				
			||||||
 | 
						} else {
 | 
				
			||||||
 | 
							currentUser, err := user.Current()
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								panic(err)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							configOutputPath = filepath.Join("/home/" + currentUser.Username + "/.config/ccat/")
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						err := os.MkdirAll(configOutputPath, 0755)
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							if os.IsExist(err) {
 | 
				
			||||||
 | 
								return errors.New("Directory already exists.")
 | 
				
			||||||
 | 
							} else {
 | 
				
			||||||
 | 
								return errors.New("Unable to create directory.")
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// Copy each folder from the embedded filesystem, into the destination path
 | 
				
			||||||
 | 
						err = fs.WalkDir(storedConfigs, "config", func(path string, d fs.DirEntry, err error) error {
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								return err
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							if d.IsDir() { // Skip directories
 | 
				
			||||||
 | 
								return nil
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							relPath, _ := filepath.Rel("config", path)
 | 
				
			||||||
 | 
							dstPath := filepath.Join(configOutputPath, relPath) // Destination path
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							data, err := os.ReadFile(path)
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								return err
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							if err := os.WriteFile(dstPath, data, 0644); err != nil {
 | 
				
			||||||
 | 
								return err
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							return nil
 | 
				
			||||||
 | 
						})
 | 
				
			||||||
 | 
						return nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// loadConfig takes in the filename of a config file. It reads the file,
 | 
					// loadConfig takes in the filename of a config file. It reads the file,
 | 
				
			||||||
// and returns a stack of RegColors, with the item at the bottom being the one that
 | 
					// and returns a stack of RegColors, with the item at the bottom being the one that
 | 
				
			||||||
// was read first. This ensures that, _when accessing the RegColors in the stack, the last
 | 
					// was read first. This ensures that, _when accessing the RegColors in the stack, the last
 | 
				
			||||||
 
 | 
				
			|||||||
		Reference in New Issue
	
	Block a user