mirror of
https://github.com/charmbracelet/glow
synced 2024-11-10 06:04:18 +00:00
84 lines
2 KiB
Go
84 lines
2 KiB
Go
package main
|
||
|
||
import (
|
||
"fmt"
|
||
"os"
|
||
"path"
|
||
"path/filepath"
|
||
|
||
"github.com/charmbracelet/x/editor"
|
||
"github.com/spf13/cobra"
|
||
"github.com/spf13/viper"
|
||
)
|
||
|
||
const defaultConfig = `# style name or JSON path (default "auto")
|
||
style: "auto"
|
||
# mouse support (TUI-mode only)
|
||
mouse: false
|
||
# use pager to display markdown
|
||
pager: false
|
||
# word-wrap at width
|
||
width: 80
|
||
`
|
||
|
||
var configCmd = &cobra.Command{
|
||
Use: "config",
|
||
Hidden: false,
|
||
Short: "Edit the glow config file",
|
||
Long: paragraph(fmt.Sprintf("\n%s the glow config file. We’ll use EDITOR to determine which editor to use. If the config file doesn't exist, it will be created.", keyword("Edit"))),
|
||
Example: paragraph("glow config\nglow config --config path/to/config.yml"),
|
||
Args: cobra.NoArgs,
|
||
RunE: func(*cobra.Command, []string) error {
|
||
if err := ensureConfigFile(); err != nil {
|
||
return err
|
||
}
|
||
|
||
c, err := editor.Cmd("Glow", configFile)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
c.Stdin = os.Stdin
|
||
c.Stdout = os.Stdout
|
||
c.Stderr = os.Stderr
|
||
if err := c.Run(); err != nil {
|
||
return err
|
||
}
|
||
|
||
fmt.Println("Wrote config file to:", configFile)
|
||
return nil
|
||
},
|
||
}
|
||
|
||
func ensureConfigFile() error {
|
||
if configFile == "" {
|
||
configFile = viper.GetViper().ConfigFileUsed()
|
||
if err := os.MkdirAll(filepath.Dir(configFile), 0o755); err != nil {
|
||
return fmt.Errorf("could not write configuration file: %w", err)
|
||
}
|
||
}
|
||
|
||
if ext := path.Ext(configFile); ext != ".yaml" && ext != ".yml" {
|
||
return fmt.Errorf("'%s' is not a supported configuration type: use '%s' or '%s'", ext, ".yaml", ".yml")
|
||
}
|
||
|
||
if _, err := os.Stat(configFile); os.IsNotExist(err) {
|
||
// File doesn't exist yet, create all necessary directories and
|
||
// write the default config file
|
||
if err := os.MkdirAll(filepath.Dir(configFile), 0o700); err != nil {
|
||
return err
|
||
}
|
||
|
||
f, err := os.Create(configFile)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
defer func() { _ = f.Close() }()
|
||
|
||
if _, err := f.WriteString(defaultConfig); err != nil {
|
||
return err
|
||
}
|
||
} else if err != nil { // some other error occurred
|
||
return err
|
||
}
|
||
return nil
|
||
}
|