-
-
Notifications
You must be signed in to change notification settings - Fork 6
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #19 from ysicing/feat-207
Feat 207
- Loading branch information
Showing
21 changed files
with
598 additions
and
55 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
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,174 @@ | ||
| /* | ||
| * Copyright (c) 2021 ysicing <i@ysicing.me> | ||
| */ | ||
|
|
||
| package cmd | ||
|
|
||
| import ( | ||
| "context" | ||
| "github.com/ergoapi/util/file" | ||
| "github.com/manifoldco/promptui" | ||
| "github.com/spf13/cobra" | ||
| "github.com/ysicing/ergo/cmd/flags" | ||
| "github.com/ysicing/ergo/common" | ||
| "github.com/ysicing/ergo/pkg/ergo/cloud" | ||
| "github.com/ysicing/ergo/pkg/ergo/cloud/qcloud" | ||
| "github.com/ysicing/ergo/pkg/util/factory" | ||
| "github.com/ysicing/ergo/pkg/util/log" | ||
| "github.com/ysicing/ergo/pkg/util/ssh" | ||
| "os" | ||
| "strings" | ||
| ) | ||
|
|
||
| type CvmOption struct { | ||
| *flags.GlobalFlags | ||
| log log.Logger | ||
| action string | ||
| } | ||
|
|
||
| // newCvmCmd ergo cvm | ||
| func newCvmCmd(f factory.Factory) *cobra.Command { | ||
| opt := &CvmOption{ | ||
| GlobalFlags: globalFlags, | ||
| log: f.GetLog(), | ||
| } | ||
| cvm := &cobra.Command{ | ||
| Use: "cvm [flags]", | ||
| Short: "开通竞价机器", | ||
| Version: "2.0.7", | ||
| RunE: func(cobraCmd *cobra.Command, args []string) error { | ||
| return opt.Init() | ||
| }, | ||
| } | ||
| cvm.PersistentFlags().StringVarP(&opt.action, "action", "a", "", "操作 \"create\",\"new\",\"add\",\"destroy\",\"del\",\"rm\",\"list\" ") | ||
|
|
||
| return cvm | ||
| } | ||
|
|
||
| func (c *CvmOption) Init() error { | ||
| var aid, akey, provider, region string | ||
| cvmfile := common.GetDefaultCfgPathByName("cloud") | ||
| c.log.Debugf("load cloud cfg: %v", cvmfile) | ||
| if !file.CheckFileExists(cvmfile) { | ||
| c.log.Debugf("not found %v, will gen one", cvmfile) | ||
| pprompt := promptui.Prompt{ | ||
| Label: "provider", | ||
| } | ||
| provider, _ = pprompt.Run() | ||
| aidprompt := promptui.Prompt{ | ||
| Label: "aid", | ||
| } | ||
| aid, _ = aidprompt.Run() | ||
| akeyprompt := promptui.Prompt{ | ||
| Label: "akey", | ||
| } | ||
| akey, _ = akeyprompt.Run() | ||
| regionprompt := promptui.Prompt{ | ||
| Label: "region", | ||
| } | ||
| region, _ = regionprompt.Run() | ||
| c.log.Debugf("%v, %v, %v, %v", aid, akey, provider, region) | ||
| configs := cloud.Configs{} | ||
| configs.Add(cloud.Config{ | ||
| Provider: provider, | ||
| Secrets: cloud.Secrets{ | ||
| AID: aid, | ||
| AKey: akey, | ||
| }, | ||
| Regions: []string{region}, | ||
| }) | ||
| if err := configs.Save(cvmfile); err != nil { | ||
| return err | ||
| } | ||
| return nil | ||
| } else { | ||
| configs, err := cloud.LoadCloudConfigs(cvmfile) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| selectitem := append(configs.Configs, cloud.Config{ | ||
| Provider: "new", | ||
| }) | ||
| c.log.Debugf("加载配置成功: %v", selectitem) | ||
| ps := promptui.Select{ | ||
| Label: "选择凭证", | ||
| Items: selectitem, | ||
| Size: 4, | ||
| Templates: &promptui.SelectTemplates{ | ||
| Label: "{{ . }}", | ||
| Active: "\U0001F449 {{ .Provider | cyan }}", | ||
| Inactive: " {{ .Provider | cyan }}", | ||
| Selected: "\U0001F389 {{ .Provider | red | cyan }}", | ||
| }, | ||
| } | ||
| psid, _, _ := ps.Run() | ||
| if selectitem[psid].Provider == "new" { | ||
| c.log.Debugf("found %v, will add one", cvmfile) | ||
| pprompt := promptui.Prompt{ | ||
| Label: "provider", | ||
| } | ||
| provider, _ = pprompt.Run() | ||
| aidprompt := promptui.Prompt{ | ||
| Label: "aid", | ||
| } | ||
| aid, _ = aidprompt.Run() | ||
| akeyprompt := promptui.Prompt{ | ||
| Label: "akey", | ||
| } | ||
| akey, _ = akeyprompt.Run() | ||
| regionprompt := promptui.Prompt{ | ||
| Label: "region", | ||
| } | ||
| region, _ = regionprompt.Run() | ||
| c.log.Debugf("%v, %v, %v, %v", aid, akey, provider, region) | ||
| configs.Add(cloud.Config{ | ||
| Provider: provider, | ||
| Secrets: cloud.Secrets{ | ||
| AID: aid, | ||
| AKey: akey, | ||
| }, | ||
| Regions: []string{region}, | ||
| }) | ||
| if err := configs.Save(cvmfile); err != nil { | ||
| return err | ||
| } | ||
| args := os.Args | ||
| return ssh.RunCmd(args[0], "cvm") | ||
| } | ||
| c.log.Debugf("select %v", selectitem[psid]) | ||
| if len(selectitem[psid].Regions) != 0 { | ||
| region = selectitem[psid].Regions[0] | ||
| } | ||
| provider = selectitem[psid].Provider | ||
| aid = selectitem[psid].Secrets.AID | ||
| akey = selectitem[psid].Secrets.AKey | ||
| } | ||
| // c.log.Debugf("%v %v , %v %v", c.action, provider, cloud.ProviderQcloud.Value(), provider == cloud.ProviderQcloud.Value()) | ||
| if provider == cloud.ProviderQcloud.Value() { | ||
| // 腾讯云默认南京地域 | ||
| if !strings.HasPrefix(region, "ap") { | ||
| region = "ap-nanjing" | ||
| } | ||
| p, err := qcloud.NewCvm(qcloud.WithLog(c.log), qcloud.WithApi(aid, akey), qcloud.WithRegion(region)) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| switch c.action { | ||
| case "status": | ||
| return p.Status(context.TODO(), cloud.StatusOption{}) | ||
| case "create", "new", "add": | ||
| return p.Create(context.TODO(), cloud.CreateOption{}) | ||
| case "destroy", "del", "rm": | ||
| return p.Destroy(context.TODO(), cloud.DestroyOption{}) | ||
| case "halt": | ||
| return p.Halt(context.TODO(), cloud.HaltOption{}) | ||
| case "up": | ||
| return p.Up(context.TODO(), cloud.UpOption{}) | ||
| case "snapshot": | ||
| return p.Snapshot(context.TODO(), cloud.SnapshotOption{}) | ||
| default: | ||
| return p.List(context.TODO(), cloud.ListOption{}) | ||
| } | ||
| } | ||
| return 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
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
Oops, something went wrong.