apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-05-02 04:12:47 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

feat: Enhance connector management UI with improved loading states, add document count display for connectors, and implement indexing progress indicators for better user feedback.

Browse source
This commit is contained in:
Anish Sarkar 2025-12-31 17:21:26 +05:30
parent 3ac806dcdf
commit 163df8fda7
10 changed files with 287 additions and 95 deletions
Download patch file Download diff file Expand all files Collapse all files

64
surfsense_web/components/assistant-ui/connector-popup/utils/connector-document-mapping.ts Normal file
View file

@ -0,0 +1,64 @@
"use client";
/**
* Maps SearchSourceConnectorType to DocumentType for fetching document counts
*
* Note: Some connectors don't have a direct 1:1 mapping to document types:
* - Search API connectors (TAVILY_API, SEARXNG_API, etc.) don't index documents
* - WEBCRAWLER_CONNECTOR maps to CRAWLED_URL document type
* - GOOGLE_DRIVE_CONNECTOR maps to GOOGLE_DRIVE_FILE document type
*/
export const CONNECTOR_TO_DOCUMENT_TYPE: Record<string, string> = {
// Direct mappings (connector type matches document type)
SLACK_CONNECTOR: "SLACK_CONNECTOR",
NOTION_CONNECTOR: "NOTION_CONNECTOR",
GITHUB_CONNECTOR: "GITHUB_CONNECTOR",
LINEAR_CONNECTOR: "LINEAR_CONNECTOR",
DISCORD_CONNECTOR: "DISCORD_CONNECTOR",
JIRA_CONNECTOR: "JIRA_CONNECTOR",
CONFLUENCE_CONNECTOR: "CONFLUENCE_CONNECTOR",
CLICKUP_CONNECTOR: "CLICKUP_CONNECTOR",
GOOGLE_CALENDAR_CONNECTOR: "GOOGLE_CALENDAR_CONNECTOR",
GOOGLE_GMAIL_CONNECTOR: "GOOGLE_GMAIL_CONNECTOR",
AIRTABLE_CONNECTOR: "AIRTABLE_CONNECTOR",
LUMA_CONNECTOR: "LUMA_CONNECTOR",
ELASTICSEARCH_CONNECTOR: "ELASTICSEARCH_CONNECTOR",
BOOKSTACK_CONNECTOR: "BOOKSTACK_CONNECTOR",
// Special mappings (connector type differs from document type)
GOOGLE_DRIVE_CONNECTOR: "GOOGLE_DRIVE_FILE",
WEBCRAWLER_CONNECTOR: "CRAWLED_URL",
};
/**
* Get the document type for a given connector type
* Returns undefined if the connector doesn't index documents (e.g., search APIs)
*/
export function getDocumentTypeForConnector(
connectorType: string
): string | undefined {
return CONNECTOR_TO_DOCUMENT_TYPE[connectorType];
}
/**
* Get document count for a specific connector type from document type counts
*/
export function getDocumentCountForConnector(
connectorType: string,
documentTypeCounts: Record<string, number> | undefined
): number | undefined {
if (!documentTypeCounts) return undefined;
const documentType = getDocumentTypeForConnector(connectorType);
if (!documentType) return undefined;
return documentTypeCounts[documentType];
}
/**
* Check if a connector type is indexable (produces documents)
*/
export function isIndexableConnectorType(connectorType: string): boolean {
return connectorType in CONNECTOR_TO_DOCUMENT_TYPE;
}