-
Notifications
You must be signed in to change notification settings - Fork 37
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat: use cobra, implement config and pod ls
- Loading branch information
1 parent
c85ea03
commit d6bf47f
Showing
9 changed files
with
358 additions
and
0 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
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,97 @@ | ||
| package api | ||
|
|
||
| import ( | ||
| "encoding/json" | ||
| "fmt" | ||
| "io/ioutil" | ||
| ) | ||
|
|
||
| type PodOut struct { | ||
| Data *PodData `json:"data"` | ||
| Errors []*GraphQLError `json:"errors"` | ||
| } | ||
| type GraphQLError struct { | ||
| Message string | ||
| } | ||
| type PodData struct { | ||
| Myself *MySelfData | ||
| } | ||
| type MySelfData struct { | ||
| Pods []*Pod | ||
| } | ||
| type Pod struct { | ||
| Id string | ||
| Name string | ||
| ImageName string | ||
| DesiredStatus string | ||
| Machine *Machine | ||
| } | ||
| type Machine struct { | ||
| GpuDisplayName string | ||
| } | ||
|
|
||
| func QueryPods() (pods []*Pod, err error) { | ||
| input := Input{ | ||
| Query: ` | ||
| query myPods { | ||
| myself { | ||
| pods { | ||
| id | ||
| machineId | ||
| name | ||
| dockerId | ||
| dockerArgs | ||
| imageName | ||
| port | ||
| ports | ||
| podType | ||
| gpuCount | ||
| vcpuCount | ||
| containerDiskInGb | ||
| memoryInGb | ||
| volumeInGb | ||
| volumeMountPath | ||
| desiredStatus | ||
| uptimeSeconds | ||
| costPerHr | ||
| env | ||
| lastStatusChange | ||
| machine { | ||
| gpuDisplayName | ||
| } | ||
| } | ||
| } | ||
| } | ||
| `, | ||
| } | ||
| res, err := Query(input) | ||
| if err != nil { | ||
| err = fmt.Errorf("QueryPods: %s", err.Error()) | ||
| return | ||
| } | ||
| if res.StatusCode != 200 { | ||
| err = fmt.Errorf("QueryPods: statuscode %d", res.StatusCode) | ||
| return | ||
| } | ||
| defer res.Body.Close() | ||
| rawData, err := ioutil.ReadAll(res.Body) | ||
| if err != nil { | ||
| err = fmt.Errorf("QueryPods: %s", err.Error()) | ||
| return | ||
| } | ||
| data := &PodOut{} | ||
| if err = json.Unmarshal(rawData, data); err != nil { | ||
| err = fmt.Errorf("QueryPods: %s", err.Error()) | ||
| return | ||
| } | ||
| if len(data.Errors) > 0 { | ||
| err = fmt.Errorf("pods query error: %s", data.Errors[0].Message) | ||
| return | ||
| } | ||
| if data == nil || data.Data == nil || data.Data.Myself == nil || data.Data.Myself.Pods == nil { | ||
| err = fmt.Errorf("QueryPods: data is nil: %s", string(rawData)) | ||
| return | ||
| } | ||
| pods = data.Data.Myself.Pods | ||
| return | ||
| } |
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,32 @@ | ||
| package api | ||
|
|
||
| import ( | ||
| "bytes" | ||
| "encoding/json" | ||
| "net/http" | ||
| "time" | ||
|
|
||
| "github.com/spf13/viper" | ||
| ) | ||
|
|
||
| var apiUrl = "https://api.dev.runpod.io/graphql" | ||
|
|
||
| type Input struct { | ||
| Query string `json:"query"` | ||
| Variables map[string]string `json:"variables"` | ||
| } | ||
|
|
||
| func Query(input Input) (res *http.Response, err error) { | ||
| jsonValue, err := json.Marshal(input) | ||
| if err != nil { | ||
| return | ||
| } | ||
| req, err := http.NewRequest("POST", apiUrl+"?api_key="+viper.GetString("apiKey"), bytes.NewBuffer(jsonValue)) | ||
| if err != nil { | ||
| return | ||
| } | ||
| req.Header.Add("Content-Type", "application/json") | ||
|
|
||
| client := &http.Client{Timeout: time.Second * 5} | ||
| return client.Do(req) | ||
| } |
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,37 @@ | ||
| package config | ||
|
|
||
| import ( | ||
| "errors" | ||
| "fmt" | ||
| "strings" | ||
|
|
||
| "github.com/spf13/cobra" | ||
| "github.com/spf13/viper" | ||
| ) | ||
|
|
||
| var ConfigFile string | ||
| var ApiKey string | ||
|
|
||
| // ConfigCmd represents the config command | ||
| var ConfigCmd = &cobra.Command{ | ||
| Use: "config", | ||
| Short: "CLI Config", | ||
| Long: "RunPod CLI Config Settings", | ||
| Run: func(c *cobra.Command, args []string) { | ||
| ApiKey = strings.TrimSpace(ApiKey) | ||
| if len(ApiKey) == 0 { | ||
| cobra.CheckErr(errors.New("apiKey cannot be empty")) | ||
| } | ||
| err := viper.WriteConfig() | ||
| cobra.CheckErr(err) | ||
|
|
||
| fmt.Println("saved apiKey into config file: " + ConfigFile) | ||
| }, | ||
| } | ||
|
|
||
| func init() { | ||
| ConfigCmd.Flags().StringVar(&ApiKey, "apiKey", "", "runpod api key") | ||
| ConfigCmd.MarkFlagRequired("apiKey") | ||
| viper.BindPFlag("apiKey", ConfigCmd.Flags().Lookup("apiKey")) | ||
| viper.SetDefault("apiKey", "") | ||
| } |
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,23 @@ | ||
| package pod | ||
|
|
||
| import ( | ||
| "cli/api" | ||
| "fmt" | ||
|
|
||
| "github.com/spf13/cobra" | ||
| ) | ||
|
|
||
| // lsCmd represents the ls command | ||
| var lsCmd = &cobra.Command{ | ||
| Use: "ls", | ||
| Short: "list of pods", | ||
| Long: `List of pods in RUNNING status.`, | ||
| Run: func(cmd *cobra.Command, args []string) { | ||
| pods, err := api.QueryPods() | ||
| cobra.CheckErr(err) | ||
|
|
||
| for _, p := range pods { | ||
| fmt.Println(p.Id, p.Name, p.ImageName, p.DesiredStatus) | ||
| } | ||
| }, | ||
| } |
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,21 @@ | ||
| package pod | ||
|
|
||
| import ( | ||
| "github.com/spf13/cobra" | ||
| ) | ||
|
|
||
| // PodCmd represents the pod command | ||
| var PodCmd = &cobra.Command{ | ||
| Use: "pod", | ||
| Short: "A brief description of your command", | ||
| Long: `A longer description that spans multiple lines and likely contains examples | ||
| and usage of using your command. For example: | ||
| Cobra is a CLI library for Go that empowers applications. | ||
| This application is a tool to generate the needed files | ||
| to quickly create a Cobra application.`, | ||
| } | ||
|
|
||
| func init() { | ||
| PodCmd.AddCommand(lsCmd) | ||
| } |
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,54 @@ | ||
| package cmd | ||
|
|
||
| import ( | ||
| "os" | ||
|
|
||
| "cli/cmd/config" | ||
| "cli/cmd/pod" | ||
|
|
||
| "github.com/spf13/cobra" | ||
| "github.com/spf13/viper" | ||
| ) | ||
|
|
||
| // rootCmd represents the base command when called without any subcommands | ||
| var rootCmd = &cobra.Command{ | ||
| Use: "runpod-cli", | ||
| Short: "RunPod CLI", | ||
| Long: "RUNPOD CLI", | ||
| } | ||
|
|
||
| // Execute adds all child commands to the root command and sets flags appropriately. | ||
| // This is called by main.main(). It only needs to happen once to the rootCmd. | ||
| func Execute() { | ||
| err := rootCmd.Execute() | ||
| if err != nil { | ||
| os.Exit(1) | ||
| } | ||
| } | ||
|
|
||
| func init() { | ||
| cobra.OnInitialize(initConfig) | ||
| rootCmd.AddCommand(pod.PodCmd) | ||
| rootCmd.AddCommand(config.ConfigCmd) | ||
| } | ||
|
|
||
| // initConfig reads in config file and ENV variables if set. | ||
| func initConfig() { | ||
| home, err := os.UserHomeDir() | ||
| cobra.CheckErr(err) | ||
|
|
||
| viper.AddConfigPath(home) | ||
| viper.SetConfigType("yaml") | ||
| viper.SetConfigName(".runpod") | ||
| config.ConfigFile = home + "/.runpod.yaml" | ||
|
|
||
| viper.AutomaticEnv() // read in environment variables that match | ||
|
|
||
| // If a config file is found, read it in. | ||
| if err := viper.ReadInConfig(); err == nil { | ||
| // fmt.Fprintln(os.Stderr, "Using config file:", viper.ConfigFileUsed()) | ||
| } else { | ||
| err := viper.WriteConfigAs(config.ConfigFile) | ||
| cobra.CheckErr(err) | ||
| } | ||
| } |
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,26 @@ | ||
| module cli | ||
|
|
||
| go 1.17 | ||
|
|
||
| require ( | ||
| github.com/spf13/cobra v1.4.0 | ||
| github.com/spf13/viper v1.10.1 | ||
| ) | ||
|
|
||
| require ( | ||
| github.com/fsnotify/fsnotify v1.5.1 // indirect | ||
| github.com/hashicorp/hcl v1.0.0 // indirect | ||
| github.com/inconshreveable/mousetrap v1.0.0 // indirect | ||
| github.com/magiconair/properties v1.8.5 // indirect | ||
| github.com/mitchellh/mapstructure v1.4.3 // indirect | ||
| github.com/pelletier/go-toml v1.9.4 // indirect | ||
| github.com/spf13/afero v1.6.0 // indirect | ||
| github.com/spf13/cast v1.4.1 // indirect | ||
| github.com/spf13/jwalterweatherman v1.1.0 // indirect | ||
| github.com/spf13/pflag v1.0.5 // indirect | ||
| github.com/subosito/gotenv v1.2.0 // indirect | ||
| golang.org/x/sys v0.0.0-20211210111614-af8b64212486 // indirect | ||
| golang.org/x/text v0.3.7 // indirect | ||
| gopkg.in/ini.v1 v1.66.2 // indirect | ||
| gopkg.in/yaml.v2 v2.4.0 // indirect | ||
| ) |
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,61 @@ | ||
| github.com/cpuguy83/go-md2man/v2 v2.0.1/go.mod h1:tgQtvFlXSQOSOSIRvRPT7W67SCa46tRHOmNcaadrF8o= | ||
| github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= | ||
| github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= | ||
| github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= | ||
| github.com/fsnotify/fsnotify v1.5.1 h1:mZcQUHVQUQWoPXXtuf9yuEXKudkV2sx1E06UadKWpgI= | ||
| github.com/fsnotify/fsnotify v1.5.1/go.mod h1:T3375wBYaZdLLcVNkcVbzGHY7f1l/uK5T5Ai1i3InKU= | ||
| github.com/hashicorp/hcl v1.0.0 h1:0Anlzjpi4vEasTeNFn2mLJgTSwt0+6sfsiTG8qcWGx4= | ||
| github.com/hashicorp/hcl v1.0.0/go.mod h1:E5yfLk+7swimpb2L/Alb/PJmXilQ/rhwaUYs4T20WEQ= | ||
| github.com/inconshreveable/mousetrap v1.0.0 h1:Z8tu5sraLXCXIcARxBp/8cbvlwVa7Z1NHg9XEKhtSvM= | ||
| github.com/inconshreveable/mousetrap v1.0.0/go.mod h1:PxqpIevigyE2G7u3NXJIT2ANytuPF1OarO4DADm73n8= | ||
| github.com/kr/fs v0.1.0/go.mod h1:FFnZGqtBN9Gxj7eW1uZ42v5BccTP0vu6NEaFoC2HwRg= | ||
| github.com/magiconair/properties v1.8.5 h1:b6kJs+EmPFMYGkow9GiUyCyOvIwYetYJ3fSaWak/Gls= | ||
| github.com/magiconair/properties v1.8.5/go.mod h1:y3VJvCyxH9uVvJTWEGAELF3aiYNyPKd5NZ3oSwXrF60= | ||
| github.com/mitchellh/mapstructure v1.4.3 h1:OVowDSCllw/YjdLkam3/sm7wEtOy59d8ndGgCcyj8cs= | ||
| github.com/mitchellh/mapstructure v1.4.3/go.mod h1:bFUtVrKA4DC2yAKiSyO/QUcy7e+RRV2QTWOzhPopBRo= | ||
| github.com/pelletier/go-toml v1.9.4 h1:tjENF6MfZAg8e4ZmZTeWaWiT2vXtsoO6+iuOjFhECwM= | ||
| github.com/pelletier/go-toml v1.9.4/go.mod h1:u1nR/EPcESfeI/szUZKdtJ0xRNbUoANCkoOuaOx1Y+c= | ||
| github.com/pkg/errors v0.8.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0= | ||
| github.com/pkg/sftp v1.10.1/go.mod h1:lYOWFsE0bwd1+KfKJaKeuokY15vzFx25BLbzYYoAxZI= | ||
| github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= | ||
| github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= | ||
| github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM= | ||
| github.com/spf13/afero v1.6.0 h1:xoax2sJ2DT8S8xA2paPFjDCScCNeWsg75VG0DLRreiY= | ||
| github.com/spf13/afero v1.6.0/go.mod h1:Ai8FlHk4v/PARR026UzYexafAt9roJ7LcLMAmO6Z93I= | ||
| github.com/spf13/cast v1.4.1 h1:s0hze+J0196ZfEMTs80N7UlFt0BDuQ7Q+JDnHiMWKdA= | ||
| github.com/spf13/cast v1.4.1/go.mod h1:Qx5cxh0v+4UWYiBimWS+eyWzqEqokIECu5etghLkUJE= | ||
| github.com/spf13/cobra v1.4.0 h1:y+wJpx64xcgO1V+RcnwW0LEHxTKRi2ZDPSBjWnrg88Q= | ||
| github.com/spf13/cobra v1.4.0/go.mod h1:Wo4iy3BUC+X2Fybo0PDqwJIv3dNRiZLHQymsfxlB84g= | ||
| github.com/spf13/jwalterweatherman v1.1.0 h1:ue6voC5bR5F8YxI5S67j9i582FU4Qvo2bmqnqMYADFk= | ||
| github.com/spf13/jwalterweatherman v1.1.0/go.mod h1:aNWZUN0dPAAO/Ljvb5BEdw96iTZ0EXowPYD95IqWIGo= | ||
| github.com/spf13/pflag v1.0.5 h1:iy+VFUOCP1a+8yFto/drg2CJ5u0yRoB7fZw3DKv/JXA= | ||
| github.com/spf13/pflag v1.0.5/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg= | ||
| github.com/spf13/viper v1.10.1 h1:nuJZuYpG7gTj/XqiUwg8bA0cp1+M2mC3J4g5luUYBKk= | ||
| github.com/spf13/viper v1.10.1/go.mod h1:IGlFPqhNAPKRxohIzWpI5QEy4kuI7tcl5WvR+8qy1rU= | ||
| github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= | ||
| github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs= | ||
| github.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4= | ||
| github.com/stretchr/testify v1.7.0 h1:nwc3DEeHmmLAfoZucVR881uASk0Mfjw8xYJ99tb5CcY= | ||
| github.com/subosito/gotenv v1.2.0 h1:Slr1R9HxAlEKefgq5jn9U+DnETlIUa6HfgEzj0g5d7s= | ||
| github.com/subosito/gotenv v1.2.0/go.mod h1:N0PQaV/YGNqwC0u51sEeR/aUtSLEXKX9iv69rRypqCw= | ||
| golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w= | ||
| golang.org/x/crypto v0.0.0-20190820162420-60c769a6c586/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI= | ||
| golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg= | ||
| golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= | ||
| golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= | ||
| golang.org/x/sys v0.0.0-20210630005230-0f9fa26af87c/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= | ||
| golang.org/x/sys v0.0.0-20211210111614-af8b64212486 h1:5hpz5aRr+W1erYCL5JRhSUBJRph7l9XkNveoExlrKYk= | ||
| golang.org/x/sys v0.0.0-20211210111614-af8b64212486/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= | ||
| golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= | ||
| golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= | ||
| golang.org/x/text v0.3.7 h1:olpwvP2KacW1ZWvsR7uQhoyTYvKAupfQrRGBFM352Gk= | ||
| golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ= | ||
| golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= | ||
| gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM= | ||
| gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= | ||
| gopkg.in/ini.v1 v1.66.2 h1:XfR1dOYubytKy4Shzc2LHrrGhU0lDCfDGG1yLPmpgsI= | ||
| gopkg.in/ini.v1 v1.66.2/go.mod h1:pNLf8WUiyNEtQjuu5G5vTm06TEv9tsIgeAvK8hOrP4k= | ||
| gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= | ||
| gopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY= | ||
| gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ= | ||
| gopkg.in/yaml.v3 v3.0.0-20210107192922-496545a6307b h1:h8qDotaEPuJATrMmW04NCwg7v22aHH28wwpauUhK9Oo= |
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,7 @@ | ||
| package main | ||
|
|
||
| import "cli/cmd" | ||
|
|
||
| func main() { | ||
| cmd.Execute() | ||
| } |