Add orgid to wac envs create (#255)

Wac requests require orgid. Was removed from yaml

Author: Emyrk

internal/cmd/ceapi.go

@@ -183,14 +183,22 @@ func getImgs(ctx context.Context, client coder.Client, conf getImgsConf) ([]code
 }
 
 func isMultiOrgMember(ctx context.Context, client coder.Client, email string) (bool, error) {
+	orgs, err := getUserOrgs(ctx, client, email)
+	if err != nil {
+		return false, err
+	}
+	return len(orgs) > 1, nil
+}
+
+func getUserOrgs(ctx context.Context, client coder.Client, email string) ([]coder.Organization, error) {
 	u, err := client.UserByEmail(ctx, email)
 	if err != nil {
-		return false, xerrors.New("email not found")
+		return nil, xerrors.New("email not found")
 	}
 
 	orgs, err := client.Organizations(ctx)
 	if err != nil {
-		return false, err
+		return nil, xerrors.New("no organizations found")
 	}
-	return len(lookupUserOrgs(u, orgs)) > 1, nil
+	return lookupUserOrgs(u, orgs), nil
 }

internal/cmd/envs.go

@@ -288,15 +288,37 @@ coder envs create-from-repo -f coder.yaml`,
 				}
 			}
 
-			multiOrgMember, err := isMultiOrgMember(ctx, client, coder.Me)
+			orgs, err := getUserOrgs(ctx, client, coder.Me)
 			if err != nil {
 				return err
 			}
 
+			multiOrgMember := len(orgs) > 1
 			if multiOrgMember && org == "" {
 				return xerrors.New("org is required for multi-org members")
 			}
 
+			var userOrg *coder.Organization
+			for i := range orgs {
+				// Look for org by name
+				if orgs[i].Name == org {
+					userOrg = &orgs[i]
+					break
+				}
+				// Or use default if the provided is blank
+				if org == "" && orgs[i].Default {
+					userOrg = &orgs[i]
+					break
+				}
+			}
+
+			if userOrg == nil {
+				if org != "" {
+					return xerrors.Errorf("Unable to locate org '%s'", org)
+				}
+				return xerrors.Errorf("Unable to locate a default organization for the user")
+			}
+
 			var rd io.Reader
 			if filepath != "" {
 				b, err := ioutil.ReadFile(filepath)
@@ -318,6 +340,7 @@ coder envs create-from-repo -f coder.yaml`,
 			}
 
 			env, err := client.CreateEnvironment(ctx, coder.CreateEnvironmentRequest{
+				OrgID:    userOrg.ID,
 				Template: tpl,
 			})
 			if err != nil {