44"errors"
55"fmt"
66"os"
7+ "slices"
8+ "strings"
79
810"github.com/coder/coder/v2/cli/clibase"
911"github.com/coder/coder/v2/cli/cliui"
@@ -22,6 +24,7 @@ func (r *RootCmd) organizations() *clibase.Cmd {
2224},
2325Children : []* clibase.Cmd {
2426r .currentOrganization (),
27+ r .switchOrganization (),
2528},
2629}
2730
@@ -52,13 +55,44 @@ func (r *RootCmd) switchOrganization() *clibase.Cmd {
5255),
5356Options : clibase.OptionSet {},
5457Handler :func (inv * clibase.Invocation )error {
58+ orgArg := inv .Args [0 ]
5559conf := r .createConfig ()
60+ orgs ,err := client .OrganizationsByUser (inv .Context (),codersdk .Me )
61+ if err != nil {
62+ return fmt .Errorf ("failed to get organizations: %w" ,err )
63+ }
64+
5665// If the user passes an empty string, we want to remove the organization
57- if inv . Args [ 0 ] == "" {
66+ if orgArg == "" {
5867err := conf .Organization ().Delete ()
5968if err != nil && ! errors .Is (err ,os .ErrNotExist ) {
6069return fmt .Errorf ("failed to unset organization: %w" ,err )
6170}
71+ }else {
72+ index := slices .IndexFunc (orgs ,func (org codersdk.Organization )bool {
73+ return org .Name == orgArg || org .ID .String ()== orgArg
74+ })
75+ if index < 0 {
76+ names := make ([]string ,0 ,len (orgs ))
77+ for _ ,org := range orgs {
78+ names = append (names ,org .Name )
79+ }
80+
81+ // Using this error for better error message formatting
82+ err := & codersdk.Error {
83+ Response : codersdk.Response {
84+ Message :fmt .Sprintf ("Organization %q not found." ,orgArg ),
85+ Detail :"Ensure the organization argument is correct and you are a member of it." ,
86+ },
87+ Helper :fmt .Sprintf ("Valid organizations you can switch to: %q" ,strings .Join (names ,", " )),
88+ }
89+ return err
90+ }
91+
92+ err := conf .Organization ().Write (orgs [index ].ID .String ())
93+ if err != nil {
94+ return fmt .Errorf ("failed to write organization to config file: %w" ,err )
95+ }
6296}
6397
6498return nil