Fix behavior on missing data in CAS

- FindMissingBlob used to identify when a blob is missing from cas
- Updated the following screens:
   - Browser (Display an error message if the data is missing)
       Command
       Action
       Historical Execute Response
   - Compare Action (Allows new comparison in case of missing data)
   - CAS Log Viewer (Displays that the data is missing)
   - Critical Path (Hide critical path)
   - Profile (Disable the button but display the reason through a ToolTip)

Author: Arsal34

frontend/src/components/BrowserActionPages/index.tsx

@@ -11,6 +11,7 @@ import { initialSizeClassCacheClient } from "@/grpc/initialSizeClassCacheClient"
 import type { BrowserSearchParams } from "@/routes/browser.$";
 import type { BrowserPageParams } from "@/types/BrowserPageParams";
 import { BrowserPageType } from "@/types/BrowserPageType";
+import { useCheckDataExists } from "@/utils/fetchCasObject";
 import BrowserActionGrid from "../BrowserActionGrid";
 import { fetchBrowserActionGrid } from "../BrowserActionGrid/fetch";
 import { CompareActionButtons } from "../CompareAction/CompareTargetButton";
@@ -26,6 +27,13 @@ interface Params {
 export const BrowserActionPages: React.FC<Params> = ({ params, search }) => {
   const [prefersCompareSideBySide, setPrefersCompareSideBySide] =
     useState(true);
+
+  const { exists, isLoading } = useCheckDataExists(
+    params.instanceName,
+    [params.digest],
+    params.digestFunction,
+  );
+
   const actionQuery = useQuery({
     queryKey: ["browserActionGrid", params.digest.hash],
     queryFn: () => {
@@ -38,10 +46,12 @@ export const BrowserActionPages: React.FC<Params> = ({ params, search }) => {
       );
     },
     staleTime: 5 * 60 * 1000,
+    enabled: exists === true,
   });
+
   const compareActionQuery = useQuery({
     queryKey: ["browserActionGrid", search.comparedAction?.digest.hash],
-    enabled: !!search.comparedAction,
+    enabled: !!search.comparedAction && exists === true,
     queryFn: () => {
       if (!search.comparedAction) {
         return undefined;
@@ -57,6 +67,17 @@ export const BrowserActionPages: React.FC<Params> = ({ params, search }) => {
     staleTime: 5 * 60 * 1000,
   });
 
+  if (exists === false && !isLoading) {
+    return (
+      <PortalAlert
+        showIcon
+        type="error"
+        title="Error fetching the operation"
+        description={"The CAS contains no data for this action"}
+      />
+    );
+  }
+
   if (
     actionQuery.isPending ||
     (compareActionQuery.isPending && search.comparedAction)
@@ -69,16 +90,22 @@ export const BrowserActionPages: React.FC<Params> = ({ params, search }) => {
     compareActionQuery.isError ||
     (!compareActionQuery.data && search.comparedAction)
   ) {
+    let errorMessage: string;
+    if (
+      actionQuery.error?.message.endsWith("Object not found") ||
+      compareActionQuery.error?.message.endsWith("Object not found")
+    ) {
+      errorMessage = "The target action was not found";
+    } else {
+      errorMessage =
+        "Unknown error occurred while fetching data from the server.";
+    }
     return (
       <PortalAlert
         showIcon
         type="error"
-        title="Error fetching action"
-        description={
-          actionQuery.error?.message ||
-          compareActionQuery.error?.message ||
-          "Unknown error occurred while fetching data from the server."
-        }
+        title="Error fetching the operation"
+        description={errorMessage}
       />
     );
   }

frontend/src/components/BrowserCommandGrid/index.tsx

@@ -4,7 +4,10 @@ import type React from "react";
 import { casByteStreamClient } from "@/grpc/casByteStreamClient";
 import { Command } from "@/lib/grpc-client/build/bazel/remote/execution/v2/remote_execution";
 import type { BrowserPageParams } from "@/types/BrowserPageParams";
-import { fetchCasObjectAndParse } from "@/utils/fetchCasObject";
+import {
+  fetchCasObjectAndParse,
+  useCheckDataExists,
+} from "@/utils/fetchCasObject";
 import { BrowserCommandDescription } from "../BrowserCommandDescription";
 import CopyBbClientdCommandButton from "../BrowserCommandDescription/CopyBbClientdCommandButton";
 import DownloadAsShellScriptButton from "../BrowserCommandDescription/DownloadAsShellScriptButton";
@@ -17,6 +20,12 @@ interface Params {
 }
 
 const BrowserCommandGrid: React.FC<Params> = ({ browserPageParams }) => {
+  const { exists, isLoading } = useCheckDataExists(
+    browserPageParams.instanceName,
+    [browserPageParams.digest],
+    browserPageParams.digestFunction,
+  );
+
   const { data, isError, isPending, error } = useQuery({
     queryKey: ["browserCommandGrid", browserPageParams],
     queryFn: () =>
@@ -27,8 +36,20 @@ const BrowserCommandGrid: React.FC<Params> = ({ browserPageParams }) => {
         browserPageParams.digest,
         Command,
       ),
+    enabled: exists === true,
   });
 
+  if (!exists && !isLoading) {
+    return (
+      <PortalAlert
+        showIcon
+        type="error"
+        title="Command not found"
+        description="The CAS contains no data for this command."
+      />
+    );
+  }
+
   if (isPending) {
     return <Spin />;
   }

frontend/src/components/BrowserResultDescription/index.tsx

@@ -1,10 +1,12 @@
-import { Descriptions, Space, Tag } from "antd";
+import { Descriptions, Space, Spin, Tag } from "antd";
 import type React from "react";
 import type { ExecuteResponse } from "@/lib/grpc-client/build/bazel/remote/execution/v2/remote_execution";
 import type { POSIXResourceUsage } from "@/lib/grpc-client/buildbarn/resourceusage/resourceusage";
 import type { BrowserPageParams } from "@/types/BrowserPageParams";
 import { digestFunctionValueToString } from "@/utils/digestFunctionUtils";
+import { useCheckDataExists } from "@/utils/fetchCasObject";
 import { CasViewer } from "../LogViewer/casViewer";
+import PortalAlert from "../PortalAlert";
 
 interface Params {
   browserPageParams: BrowserPageParams;
@@ -17,6 +19,38 @@ const BrowserResultDescription: React.FC<Params> = ({
   executeResponse,
   posixResourceUsage,
 }) => {
+  const { exists: stdOutExists, isLoading: isLoadingOut } = useCheckDataExists(
+    browserPageParams.instanceName,
+    [
+      {
+        hash: executeResponse?.result?.stdoutDigest?.hash ?? "",
+        sizeBytes: (
+          executeResponse?.result?.stdoutDigest?.sizeBytes ?? ""
+        ).toString(),
+      },
+    ],
+    browserPageParams.digestFunction,
+    executeResponse.result?.stdoutDigest !== undefined,
+  );
+
+  const { exists: stdErrExists, isLoading: isLoadingErr } = useCheckDataExists(
+    browserPageParams.instanceName,
+    [
+      {
+        hash: executeResponse?.result?.stderrDigest?.hash ?? "",
+        sizeBytes: (
+          executeResponse?.result?.stderrDigest?.sizeBytes ?? ""
+        ).toString(),
+      },
+    ],
+    browserPageParams.digestFunction,
+    executeResponse.result?.stdoutDigest !== undefined,
+  );
+
+  if (isLoadingOut && isLoadingErr) {
+    return <Spin />;
+  }
+
   const renderResult = () => {
     if (executeResponse.status !== undefined) {
       return (
@@ -59,7 +93,10 @@ const BrowserResultDescription: React.FC<Params> = ({
       >
         {renderResult()}
       </Descriptions>
-      {executeResponse.result?.stdoutDigest?.hash &&
+
+      {!executeResponse.result?.stdoutDigest?.hash ? null : stdOutExists ===
+        true ? (
+        executeResponse.result?.stdoutDigest?.hash &&
         executeResponse.result?.stdoutDigest?.sizeBytes && (
           <CasViewer
             instanceName={browserPageParams.instanceName}
@@ -74,8 +111,19 @@ const BrowserResultDescription: React.FC<Params> = ({
             title="Standard Output"
             fileName="standard_output.txt"
           />
-        )}
-      {executeResponse.result?.stderrDigest?.hash &&
+        )
+      ) : (
+        <PortalAlert
+          type="error"
+          title="Standard output logs not found"
+          description="The data does not exist"
+          showIcon
+        />
+      )}
+
+      {!executeResponse.result?.stderrDigest?.hash ? null : stdErrExists ===
+        true ? (
+        executeResponse.result?.stderrDigest?.hash &&
         executeResponse.result?.stderrDigest?.sizeBytes && (
           <CasViewer
             instanceName={browserPageParams.instanceName}
@@ -90,7 +138,15 @@ const BrowserResultDescription: React.FC<Params> = ({
             title="Standard Error"
             fileName="standard_error.txt"
           />
-        )}
+        )
+      ) : (
+        <PortalAlert
+          type="error"
+          title="Standard error logs not found"
+          description={`The data does not exist`}
+          showIcon
+        />
+      )}
     </Space>
   );
 };

frontend/src/components/CompareAction/CompareTargetButton/index.tsx

@@ -1,11 +1,13 @@
 import { Button, Space } from "antd";
 import { useEffect, useState } from "react";
 import { LinkButton } from "@/components/LinkButton";
+import { digestFunction_ValueFromJSON } from "@/lib/grpc-client/build/bazel/remote/execution/v2/remote_execution";
 import {
   type BrowserPageParams,
   BrowserPageSchema,
 } from "@/types/BrowserPageParams";
 import { BrowserPageType } from "@/types/BrowserPageType";
+import { useCheckDataExists } from "@/utils/fetchCasObject";
 import { generateBrowserSplat } from "@/utils/urlGenerator";
 
 const COMPARE_KEY = "buildbarn_compare_action";
@@ -17,6 +19,19 @@ type Props = {
 
 const CompareActionButtons: React.FC<Props> = ({ params, comparing }) => {
   const [storedData, setStoredData] = useState<BrowserPageParams | undefined>();
+
+  const { exists } = useCheckDataExists(
+    storedData?.instanceName ?? "",
+    [
+      storedData?.digest ?? {
+        hash: "",
+        sizeBytes: "",
+      },
+    ],
+    storedData?.digestFunction ?? digestFunction_ValueFromJSON(""),
+    storedData !== undefined,
+  );
+
   useEffect(() => {
     const checkSavedData = () => {
       try {
@@ -63,7 +78,8 @@ const CompareActionButtons: React.FC<Props> = ({ params, comparing }) => {
       </LinkButton>
     );
   }
-  if (!storedData) {
+
+  if (!storedData || !exists) {
     return (
       <Button type="default" onClick={setCompare}>
         Compare...

frontend/src/components/InvocationOverviewDisplay/index.tsx

@@ -2,8 +2,10 @@ import { Descriptions } from "antd";
 import type React from "react";
 import { getFragmentData } from "@/graphql/__generated__";
 import type { BazelInvocationOverviewFragment } from "@/graphql/__generated__/graphql";
+import { digestFunction_ValueFromJSON } from "@/lib/grpc-client/build/bazel/remote/execution/v2/remote_execution";
 import { FILE_DETAILS_FRAGMENT } from "@/types/GraphqlFileFragment";
 import { commandLineDataToString } from "@/utils/commandLineDataToString";
+import { useCheckDataExists } from "@/utils/fetchCasObject";
 import { CriticalPathDisplay } from "../CriticalPath";
 import { InvocationResultTag } from "../InvocationResultTag";
 import PortalDuration from "../PortalDuration";
@@ -52,6 +54,20 @@ export const InvocationOverviewDisplay: React.FC<Props> = ({ invocation }) => {
 
   const parsedProfile = getFragmentData(FILE_DETAILS_FRAGMENT, profile);
 
+  const { exists } = useCheckDataExists(
+    parsedProfile?.digest.rev2InstanceName ?? "",
+    [
+      {
+        hash: parsedProfile?.digest.hash ?? "",
+        sizeBytes: (parsedProfile?.digest.sizeBytes ?? "").toString(),
+      },
+    ],
+    digestFunction_ValueFromJSON(
+      (parsedProfile?.digest.digestFunction ?? "").toUpperCase(),
+    ),
+    parsedProfile !== undefined,
+  );
+
   return (
     <Descriptions column={1} bordered>
       <Descriptions.Item label="Status">
@@ -84,7 +100,7 @@ export const InvocationOverviewDisplay: React.FC<Props> = ({ invocation }) => {
           formatConfig={{ smallestUnit: "s" }}
         />
       </Descriptions.Item>
-      {parsedProfile && (
+      {parsedProfile && exists && (
         <Descriptions.Item label="Critical Path">
           <CriticalPathDisplay
             profile={parsedProfile}