-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
* feat: Impl update command * Fix unit test by use mock server * Fix test * Mock network call
- Loading branch information
1 parent
dc0d6d7
commit bbbadf2
Showing
9 changed files
with
485 additions
and
8 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
| Original file line number | Diff line number | Diff line change |
|---|---|---|
|
|
@@ -4,12 +4,18 @@ Copyright © 2024 Xiao Cui <[email protected]> | |
| package main | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "os" | ||
|
|
||
| "github.com/XiaoConstantine/mycli/pkg/build" | ||
| "github.com/XiaoConstantine/mycli/pkg/commands/root" | ||
| ) | ||
|
|
||
| func main() { | ||
| if len(os.Args) > 1 && os.Args[1] == "version" { | ||
| fmt.Printf("mycli version %s (%s) - built on %s\n", build.Version, build.Commit, build.Date) | ||
| os.Exit(0) | ||
| } | ||
| code := root.Run([]string{}) | ||
| os.Exit(int(code)) | ||
| } | ||
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
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,12 @@ | ||
| package build | ||
|
|
||
| var ( | ||
| // Version is the current version of the CLI. | ||
| Version = "dev" | ||
|
|
||
| // Commit is the git commit hash of the build. | ||
| Commit = "none" | ||
|
|
||
| // Date is the build date. | ||
| Date = "unknown" | ||
| ) |
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
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,199 @@ | ||
| package update | ||
|
|
||
| import ( | ||
| "encoding/json" | ||
| "fmt" | ||
| "io" | ||
| "net/http" | ||
| "os" | ||
| "path/filepath" | ||
| "runtime" | ||
| "strings" | ||
|
|
||
| "github.com/XiaoConstantine/mycli/pkg/build" | ||
| "github.com/XiaoConstantine/mycli/pkg/iostreams" | ||
| "github.com/XiaoConstantine/mycli/pkg/utils" | ||
| "github.com/spf13/cobra" | ||
| ) | ||
|
|
||
| var CheckForUpdatesFunc = CheckForUpdates | ||
|
|
||
| func NewUpdateCmd(iostream *iostreams.IOStreams) *cobra.Command { | ||
| cmd := &cobra.Command{ | ||
| Use: "update", | ||
| Short: "Update mycli to the latest version", | ||
| RunE: func(cmd *cobra.Command, args []string) error { | ||
| installDir, err := ensureInstallDirectory() | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| if err := ensurePathInZshrc(installDir); err != nil { | ||
| return err | ||
| } | ||
|
|
||
| return updateCLI(iostream) | ||
| }, | ||
| } | ||
| return cmd | ||
| } | ||
|
|
||
| func updateCLI(iostream *iostreams.IOStreams) error { | ||
| currentVersion := build.Version | ||
|
|
||
| // Ensure .mycli/bin directory exists | ||
| installDir, err := ensureInstallDirectory() | ||
| if err != nil { | ||
| return fmt.Errorf("failed to ensure install directory: %w", err) | ||
| } | ||
|
|
||
| // Get the latest release info | ||
| release, err := getLatestRelease() | ||
| if err != nil { | ||
| return fmt.Errorf("failed to get latest release: %w", err) | ||
| } | ||
|
|
||
| // Check if update is needed | ||
| if utils.CompareVersions(currentVersion, release.TagName) >= 0 { | ||
| fmt.Fprintln(iostream.Out, "You're already using the latest version of mycli.") | ||
| return nil | ||
| } | ||
|
|
||
| // Determine the asset to download based on the current OS and architecture | ||
| assetURL := "" | ||
| for _, asset := range release.Assets { | ||
| if asset.Name == fmt.Sprintf("mycli_%s_%s", runtime.GOOS, runtime.GOARCH) { | ||
| assetURL = asset.BrowserDownloadURL | ||
| break | ||
| } | ||
| } | ||
|
|
||
| if assetURL == "" { | ||
| return fmt.Errorf("no suitable release found for %s/%s", runtime.GOOS, runtime.GOARCH) | ||
| } | ||
|
|
||
| // Download the new binary | ||
| resp, err := http.Get(assetURL) | ||
| if err != nil { | ||
| return fmt.Errorf("failed to download update: %w", err) | ||
| } | ||
| defer resp.Body.Close() | ||
|
|
||
| // Determine the install location | ||
| installPath := filepath.Join(installDir, "mycli") | ||
|
|
||
| // Create a temporary file | ||
| tmpFile, err := os.CreateTemp("", "mycli-update") | ||
| if err != nil { | ||
| return fmt.Errorf("failed to create temporary file: %w", err) | ||
| } | ||
| defer os.Remove(tmpFile.Name()) | ||
|
|
||
| // Copy the downloaded content to the temporary file | ||
| _, err = io.Copy(tmpFile, resp.Body) | ||
| if err != nil { | ||
| return fmt.Errorf("failed to write update to temporary file: %w", err) | ||
| } | ||
| tmpFile.Close() | ||
|
|
||
| // Make the temporary file executable | ||
| if err := os.Chmod(tmpFile.Name(), 0755); err != nil { | ||
| return fmt.Errorf("failed to make new binary executable: %w", err) | ||
| } | ||
|
|
||
| // Replace the old binary with the new one | ||
| if err := os.Rename(tmpFile.Name(), installPath); err != nil { | ||
| return fmt.Errorf("failed to replace old binary: %w", err) | ||
| } | ||
|
|
||
| fmt.Fprintf(iostream.Out, "mycli has been updated successfully to version %s!\n", release.TagName) | ||
| return nil | ||
| } | ||
|
|
||
| var ensureInstallDirectory = func() (string, error) { | ||
| home, err := os.UserHomeDir() | ||
| if err != nil { | ||
| return "", fmt.Errorf("failed to get user home directory: %w", err) | ||
| } | ||
|
|
||
| installDir := filepath.Join(home, ".mycli", "bin") | ||
| if err := os.MkdirAll(installDir, 0755); err != nil { | ||
| return "", fmt.Errorf("failed to create install directory: %w", err) | ||
| } | ||
|
|
||
| return installDir, nil | ||
| } | ||
|
|
||
| func ensurePathInZshrc(installDir string) error { | ||
| home, err := os.UserHomeDir() | ||
| if err != nil { | ||
| return fmt.Errorf("failed to get user home directory: %w", err) | ||
| } | ||
|
|
||
| zshrcPath := filepath.Join(home, ".zshrc") | ||
| zshrcContent, err := os.ReadFile(zshrcPath) | ||
| if err != nil && !os.IsNotExist(err) { | ||
| return fmt.Errorf("failed to read .zshrc: %w", err) | ||
| } | ||
|
|
||
| pathLine := fmt.Sprintf("export PATH=\"$PATH:%s\"", installDir) | ||
| if !strings.Contains(string(zshrcContent), pathLine) { | ||
| f, err := os.OpenFile(zshrcPath, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644) | ||
| if err != nil { | ||
| return fmt.Errorf("failed to open .zshrc: %w", err) | ||
| } | ||
| defer f.Close() | ||
|
|
||
| if _, err := f.WriteString("\n" + pathLine + "\n"); err != nil { | ||
| return fmt.Errorf("failed to write to .zshrc: %w", err) | ||
| } | ||
|
|
||
| fmt.Println("Added .mycli/bin to your PATH in .zshrc. Please restart your terminal or run 'source ~/.zshrc' to apply the changes.") | ||
| } | ||
|
|
||
| return nil | ||
| } | ||
|
|
||
| const repoURL = "https://api.github.com/repos/XiaoConstantine/mycli/releases/latest" | ||
|
|
||
| type githubRelease struct { | ||
| TagName string `json:"tag_name"` | ||
| Assets []githubAsset `json:"assets"` | ||
| } | ||
|
|
||
| type githubAsset struct { | ||
| Name string `json:"name"` | ||
| BrowserDownloadURL string `json:"browser_download_url"` | ||
| } | ||
|
|
||
| var getLatestRelease = func() (*githubRelease, error) { | ||
| resp, err := http.Get(repoURL) | ||
| if err != nil { | ||
| return nil, fmt.Errorf("failed to fetch latest release: %w", err) | ||
| } | ||
| defer resp.Body.Close() | ||
|
|
||
| if resp.StatusCode != http.StatusOK { | ||
| return nil, fmt.Errorf("unexpected status code: %d", resp.StatusCode) | ||
| } | ||
|
|
||
| var release githubRelease | ||
| if err := json.NewDecoder(resp.Body).Decode(&release); err != nil { | ||
| return nil, fmt.Errorf("failed to decode release JSON: %w", err) | ||
| } | ||
|
|
||
| return &release, nil | ||
| } | ||
|
|
||
| func CheckForUpdates(iostream *iostreams.IOStreams) (bool, string, error) { | ||
| currentVersion := build.Version | ||
|
|
||
| // Get the latest release info | ||
| release, err := getLatestRelease() | ||
| if err != nil { | ||
| return false, "", fmt.Errorf("failed to get latest release: %w", err) | ||
| } | ||
|
|
||
| hasUpdate := utils.CompareVersions(currentVersion, release.TagName) < 0 | ||
| return hasUpdate, release.TagName, nil | ||
| } |
Oops, something went wrong.