-
Notifications
You must be signed in to change notification settings - Fork 1.4k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat: run autocomplete on init (#1291)
Signed-off-by: Ivan Dagelic <[email protected]>
- Loading branch information
Showing
12 changed files
with
152 additions
and
114 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file was deleted.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,139 @@ | ||
| // Copyright 2024 Daytona Platforms Inc. | ||
| // SPDX-License-Identifier: Apache-2.0 | ||
|
|
||
| package autocomplete | ||
|
|
||
| import ( | ||
| "errors" | ||
| "fmt" | ||
| "os" | ||
| "path/filepath" | ||
| "strings" | ||
|
|
||
| "github.com/spf13/cobra" | ||
| ) | ||
|
|
||
| var supportedShells = []string{"bash", "zsh", "fish", "powershell"} | ||
|
|
||
| var AutoCompleteCmd = &cobra.Command{ | ||
| Use: fmt.Sprintf("autocomplete [%s]", strings.Join(supportedShells, "|")), | ||
| Short: "Adds a completion script for your shell environment", | ||
| Args: cobra.ExactArgs(1), | ||
| RunE: func(cmd *cobra.Command, args []string) error { | ||
| shell := args[0] | ||
|
|
||
| profilePath, err := SetupAutocompletionForShell(cmd.Root(), shell) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| fmt.Println("Autocomplete script generated and injected successfully.") | ||
| fmt.Printf("Please source your %s profile to apply the changes or restart your terminal.\n", shell) | ||
| fmt.Printf("For manual sourcing, use: source %s\n", profilePath) | ||
| if shell == "bash" { | ||
| fmt.Println("Please make sure that you have bash-completion installed in order to get full autocompletion functionality.") | ||
| fmt.Println("On how to install bash-completion, please refer to the following link: https://www.daytona.io/docs/tools/cli/#daytona-autocomplete") | ||
| } | ||
|
|
||
| return nil | ||
| }, | ||
| } | ||
|
|
||
| func DetectShellAndSetupAutocompletion(rootCmd *cobra.Command) error { | ||
| shell := os.Getenv("SHELL") | ||
| if shell == "" { | ||
| return fmt.Errorf("unable to detect the shell, please use a supported one: %s", strings.Join(supportedShells, ", ")) | ||
| } | ||
|
|
||
| for _, supportedShell := range supportedShells { | ||
| if strings.Contains(shell, supportedShell) { | ||
| shell = supportedShell | ||
| break | ||
| } | ||
| } | ||
|
|
||
| _, err := SetupAutocompletionForShell(rootCmd, shell) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| return nil | ||
| } | ||
|
|
||
| func SetupAutocompletionForShell(rootCmd *cobra.Command, shell string) (string, error) { | ||
| homeDir, err := os.UserHomeDir() | ||
| if err != nil { | ||
| return "", fmt.Errorf("error finding user home directory: %s", err) | ||
| } | ||
|
|
||
| var filePath, profilePath string | ||
| switch shell { | ||
| case "bash": | ||
| filePath = filepath.Join(homeDir, ".daytona.completion_script.bash") | ||
| profilePath = filepath.Join(homeDir, ".bashrc") | ||
| case "zsh": | ||
| filePath = filepath.Join(homeDir, ".daytona.completion_script.zsh") | ||
| profilePath = filepath.Join(homeDir, ".zshrc") | ||
| case "fish": | ||
| filePath = filepath.Join(homeDir, ".config", "fish", "daytona.completion_script.fish") | ||
| profilePath = filepath.Join(homeDir, ".config", "fish", "config.fish") | ||
| case "powershell": | ||
| filePath = filepath.Join(homeDir, "daytona.completion_script.ps1") | ||
| profilePath = filepath.Join(homeDir, "Documents", "WindowsPowerShell", "Microsoft.PowerShell_profile.ps1") | ||
| default: | ||
| return "", errors.New("unsupported shell type. Please use bash, zsh, fish, or powershell") | ||
| } | ||
|
|
||
| file, err := os.Create(filePath) | ||
| if err != nil { | ||
| return "", fmt.Errorf("error creating completion script file: %s", err) | ||
| } | ||
| defer file.Close() | ||
|
|
||
| switch shell { | ||
| case "bash": | ||
| err = rootCmd.GenBashCompletion(file) | ||
| case "zsh": | ||
| err = rootCmd.GenZshCompletion(file) | ||
| case "fish": | ||
| err = rootCmd.GenFishCompletion(file, true) | ||
| case "powershell": | ||
| err = rootCmd.GenPowerShellCompletionWithDesc(file) | ||
| } | ||
|
|
||
| if err != nil { | ||
| return "", fmt.Errorf("error generating completion script: %s", err) | ||
| } | ||
|
|
||
| sourceCommand := fmt.Sprintf("\nsource %s\n", filePath) | ||
| if shell == "powershell" { | ||
| sourceCommand = fmt.Sprintf(". %s\n", filePath) | ||
| } | ||
|
|
||
| alreadyPresent := false | ||
| // Read existing content from the file | ||
| profile, err := os.ReadFile(profilePath) | ||
|
|
||
| if err != nil && !os.IsNotExist(err) { | ||
| return "", fmt.Errorf("error while reading profile (%s): %s", profilePath, err) | ||
| } | ||
|
|
||
| if strings.Contains(string(profile), strings.TrimSpace(sourceCommand)) { | ||
| alreadyPresent = true | ||
| } | ||
|
|
||
| if !alreadyPresent { | ||
| // Append the source command to the shell's profile file if not present | ||
| profile, err := os.OpenFile(profilePath, os.O_APPEND|os.O_WRONLY|os.O_CREATE, 0644) | ||
| if err != nil { | ||
| return "", fmt.Errorf("error opening profile file (%s): %s", profilePath, err) | ||
| } | ||
| defer profile.Close() | ||
|
|
||
| if _, err := profile.WriteString(sourceCommand); err != nil { | ||
| return "", fmt.Errorf("error writing to profile file (%s): %s", profilePath, err) | ||
| } | ||
| } | ||
|
|
||
| return profilePath, nil | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters