[front] Add sync badge on websites (#7546)

dust-tt · Sep 19, 2024 · 8610669 · 8610669
1 parent 8fd092b
commit 8610669
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 4 deletions.
diff --git a/front/components/vaults/VaultResourcesList.tsx b/front/components/vaults/VaultResourcesList.tsx
@@ -85,9 +85,11 @@ type VaultResourcesListProps = {
 
 const getTableColumns = ({
   isManaged,
+  isWebsite,
   isSystemVault,
 }: {
   isManaged: boolean;
+  isWebsite: boolean;
   isSystemVault: boolean;
 }) => {
   const nameColumn: ColumnDef<RowData, string> = {
@@ -233,7 +235,7 @@ const getTableColumns = ({
       actionColumn,
     ];
   }
-  return isManaged
+  return isManaged || isWebsite
     ? [nameColumn, usedByColumn, managedByColumn, lastSyncedColumn]
     : [nameColumn, usedByColumn, managedByColumn];
 };
@@ -268,6 +270,7 @@ export const VaultResourcesList = ({
 
   const isSystemVault = systemVault.sId === vault.sId;
   const isManagedCategory = category === "managed";
+  const isWebsite = category === "website";
   const isWebsiteOrFolder = isWebsiteOrFolderCategory(category);
 
   const [isLoadingByProvider, setIsLoadingByProvider] = useState<
@@ -481,6 +484,7 @@ export const VaultResourcesList = ({
           data={rows}
           columns={getTableColumns({
             isManaged: isManagedCategory,
+            isWebsite: isWebsite,
             isSystemVault,
           })}
           filter={dataSourceSearch}

diff --git a/front/lib/data_sources.ts b/front/lib/data_sources.ts
@@ -53,7 +53,7 @@ export function isFolder(
 
 export function isWebsite(
   ds: DataSource
-): ds is DataSource & { connectorProvider: "webcrawler" } {
+): ds is DataSource & WithConnector & { connectorProvider: "webcrawler" } {
   return ds.connectorProvider === "webcrawler";
 }
 

diff --git a/front/pages/api/w/[wId]/vaults/[vId]/data_source_views/index.ts b/front/pages/api/w/[wId]/vaults/[vId]/data_source_views/index.ts
@@ -17,7 +17,7 @@ import {
 import { augmentDataSourceWithConnectorDetails } from "@app/lib/api/data_sources";
 import { withSessionAuthenticationForWorkspace } from "@app/lib/api/wrappers";
 import type { Authenticator } from "@app/lib/auth";
-import { isManaged } from "@app/lib/data_sources";
+import { isManaged, isWebsite } from "@app/lib/data_sources";
 import { DataSourceResource } from "@app/lib/resources/data_source_resource";
 import { DataSourceViewResource } from "@app/lib/resources/data_source_view_resource";
 import { VaultResource } from "@app/lib/resources/vault_resource";
@@ -112,7 +112,7 @@ async function handler(
             dataSourceViews.map(async (dataSourceView) => {
               const dataSource = dataSourceView.dataSource;
 
-              if (!isManaged(dataSource)) {
+              if (!isManaged(dataSource) && !isWebsite(dataSource)) {
                 return {
                   ...dataSourceView,
                   dataSource: {