Migrate gRPC workspaceService other workspace related method (#19118)

* Migrate gRPC workspaceService other workspace related method

* Use shim and gRPC everywhere

* fix update

Author: mustard-mh

components/dashboard/src/components/SelectWorkspaceClassComponent.tsx

@@ -4,11 +4,11 @@
  * See License.AGPL.txt in the project root for license information.
  */
 
-import { SupportedWorkspaceClass } from "@gitpod/gitpod-protocol/lib/workspace-class";
 import { FC, useCallback, useEffect, useMemo } from "react";
-import WorkspaceClass from "../icons/WorkspaceClass.svg";
+import WorkspaceClassIcon from "../icons/WorkspaceClass.svg";
 import { Combobox, ComboboxElement, ComboboxSelectedItem } from "./podkit/combobox/Combobox";
 import { useWorkspaceClasses } from "../data/workspaces/workspace-classes-query";
+import { WorkspaceClass } from "@gitpod/public-api/lib/gitpod/v1/workspace_pb";
 
 interface SelectWorkspaceClassProps {
     selectedWorkspaceClass?: string;
@@ -78,7 +78,7 @@ export default function SelectWorkspaceClassComponent({
 }
 
 type WorkspaceClassDropDownElementSelectedProps = {
-    wsClass?: SupportedWorkspaceClass;
+    wsClass?: WorkspaceClass;
     loading?: boolean;
 };
 
@@ -90,7 +90,7 @@ const WorkspaceClassDropDownElementSelected: FC<WorkspaceClassDropDownElementSel
 
     return (
         <ComboboxSelectedItem
-            icon={WorkspaceClass}
+            icon={WorkspaceClassIcon}
             loading={loading}
             htmlTitle={title}
             title={<div className="truncate">{title}</div>}
@@ -106,7 +106,7 @@ const WorkspaceClassDropDownElementSelected: FC<WorkspaceClassDropDownElementSel
     );
 };
 
-function WorkspaceClassDropDownElement(props: { wsClass: SupportedWorkspaceClass }): JSX.Element {
+function WorkspaceClassDropDownElement(props: { wsClass: WorkspaceClass }): JSX.Element {
     const c = props.wsClass;
     return (
         <div className="flex ml-1 mt-1 flex-grow">

components/dashboard/src/data/setup.tsx

@@ -32,7 +32,7 @@ import * as SSHClasses from "@gitpod/public-api/lib/gitpod/v1/ssh_pb";
 // This is used to version the cache
 // If data we cache changes in a non-backwards compatible way, increment this version
 // That will bust any previous cache versions a client may have stored
-const CACHE_VERSION = "14";
+const CACHE_VERSION = "16";
 
 export function noPersistence(queryKey: QueryKey): QueryKey {
     return [...queryKey, "no-persistence"];

components/dashboard/src/data/workspaces/delete-inactive-workspaces-mutation.ts

@@ -5,7 +5,7 @@
  */
 
 import { useMutation, useQueryClient } from "@tanstack/react-query";
-import { workspacesService } from "../../service/public-api";
+import { workspaceClient } from "../../service/public-api";
 import { getListWorkspacesQueryKey, ListWorkspacesQueryResult } from "./list-workspaces-query";
 import { useCurrentOrg } from "../organizations/orgs-query";
 
@@ -22,7 +22,7 @@ export const useDeleteInactiveWorkspacesMutation = () => {
 
             for (const workspaceId of workspaceIds) {
                 try {
-                    await workspacesService.deleteWorkspace({ workspaceId });
+                    await workspaceClient.deleteWorkspace({ workspaceId });
 
                     deletedWorkspaceIds.push(workspaceId);
                 } catch (e) {

components/dashboard/src/data/workspaces/delete-workspace-mutation.ts

@@ -5,7 +5,7 @@
  */
 
 import { useMutation, useQueryClient } from "@tanstack/react-query";
-import { workspacesService } from "../../service/public-api";
+import { workspaceClient } from "../../service/public-api";
 import { getListWorkspacesQueryKey, ListWorkspacesQueryResult } from "./list-workspaces-query";
 import { useCurrentOrg } from "../organizations/orgs-query";
 
@@ -19,7 +19,7 @@ export const useDeleteWorkspaceMutation = () => {
 
     return useMutation({
         mutationFn: async ({ workspaceId }: DeleteWorkspaceArgs) => {
-            return await workspacesService.deleteWorkspace({ workspaceId });
+            return await workspaceClient.deleteWorkspace({ workspaceId });
         },
         onSuccess: (_, { workspaceId }) => {
             const queryKey = getListWorkspacesQueryKey(org.data?.id);

components/dashboard/src/data/workspaces/list-workspaces-query.ts

@@ -4,7 +4,7 @@
  * See License.AGPL.txt in the project root for license information.
  */
 
-import { useQuery } from "@tanstack/react-query";
+import { useQuery, useQueryClient } from "@tanstack/react-query";
 import { useCurrentOrg } from "../organizations/orgs-query";
 import { workspaceClient } from "../../service/public-api";
 import { Workspace } from "@gitpod/public-api/lib/gitpod/v1/workspace_pb";
@@ -57,3 +57,19 @@ export function getListWorkspacesQueryKey(orgId?: string) {
     }
     return ["workspaces", "list", orgId];
 }
+
+export const useUpdateWorkspaceInCache = () => {
+    const queryClient = useQueryClient();
+    const org = useCurrentOrg();
+    return (newWorkspace: Workspace) => {
+        const queryKey = getListWorkspacesQueryKey(org.data?.id);
+        queryClient.setQueryData<ListWorkspacesQueryResult>(queryKey, (oldWorkspacesData) => {
+            return oldWorkspacesData?.map((info) => {
+                if (info.id !== newWorkspace.id) {
+                    return info;
+                }
+                return newWorkspace;
+            });
+        });
+    };
+};

components/dashboard/src/data/workspaces/stop-workspace-mutation.ts

@@ -5,7 +5,7 @@
  */
 
 import { useMutation } from "@tanstack/react-query";
-import { workspacesService } from "../../service/public-api";
+import { workspaceClient } from "../../service/public-api";
 
 type StopWorkspaceArgs = {
     workspaceId: string;
@@ -15,7 +15,7 @@ export const useStopWorkspaceMutation = () => {
     // No need to manually update workspace in cache here, we'll receive messages over the ws that will update it
     return useMutation({
         mutationFn: async ({ workspaceId }: StopWorkspaceArgs) => {
-            return workspacesService.stopWorkspace({ workspaceId });
+            return workspaceClient.stopWorkspace({ workspaceId });
         },
     });
 };

components/dashboard/src/data/workspaces/toggle-workspace-pinned-mutation.ts

@@ -0,0 +1,26 @@
+/**
+ * Copyright (c) 2023 Gitpod GmbH. All rights reserved.
+ * Licensed under the GNU Affero General Public License (AGPL).
+ * See License.AGPL.txt in the project root for license information.
+ */
+
+import { useMutation } from "@tanstack/react-query";
+import { useUpdateWorkspaceInCache } from "./list-workspaces-query";
+import { UpdateWorkspaceRequest } from "@gitpod/public-api/lib/gitpod/v1/workspace_pb";
+import { PartialMessage } from "@bufbuild/protobuf";
+import { workspaceClient } from "../../service/public-api";
+
+export const useUpdateWorkspaceMutation = () => {
+    const updateWorkspace = useUpdateWorkspaceInCache();
+
+    return useMutation({
+        mutationFn: async (data: PartialMessage<UpdateWorkspaceRequest>) => {
+            return await workspaceClient.updateWorkspace(data);
+        },
+        onSuccess: (data) => {
+            if (data.workspace) {
+                updateWorkspace(data.workspace);
+            }
+        },
+    });
+};