fix refresh tokens (#1253)

* fix refresh tokens * fix context key case * add more context keys
astronomer · Jun 13, 2023 · a52f13f · a52f13f
1 parent 63e93da
commit a52f13f
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 120 deletions.
diff --git a/cmd/cloud/migration.go b/cmd/cloud/migration.go
diff --git a/cmd/cloud/migration_test.go b/cmd/cloud/migration_test.go
diff --git a/cmd/cloud/setup.go b/cmd/cloud/setup.go
@@ -125,10 +125,7 @@ func Setup(cmd *cobra.Command, client astro.Client, coreClient astrocore.CoreCli
 	if err != nil {
 		return err
 	}
-	err = migrateCloudConfig(coreClient)
-	if err != nil {
-		return err
-	}
+
 	return nil
 }
 
@@ -177,6 +174,18 @@ func checkToken(client astro.Client, coreClient astrocore.CoreClient, out io.Wri
 		if err != nil {
 			return err
 		}
+		err = c.SetContextKey("organization", c.Organization)
+		if err != nil {
+			return err
+		}
+		err = c.SetContextKey("organization_short_name", c.OrganizationShortName)
+		if err != nil {
+			return err
+		}
+		err = c.SetContextKey("organization_product", c.OrganizationProduct)
+		if err != nil {
+			return err
+		}
 	}
 	return nil
 }