mirror of
https://github.com/JanDeDobbeleer/oh-my-posh.git
synced 2025-02-02 05:41:10 -08:00
101 lines
2.3 KiB
Go
101 lines
2.3 KiB
Go
package cli
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/jandedobbeleer/oh-my-posh/src/ansi"
|
|
"github.com/jandedobbeleer/oh-my-posh/src/engine"
|
|
"github.com/jandedobbeleer/oh-my-posh/src/platform"
|
|
"github.com/jandedobbeleer/oh-my-posh/src/shell"
|
|
"github.com/jandedobbeleer/oh-my-posh/src/upgrade"
|
|
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var (
|
|
printOutput bool
|
|
strict bool
|
|
manual bool
|
|
|
|
initCmd = &cobra.Command{
|
|
Use: "init [bash|zsh|fish|powershell|pwsh|cmd|nu|tcsh|elvish|xonsh]",
|
|
Short: "Initialize your shell and config",
|
|
Long: `Initialize your shell and config.
|
|
|
|
See the documentation to initialize your shell: https://ohmyposh.dev/docs/installation/prompt.`,
|
|
ValidArgs: []string{
|
|
"bash",
|
|
"zsh",
|
|
"fish",
|
|
"powershell",
|
|
"pwsh",
|
|
"cmd",
|
|
"nu",
|
|
"tcsh",
|
|
"elvish",
|
|
"xonsh",
|
|
},
|
|
Args: NoArgsOrOneValidArg,
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
if len(args) == 0 {
|
|
_ = cmd.Help()
|
|
return
|
|
}
|
|
runInit(args[0])
|
|
},
|
|
}
|
|
)
|
|
|
|
func init() {
|
|
initCmd.Flags().BoolVarP(&printOutput, "print", "p", false, "print the init script")
|
|
initCmd.Flags().BoolVarP(&strict, "strict", "s", false, "run in strict mode")
|
|
initCmd.Flags().BoolVarP(&manual, "manual", "m", false, "enable/disable manual mode")
|
|
_ = initCmd.MarkPersistentFlagRequired("config")
|
|
RootCmd.AddCommand(initCmd)
|
|
}
|
|
|
|
func runInit(shellName string) {
|
|
env := &platform.Shell{
|
|
CmdFlags: &platform.Flags{
|
|
Shell: shellName,
|
|
Config: config,
|
|
Strict: strict,
|
|
Manual: manual,
|
|
},
|
|
}
|
|
env.Init()
|
|
defer env.Close()
|
|
|
|
cfg := engine.LoadConfig(env)
|
|
|
|
shell.Transient = cfg.TransientPrompt != nil
|
|
shell.ErrorLine = cfg.ErrorLine != nil || cfg.ValidLine != nil
|
|
shell.Tooltips = len(cfg.Tooltips) > 0
|
|
shell.ShellIntegration = cfg.ShellIntegration
|
|
shell.PromptMark = shellName == shell.FISH && cfg.ITermFeatures != nil && cfg.ITermFeatures.Contains(ansi.PromptMark)
|
|
|
|
for i, block := range cfg.Blocks {
|
|
// only fetch cursor position when relevant
|
|
if !cfg.DisableCursorPositioning && (i == 0 && block.Newline) {
|
|
shell.CursorPositioning = true
|
|
}
|
|
if block.Type == engine.RPrompt {
|
|
shell.RPrompt = true
|
|
}
|
|
}
|
|
|
|
// allow overriding the upgrade notice from the config
|
|
if cfg.DisableNotice {
|
|
env.Cache().Set(upgrade.CACHEKEY, "disabled", -1)
|
|
}
|
|
|
|
if printOutput {
|
|
init := shell.PrintInit(env)
|
|
fmt.Print(init)
|
|
return
|
|
}
|
|
|
|
init := shell.Init(env)
|
|
fmt.Print(init)
|
|
}
|