apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-07-22 23:31:12 +02:00
Code Issues Projects Releases Packages Wiki Activity

Merge pull request #9 from okapteinis/claude/auth-token-key-refactor-01WNkTGGrYURxe9JATA3gX6y

Browse source 
Refactor auth token usage to use AUTH_TOKEN_KEY constant
This commit is contained in:
Ojārs Kapteinis 2025-11-19 00:27:43 +02:00 • committed by GitHub
commit 9248dd8b35
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
33 changed files with 96 additions and 56 deletions
Download patch file Download diff file Expand all files Collapse all files

4
surfsense_backend/.env.example
View file

@ -28,6 +28,10 @@ NEXT_FRONTEND_URL=http://localhost:3000
AUTH_TYPE=LOCAL
REGISTRATION_ENABLED=TRUE or FALSE
# CORS Configuration (comma-separated list of allowed origins)
# Use * to allow all origins, or specify domains like: https://example.com,https://app.example.com
CORS_ORIGINS=*
# Google OAuth Credentials (OPTIONAL - Required only for Gmail and Google Calendar connectors)
GOOGLE_OAUTH_CLIENT_ID=your_google_client_id
GOOGLE_OAUTH_CLIENT_SECRET=your_google_client_secret

2
surfsense_backend/app/app.py
View file

@ -50,7 +50,7 @@ app.add_middleware(ProxyHeadersMiddleware, trusted_hosts="*")
# Add CORS middleware
app.add_middleware(
CORSMiddleware,
allow_origins=["https://ai.kapteinis.lv"], # Only allow our domain
allow_origins=config.CORS_ORIGINS, # Configurable via CORS_ORIGINS env var
allow_credentials=True,
allow_methods=["*"], # Allows all methods
allow_headers=["*"], # Allows all headers

5
surfsense_backend/app/config/__init__.py
View file

@ -135,6 +135,11 @@ class Config:
AUTH_TYPE = os.getenv("AUTH_TYPE")
REGISTRATION_ENABLED = os.getenv("REGISTRATION_ENABLED", "TRUE").upper() == "TRUE"
# CORS Configuration
# Comma-separated list of allowed origins, defaults to all origins if not set
_cors_origins_str = os.getenv("CORS_ORIGINS", "*")
CORS_ORIGINS = [origin.strip() for origin in _cors_origins_str.split(",") if origin.strip()]
# Google OAuth
GOOGLE_OAUTH_CLIENT_ID = os.getenv("GOOGLE_OAUTH_CLIENT_ID")
GOOGLE_OAUTH_CLIENT_SECRET = os.getenv("GOOGLE_OAUTH_CLIENT_SECRET")

3
surfsense_web/app/auth/callback/page.tsx
View file

@ -1,5 +1,6 @@
import { Suspense } from "react";
import TokenHandler from "@/components/TokenHandler";
import { AUTH_TOKEN_KEY } from "@/lib/constants";
export default function AuthCallbackPage() {
return (
@ -15,7 +16,7 @@ export default function AuthCallbackPage() {
<TokenHandler
redirectPath="/dashboard"
tokenParamName="token"
storageKey="surfsense_bearer_token"
storageKey={AUTH_TOKEN_KEY}
/>
</Suspense>
</div>

3
surfsense_web/app/dashboard/[search_space_id]/connectors/add/airtable-connector/page.tsx
View file

@ -22,6 +22,7 @@ import {
type SearchSourceConnector,
useSearchSourceConnectors,
} from "@/hooks/use-search-source-connectors";
import { AUTH_TOKEN_KEY } from "@/lib/constants";
export default function AirtableConnectorPage() {
const router = useRouter();
@ -51,7 +52,7 @@ export default function AirtableConnectorPage() {
{
method: "GET",
headers: {
Authorization: `Bearer ${localStorage.getItem("surfsense_bearer_token")}`,
Authorization: `Bearer ${localStorage.getItem(AUTH_TOKEN_KEY)}`,
},
}
);

3
surfsense_web/app/dashboard/[search_space_id]/connectors/add/github-connector/page.tsx
View file

@ -38,6 +38,7 @@ import { Input } from "@/components/ui/input";
import { Tabs, TabsContent, TabsList, TabsTrigger } from "@/components/ui/tabs";
import { EnumConnectorName } from "@/contracts/enums/connector";
import { getConnectorIcon } from "@/contracts/enums/connectorIcons";
import { AUTH_TOKEN_KEY } from "@/lib/constants";
// Assuming useSearchSourceConnectors hook exists and works similarly
import { useSearchSourceConnectors } from "@/hooks/use-search-source-connectors";
@ -101,7 +102,7 @@ export default function GithubConnectorPage() {
setConnectorName(values.name); // Store the name
setValidatedPat(values.github_pat); // Store the PAT temporarily
try {
const token = localStorage.getItem("surfsense_bearer_token");
const token = localStorage.getItem(AUTH_TOKEN_KEY);
if (!token) {
throw new Error("No authentication token found");
}

3
surfsense_web/app/dashboard/[search_space_id]/connectors/add/google-calendar-connector/page.tsx
View file

@ -24,6 +24,7 @@ import {
type SearchSourceConnector,
useSearchSourceConnectors,
} from "@/hooks/use-search-source-connectors";
import { AUTH_TOKEN_KEY } from "@/lib/constants";
export default function GoogleCalendarConnectorPage() {
const router = useRouter();
@ -56,7 +57,7 @@ export default function GoogleCalendarConnectorPage() {
{
method: "GET",
headers: {
Authorization: `Bearer ${localStorage.getItem("surfsense_bearer_token")}`,
Authorization: `Bearer ${localStorage.getItem(AUTH_TOKEN_KEY)}`,
},
}
);

3
surfsense_web/app/dashboard/[search_space_id]/connectors/add/google-gmail-connector/page.tsx
View file

@ -24,6 +24,7 @@ import {
type SearchSourceConnector,
useSearchSourceConnectors,
} from "@/hooks/use-search-source-connectors";
import { AUTH_TOKEN_KEY } from "@/lib/constants";
export default function GoogleGmailConnectorPage() {
const router = useRouter();
@ -55,7 +56,7 @@ export default function GoogleGmailConnectorPage() {
{
method: "GET",
headers: {
Authorization: `Bearer ${localStorage.getItem("surfsense_bearer_token")}`,
Authorization: `Bearer ${localStorage.getItem(AUTH_TOKEN_KEY)}`,
},
}
);

3
surfsense_web/app/dashboard/[search_space_id]/documents/webpage/page.tsx
View file

@ -16,6 +16,7 @@ import {
CardTitle,
} from "@/components/ui/card";
import { Label } from "@/components/ui/label";
import { AUTH_TOKEN_KEY } from "@/lib/constants";
// URL validation regex - updated to support percent-encoded URLs (e.g., Latvian characters)
const urlRegex = /^https?:\/\/[^\s]+$/;
@ -69,7 +70,7 @@ export default function WebpageCrawler() {
method: "POST",
headers: {
"Content-Type": "application/json",
Authorization: `Bearer ${localStorage.getItem("surfsense_bearer_token")}`,
Authorization: `Bearer ${localStorage.getItem(AUTH_TOKEN_KEY)}`,
},
body: JSON.stringify({
document_type: "CRAWLED_URL",

3
surfsense_web/app/dashboard/[search_space_id]/onboard/page.tsx
View file

@ -12,6 +12,7 @@ import { Button } from "@/components/ui/button";
import { Card, CardContent, CardDescription, CardHeader, CardTitle } from "@/components/ui/card";
import { Progress } from "@/components/ui/progress";
import { useGlobalLLMConfigs, useLLMConfigs, useLLMPreferences } from "@/hooks/use-llm-configs";
import { AUTH_TOKEN_KEY } from "@/lib/constants";
const TOTAL_STEPS = 2;
@ -38,7 +39,7 @@ const OnboardPage = () => {
// Check if user is authenticated
useEffect(() => {
const token = localStorage.getItem("surfsense_bearer_token");
const token = localStorage.getItem(AUTH_TOKEN_KEY);
if (!token) {
router.push("/login");
return;

3
surfsense_web/app/dashboard/layout.tsx
View file

@ -6,6 +6,7 @@ import { useEffect, useState } from "react";
import { AnnouncementBanner } from "@/components/announcement-banner";
import { Card, CardContent, CardDescription, CardHeader, CardTitle } from "@/components/ui/card";
import { baseApiService } from "@/lib/apis/base-api.service";
import { AUTH_TOKEN_KEY } from "@/lib/constants";
interface DashboardLayoutProps {
children: React.ReactNode;
@ -17,7 +18,7 @@ export default function DashboardLayout({ children }: DashboardLayoutProps) {
useEffect(() => {
// Check if user is authenticated
const token = localStorage.getItem("surfsense_bearer_token");
const token = localStorage.getItem(AUTH_TOKEN_KEY);
if (!token) {
router.push("/login");
return;

3
surfsense_web/app/dashboard/page.tsx
View file

@ -35,6 +35,7 @@ import { Spotlight } from "@/components/ui/spotlight";
import { Tilt } from "@/components/ui/tilt";
import { useUser } from "@/hooks";
import { useSearchSpaces } from "@/hooks/use-search-spaces";
import { AUTH_TOKEN_KEY } from "@/lib/constants";
/**
* Formats a date string into a readable format
@ -177,7 +178,7 @@ const DashboardPage = () => {
{
method: "DELETE",
headers: {
Authorization: `Bearer ${localStorage.getItem("surfsense_bearer_token")}`,
Authorization: `Bearer ${localStorage.getItem(AUTH_TOKEN_KEY)}`,
},
}
);

3
surfsense_web/app/dashboard/searchspaces/page.tsx
View file

@ -4,6 +4,7 @@ import { motion } from "motion/react";
import { useRouter } from "next/navigation";
import { toast } from "sonner";
import { SearchSpaceForm } from "@/components/search-space-form";
import { AUTH_TOKEN_KEY } from "@/lib/constants";
export default function SearchSpacesPage() {
const router = useRouter();
const handleCreateSearchSpace = async (data: { name: string; description: string }) => {
@ -14,7 +15,7 @@ export default function SearchSpacesPage() {
method: "POST",
headers: {
"Content-Type": "application/json",
Authorization: `Bearer ${localStorage.getItem("surfsense_bearer_token")}`,
Authorization: `Bearer ${localStorage.getItem(AUTH_TOKEN_KEY)}`,
},
body: JSON.stringify(data),
}

5
surfsense_web/app/dashboard/site-settings/page.tsx
View file

@ -6,6 +6,7 @@ import { toast } from "sonner";
import { useSiteConfig } from "@/contexts/SiteConfigContext";
import { Card, CardContent, CardDescription, CardHeader, CardTitle } from "@/components/ui/card";
import { Loader2 } from "lucide-react";
import { AUTH_TOKEN_KEY } from "@/lib/constants";
interface SiteConfigForm {
// Header/Navbar toggles
@ -67,7 +68,7 @@ export default function SiteSettingsPage() {
const checkSuperuser = async () => {
try {
const backendUrl = process.env.NEXT_PUBLIC_FASTAPI_BACKEND_URL || "http://localhost:8000";
const token = localStorage.getItem("surfsense_bearer_token");
const token = localStorage.getItem(AUTH_TOKEN_KEY);
if (!token) {
router.push("/login");
@ -150,7 +151,7 @@ export default function SiteSettingsPage() {
setIsSaving(true);
try {
const backendUrl = process.env.NEXT_PUBLIC_FASTAPI_BACKEND_URL || "http://localhost:8000";
const token = localStorage.getItem("surfsense_bearer_token");
const token = localStorage.getItem(AUTH_TOKEN_KEY);
const response = await fetch(`${backendUrl}/api/v1/site-config`, {
method: "PUT",

3
surfsense_web/atoms/chats/chat-mutation.atoms.ts
View file

@ -2,13 +2,14 @@ import { atomWithMutation } from "jotai-tanstack-query";
import { toast } from "sonner";
import type { Chat } from "@/app/dashboard/[search_space_id]/chats/chats-client";
import { chatApiService } from "@/lib/apis/chats-api.service";
import { AUTH_TOKEN_KEY } from "@/lib/constants";
import { cacheKeys } from "@/lib/query-client/cache-keys";
import { queryClient } from "@/lib/query-client/client";
import { activeSearchSpaceIdAtom } from "../seach-spaces/seach-space-queries.atom";
export const deleteChatMutationAtom = atomWithMutation((get) => {
const searchSpaceId = get(activeSearchSpaceIdAtom);
const authToken = localStorage.getItem("surfsense_bearer_token");
const authToken = localStorage.getItem(AUTH_TOKEN_KEY);
return {
mutationKey: cacheKeys.activeSearchSpace.chats(searchSpaceId ?? ""),

5
surfsense_web/atoms/chats/chat-querie.atoms.ts
View file

@ -4,6 +4,7 @@ import type { ChatDetails } from "@/app/dashboard/[search_space_id]/chats/chats-
import type { PodcastItem } from "@/app/dashboard/[search_space_id]/podcasts/podcasts-client";
import { activeSearchSpaceIdAtom } from "@/atoms/seach-spaces/seach-space-queries.atom";
import { chatApiService } from "@/lib/apis/chats-api.service";
import { AUTH_TOKEN_KEY } from "@/lib/constants";
import { getPodcastByChatId } from "@/lib/apis/podcasts.api";
import { cacheKeys } from "@/lib/query-client/cache-keys";
@ -17,7 +18,7 @@ export const activeChatIdAtom = atom<string | null>(null);
export const activeChatAtom = atomWithQuery<ActiveChatState>((get) => {
const activeChatId = get(activeChatIdAtom);
const authToken = localStorage.getItem("surfsense_bearer_token");
const authToken = localStorage.getItem(AUTH_TOKEN_KEY);
return {
queryKey: cacheKeys.activeSearchSpace.activeChat(activeChatId ?? ""),
@ -42,7 +43,7 @@ export const activeChatAtom = atomWithQuery<ActiveChatState>((get) => {
export const activeSearchSpaceChatsAtom = atomWithQuery((get) => {
const searchSpaceId = get(activeSearchSpaceIdAtom);
const authToken = localStorage.getItem("surfsense_bearer_token");
const authToken = localStorage.getItem(AUTH_TOKEN_KEY);
return {
queryKey: cacheKeys.activeSearchSpace.chats(searchSpaceId ?? ""),

3
surfsense_web/components/TokenHandler.tsx
View file

@ -3,6 +3,7 @@
import { useRouter, useSearchParams } from "next/navigation";
import { useEffect } from "react";
import { baseApiService } from "@/lib/apis/base-api.service";
import { AUTH_TOKEN_KEY } from "@/lib/constants";
interface TokenHandlerProps {
redirectPath?: string; // Path to redirect after storing token
@ -20,7 +21,7 @@ interface TokenHandlerProps {
const TokenHandler = ({
redirectPath = "/",
tokenParamName = "token",
storageKey = "surfsense_bearer_token",
storageKey = AUTH_TOKEN_KEY,
}: TokenHandlerProps) => {
const router = useRouter();
const searchParams = useSearchParams();

3
surfsense_web/components/UserDropdown.tsx
View file

@ -4,6 +4,7 @@ import { BadgeCheck, LogOut, Settings } from "lucide-react";
import { useRouter } from "next/navigation";
import { Avatar, AvatarFallback, AvatarImage } from "@/components/ui/avatar";
import { baseApiService } from "@/lib/apis/base-api.service";
import { AUTH_TOKEN_KEY } from "@/lib/constants";
import { Button } from "@/components/ui/button";
import {
DropdownMenu,
@ -29,7 +30,7 @@ export function UserDropdown({
const handleLogout = () => {
try {
if (typeof window !== "undefined") {
localStorage.removeItem("surfsense_bearer_token");
localStorage.removeItem(AUTH_TOKEN_KEY);
// Clear the baseApiService token to prevent stale auth state
baseApiService.setBearerToken("");
router.push("/");

3
surfsense_web/components/chat/ChatPanel/ChatPanelContainer.tsx
View file

@ -7,6 +7,7 @@ import { activeChathatUIAtom } from "@/atoms/chats/ui.atoms";
import { generatePodcast } from "@/lib/apis/podcasts.api";
import { cn } from "@/lib/utils";
import { ChatPanelView } from "./ChatPanelView";
import { AUTH_TOKEN_KEY } from "@/lib/constants";
export interface GeneratePodcastRequest {
type: "CHAT" | "DOCUMENT";
@ -23,7 +24,7 @@ export function ChatPanelContainer() {
error: chatError,
} = useAtomValue(activeChatAtom);
const activeChatIdState = useAtomValue(activeChatIdAtom);
const authToken = localStorage.getItem("surfsense_bearer_token");
const authToken = localStorage.getItem(AUTH_TOKEN_KEY);
const { isChatPannelOpen } = useAtomValue(activeChathatUIAtom);
const handleGeneratePodcast = async (request: GeneratePodcastRequest) => {

3
surfsense_web/components/chat/ChatPanel/PodcastPlayer/PodcastPlayer.tsx
View file

@ -8,6 +8,7 @@ import type { PodcastItem } from "@/app/dashboard/[search_space_id]/podcasts/pod
import { Button } from "@/components/ui/button";
import { Slider } from "@/components/ui/slider";
import { PodcastPlayerCompactSkeleton } from "./PodcastPlayerCompactSkeleton";
import { AUTH_TOKEN_KEY } from "@/lib/constants";
interface PodcastPlayerProps {
podcast: PodcastItem | null;
@ -56,7 +57,7 @@ export function PodcastPlayer({
const loadPodcast = async () => {
setIsFetching(true);
try {
const token = localStorage.getItem("surfsense_bearer_token");
const token = localStorage.getItem(AUTH_TOKEN_KEY);
if (!token) {
throw new Error("Authentication token not found.");
}

3
surfsense_web/components/sources/DocumentUploadTab.tsx
View file

@ -15,6 +15,7 @@ import { Card, CardContent, CardDescription, CardHeader, CardTitle } from "@/com
import { Progress } from "@/components/ui/progress";
import { Separator } from "@/components/ui/separator";
import { GridPattern } from "./GridPattern";
import { AUTH_TOKEN_KEY } from "@/lib/constants";
interface DocumentUploadTabProps {
searchSpaceId: string;
@ -169,7 +170,7 @@ export function DocumentUploadTab({ searchSpaceId }: DocumentUploadTabProps) {
{
method: "POST",
headers: {
Authorization: `Bearer ${window.localStorage.getItem("surfsense_bearer_token")}`,
Authorization: `Bearer ${window.localStorage.getItem(AUTH_TOKEN_KEY)}`,
},
body: formData,
}

3
surfsense_web/components/sources/YouTubeTab.tsx
View file

@ -19,6 +19,7 @@ import {
CardTitle,
} from "@/components/ui/card";
import { Label } from "@/components/ui/label";
import { AUTH_TOKEN_KEY } from "@/lib/constants";
const youtubeRegex =
/^(https:\/\/)?(www\.)?(youtube\.com\/watch\?v=|youtu\.be\/)([a-zA-Z0-9_-]{11})$/;
@ -72,7 +73,7 @@ export function YouTubeTab({ searchSpaceId }: YouTubeTabProps) {
method: "POST",
headers: {
"Content-Type": "application/json",
Authorization: `Bearer ${localStorage.getItem("surfsense_bearer_token")}`,
Authorization: `Bearer ${localStorage.getItem(AUTH_TOKEN_KEY)}`,
},
body: JSON.stringify({
document_type: "YOUTUBE_VIDEO",

3
surfsense_web/hooks/use-api-key.ts
View file

@ -1,5 +1,6 @@
import { useCallback, useEffect, useState } from "react";
import { toast } from "sonner";
import { AUTH_TOKEN_KEY } from "@/lib/constants";
interface UseApiKeyReturn {
apiKey: string | null;
@ -17,7 +18,7 @@ export function useApiKey(): UseApiKeyReturn {
// Load API key from localStorage
const loadApiKey = () => {
try {
const token = localStorage.getItem("surfsense_bearer_token");
const token = localStorage.getItem(AUTH_TOKEN_KEY);
setApiKey(token);
} catch (error) {
console.error("Error loading API key:", error);

3
surfsense_web/hooks/use-chat.ts
View file

@ -3,6 +3,7 @@ import { useCallback, useEffect, useState } from "react";
import type { ChatDetails } from "@/app/dashboard/[search_space_id]/chats/chats-client";
import type { ResearchMode } from "@/components/chat";
import type { Document } from "@/hooks/use-documents";
import { AUTH_TOKEN_KEY } from "@/lib/constants";
interface UseChatStateProps {
search_space_id: string;
@ -22,7 +23,7 @@ export function useChatState({ chat_id }: UseChatStateProps) {
const [topK, setTopK] = useState<number>(5);
useEffect(() => {
const bearerToken = localStorage.getItem("surfsense_bearer_token");
const bearerToken = localStorage.getItem(AUTH_TOKEN_KEY);
setToken(bearerToken);
}, []);

3
surfsense_web/hooks/use-connector-edit-page.ts
View file

@ -15,6 +15,7 @@ import {
type SearchSourceConnector,
useSearchSourceConnectors,
} from "@/hooks/use-search-source-connectors";
import { AUTH_TOKEN_KEY } from "@/lib/constants";
const normalizeListInput = (value: unknown): string[] => {
if (Array.isArray(value)) {
@ -174,7 +175,7 @@ export function useConnectorEditPage(connectorId: number, searchSpaceId: string)
setIsFetchingRepos(true);
setFetchedRepos(null);
try {
const token = localStorage.getItem("surfsense_bearer_token");
const token = localStorage.getItem(AUTH_TOKEN_KEY);
if (!token) throw new Error("No auth token");
const response = await fetch(
`${process.env.NEXT_PUBLIC_FASTAPI_BACKEND_URL}/api/v1/github/repositories`,

12
surfsense_web/hooks/use-connectors.ts
View file

@ -1,3 +1,5 @@
import { AUTH_TOKEN_KEY } from "@/lib/constants";
// Types for connector API
export interface ConnectorConfig {
[key: string]: string;
@ -38,7 +40,7 @@ export const ConnectorService = {
method: "POST",
headers: {
"Content-Type": "application/json",
Authorization: `Bearer ${localStorage.getItem("surfsense_bearer_token")}`,
Authorization: `Bearer ${localStorage.getItem(AUTH_TOKEN_KEY)}`,
},
body: JSON.stringify(data),
}
@ -58,7 +60,7 @@ export const ConnectorService = {
`${process.env.NEXT_PUBLIC_FASTAPI_BACKEND_URL}/api/v1/search-source-connectors?skip=${skip}&limit=${limit}`,
{
headers: {
Authorization: `Bearer ${localStorage.getItem("surfsense_bearer_token")}`,
Authorization: `Bearer ${localStorage.getItem(AUTH_TOKEN_KEY)}`,
},
}
);
@ -77,7 +79,7 @@ export const ConnectorService = {
`${process.env.NEXT_PUBLIC_FASTAPI_BACKEND_URL}/api/v1/search-source-connectors/${connectorId}`,
{
headers: {
Authorization: `Bearer ${localStorage.getItem("surfsense_bearer_token")}`,
Authorization: `Bearer ${localStorage.getItem(AUTH_TOKEN_KEY)}`,
},
}
);
@ -98,7 +100,7 @@ export const ConnectorService = {
method: "PUT",
headers: {
"Content-Type": "application/json",
Authorization: `Bearer ${localStorage.getItem("surfsense_bearer_token")}`,
Authorization: `Bearer ${localStorage.getItem(AUTH_TOKEN_KEY)}`,
},
body: JSON.stringify(data),
}
@ -119,7 +121,7 @@ export const ConnectorService = {
{
method: "DELETE",
headers: {
Authorization: `Bearer ${localStorage.getItem("surfsense_bearer_token")}`,
Authorization: `Bearer ${localStorage.getItem(AUTH_TOKEN_KEY)}`,
},
}
);

3
surfsense_web/hooks/use-document-by-chunk.ts
View file

@ -1,6 +1,7 @@
"use client";
import { useCallback, useState } from "react";
import { toast } from "sonner";
import { AUTH_TOKEN_KEY } from "@/lib/constants";
export interface Chunk {
id: number;
@ -53,7 +54,7 @@ export function useDocumentByChunk() {
`${process.env.NEXT_PUBLIC_FASTAPI_BACKEND_URL}/api/v1/documents/by-chunk/${chunkId}`,
{
headers: {
Authorization: `Bearer ${localStorage.getItem("surfsense_bearer_token")}`,
Authorization: `Bearer ${localStorage.getItem(AUTH_TOKEN_KEY)}`,
"Content-Type": "application/json",
},
method: "GET",

3
surfsense_web/hooks/use-document-types.ts
View file

@ -1,4 +1,5 @@
import { useCallback, useEffect, useState } from "react";
import { AUTH_TOKEN_KEY } from "@/lib/constants";
export interface DocumentTypeCount {
type: string;
@ -23,7 +24,7 @@ export const useDocumentTypes = (searchSpaceId?: number, lazy: boolean = false)
try {
setIsLoading(true);
setError(null);
const token = localStorage.getItem("surfsense_bearer_token");
const token = localStorage.getItem(AUTH_TOKEN_KEY);
if (!token) {
throw new Error("No authentication token found");

9
surfsense_web/hooks/use-documents.ts
View file

@ -2,6 +2,7 @@
import { useCallback, useEffect, useState } from "react";
import { toast } from "sonner";
import { normalizeListResponse } from "@/lib/pagination";
import { AUTH_TOKEN_KEY } from "@/lib/constants";
export interface Document {
id: number;
@ -82,7 +83,7 @@ export function useDocuments(searchSpaceId: number, options?: UseDocumentsOption
`${process.env.NEXT_PUBLIC_FASTAPI_BACKEND_URL}/api/v1/documents?${params.toString()}`,
{
headers: {
Authorization: `Bearer ${localStorage.getItem("surfsense_bearer_token")}`,
Authorization: `Bearer ${localStorage.getItem(AUTH_TOKEN_KEY)}`,
},
method: "GET",
}
@ -163,7 +164,7 @@ export function useDocuments(searchSpaceId: number, options?: UseDocumentsOption
`${process.env.NEXT_PUBLIC_FASTAPI_BACKEND_URL}/api/v1/documents/search?${params.toString()}`,
{
headers: {
Authorization: `Bearer ${localStorage.getItem("surfsense_bearer_token")}`,
Authorization: `Bearer ${localStorage.getItem(AUTH_TOKEN_KEY)}`,
},
method: "GET",
}
@ -197,7 +198,7 @@ export function useDocuments(searchSpaceId: number, options?: UseDocumentsOption
`${process.env.NEXT_PUBLIC_FASTAPI_BACKEND_URL}/api/v1/documents/${documentId}`,
{
headers: {
Authorization: `Bearer ${localStorage.getItem("surfsense_bearer_token")}`,
Authorization: `Bearer ${localStorage.getItem(AUTH_TOKEN_KEY)}`,
},
method: "DELETE",
}
@ -232,7 +233,7 @@ export function useDocuments(searchSpaceId: number, options?: UseDocumentsOption
`${process.env.NEXT_PUBLIC_FASTAPI_BACKEND_URL}/api/v1/documents/type-counts?${params.toString()}`,
{
headers: {
Authorization: `Bearer ${localStorage.getItem("surfsense_bearer_token")}`,
Authorization: `Bearer ${localStorage.getItem(AUTH_TOKEN_KEY)}`,
},
method: "GET",
}

15
surfsense_web/hooks/use-llm-configs.ts
View file

@ -1,6 +1,7 @@
"use client";
import { useEffect, useState } from "react";
import { toast } from "sonner";
import { AUTH_TOKEN_KEY } from "@/lib/constants";
export interface LLMConfig {
id: number;
@ -65,7 +66,7 @@ export function useLLMConfigs(searchSpaceId: number | null) {
`${process.env.NEXT_PUBLIC_FASTAPI_BACKEND_URL}/api/v1/llm-configs?search_space_id=${searchSpaceId}`,
{
headers: {
Authorization: `Bearer ${localStorage.getItem("surfsense_bearer_token")}`,
Authorization: `Bearer ${localStorage.getItem(AUTH_TOKEN_KEY)}`,
},
method: "GET",
}
@ -98,7 +99,7 @@ export function useLLMConfigs(searchSpaceId: number | null) {
method: "POST",
headers: {
"Content-Type": "application/json",
Authorization: `Bearer ${localStorage.getItem("surfsense_bearer_token")}`,
Authorization: `Bearer ${localStorage.getItem(AUTH_TOKEN_KEY)}`,
},
body: JSON.stringify(config),
}
@ -127,7 +128,7 @@ export function useLLMConfigs(searchSpaceId: number | null) {
{
method: "DELETE",
headers: {
Authorization: `Bearer ${localStorage.getItem("surfsense_bearer_token")}`,
Authorization: `Bearer ${localStorage.getItem(AUTH_TOKEN_KEY)}`,
},
}
);
@ -157,7 +158,7 @@ export function useLLMConfigs(searchSpaceId: number | null) {
method: "PUT",
headers: {
"Content-Type": "application/json",
Authorization: `Bearer ${localStorage.getItem("surfsense_bearer_token")}`,
Authorization: `Bearer ${localStorage.getItem(AUTH_TOKEN_KEY)}`,
},
body: JSON.stringify(config),
}
@ -207,7 +208,7 @@ export function useLLMPreferences(searchSpaceId: number | null) {
`${process.env.NEXT_PUBLIC_FASTAPI_BACKEND_URL}/api/v1/search-spaces/${searchSpaceId}/llm-preferences`,
{
headers: {
Authorization: `Bearer ${localStorage.getItem("surfsense_bearer_token")}`,
Authorization: `Bearer ${localStorage.getItem(AUTH_TOKEN_KEY)}`,
},
method: "GET",
}
@ -245,7 +246,7 @@ export function useLLMPreferences(searchSpaceId: number | null) {
method: "PUT",
headers: {
"Content-Type": "application/json",
Authorization: `Bearer ${localStorage.getItem("surfsense_bearer_token")}`,
Authorization: `Bearer ${localStorage.getItem(AUTH_TOKEN_KEY)}`,
},
body: JSON.stringify(newPreferences),
}
@ -297,7 +298,7 @@ export function useGlobalLLMConfigs() {
`${process.env.NEXT_PUBLIC_FASTAPI_BACKEND_URL}/api/v1/global-llm-configs`,
{
headers: {
Authorization: `Bearer ${localStorage.getItem("surfsense_bearer_token")}`,
Authorization: `Bearer ${localStorage.getItem(AUTH_TOKEN_KEY)}`,
},
method: "GET",
}

13
surfsense_web/hooks/use-logs.ts
View file

@ -1,6 +1,7 @@
"use client";
import { useCallback, useEffect, useMemo, useState } from "react";
import { toast } from "sonner";
import { AUTH_TOKEN_KEY } from "@/lib/constants";
export type LogLevel = "DEBUG" | "INFO" | "WARNING" | "ERROR" | "CRITICAL";
export type LogStatus = "IN_PROGRESS" | "SUCCESS" | "FAILED";
@ -99,7 +100,7 @@ export function useLogs(searchSpaceId?: number, filters: LogFilters = {}) {
`${process.env.NEXT_PUBLIC_FASTAPI_BACKEND_URL}/api/v1/logs?${params}`,
{
headers: {
Authorization: `Bearer ${localStorage.getItem("surfsense_bearer_token")}`,
Authorization: `Bearer ${localStorage.getItem(AUTH_TOKEN_KEY)}`,
},
method: "GET",
}
@ -150,7 +151,7 @@ export function useLogs(searchSpaceId?: number, filters: LogFilters = {}) {
const response = await fetch(`${process.env.NEXT_PUBLIC_FASTAPI_BACKEND_URL}/api/v1/logs`, {
headers: {
"Content-Type": "application/json",
Authorization: `Bearer ${localStorage.getItem("surfsense_bearer_token")}`,
Authorization: `Bearer ${localStorage.getItem(AUTH_TOKEN_KEY)}`,
},
method: "POST",
body: JSON.stringify(logData),
@ -184,7 +185,7 @@ export function useLogs(searchSpaceId?: number, filters: LogFilters = {}) {
{
headers: {
"Content-Type": "application/json",
Authorization: `Bearer ${localStorage.getItem("surfsense_bearer_token")}`,
Authorization: `Bearer ${localStorage.getItem(AUTH_TOKEN_KEY)}`,
},
method: "PUT",
body: JSON.stringify(updateData),
@ -216,7 +217,7 @@ export function useLogs(searchSpaceId?: number, filters: LogFilters = {}) {
`${process.env.NEXT_PUBLIC_FASTAPI_BACKEND_URL}/api/v1/logs/${logId}`,
{
headers: {
Authorization: `Bearer ${localStorage.getItem("surfsense_bearer_token")}`,
Authorization: `Bearer ${localStorage.getItem(AUTH_TOKEN_KEY)}`,
},
method: "DELETE",
}
@ -244,7 +245,7 @@ export function useLogs(searchSpaceId?: number, filters: LogFilters = {}) {
`${process.env.NEXT_PUBLIC_FASTAPI_BACKEND_URL}/api/v1/logs/${logId}`,
{
headers: {
Authorization: `Bearer ${localStorage.getItem("surfsense_bearer_token")}`,
Authorization: `Bearer ${localStorage.getItem(AUTH_TOKEN_KEY)}`,
},
method: "GET",
}
@ -291,7 +292,7 @@ export function useLogsSummary(searchSpaceId: number, hours: number = 24) {
`${process.env.NEXT_PUBLIC_FASTAPI_BACKEND_URL}/api/v1/logs/search-space/${searchSpaceId}/summary?hours=${hours}`,
{
headers: {
Authorization: `Bearer ${localStorage.getItem("surfsense_bearer_token")}`,
Authorization: `Bearer ${localStorage.getItem(AUTH_TOKEN_KEY)}`,
},
method: "GET",
}

11
surfsense_web/hooks/use-search-source-connectors.ts
View file

@ -1,4 +1,5 @@
import { useCallback, useEffect, useState } from "react";
import { AUTH_TOKEN_KEY } from "@/lib/constants";
export interface SearchSourceConnector {
id: number;
@ -66,7 +67,7 @@ export const useSearchSourceConnectors = (lazy: boolean = false, searchSpaceId?:
try {
setIsLoading(true);
setError(null);
const token = localStorage.getItem("surfsense_bearer_token");
const token = localStorage.getItem(AUTH_TOKEN_KEY);
if (!token) {
throw new Error("No authentication token found");
@ -176,7 +177,7 @@ export const useSearchSourceConnectors = (lazy: boolean = false, searchSpaceId?:
spaceId: number
) => {
try {
const token = localStorage.getItem("surfsense_bearer_token");
const token = localStorage.getItem(AUTH_TOKEN_KEY);
if (!token) {
throw new Error("No authentication token found");
@ -222,7 +223,7 @@ export const useSearchSourceConnectors = (lazy: boolean = false, searchSpaceId?:
>
) => {
try {
const token = localStorage.getItem("surfsense_bearer_token");
const token = localStorage.getItem(AUTH_TOKEN_KEY);
if (!token) {
throw new Error("No authentication token found");
@ -262,7 +263,7 @@ export const useSearchSourceConnectors = (lazy: boolean = false, searchSpaceId?:
*/
const deleteConnector = async (connectorId: number) => {
try {
const token = localStorage.getItem("surfsense_bearer_token");
const token = localStorage.getItem(AUTH_TOKEN_KEY);
if (!token) {
throw new Error("No authentication token found");
@ -302,7 +303,7 @@ export const useSearchSourceConnectors = (lazy: boolean = false, searchSpaceId?:
endDate?: string
) => {
try {
const token = localStorage.getItem("surfsense_bearer_token");
const token = localStorage.getItem(AUTH_TOKEN_KEY);
if (!token) {
throw new Error("No authentication token found");

5
surfsense_web/hooks/use-search-spaces.ts
View file

@ -2,6 +2,7 @@
import { useEffect, useState } from "react";
import { toast } from "sonner";
import { AUTH_TOKEN_KEY } from "@/lib/constants";
interface SearchSpace {
id: number;
@ -24,7 +25,7 @@ export function useSearchSpaces() {
`${process.env.NEXT_PUBLIC_FASTAPI_BACKEND_URL}/api/v1/searchspaces`,
{
headers: {
Authorization: `Bearer ${localStorage.getItem("surfsense_bearer_token")}`,
Authorization: `Bearer ${localStorage.getItem(AUTH_TOKEN_KEY)}`,
},
method: "GET",
}
@ -57,7 +58,7 @@ export function useSearchSpaces() {
`${process.env.NEXT_PUBLIC_FASTAPI_BACKEND_URL}/api/v1/searchspaces`,
{
headers: {
Authorization: `Bearer ${localStorage.getItem("surfsense_bearer_token")}`,
Authorization: `Bearer ${localStorage.getItem(AUTH_TOKEN_KEY)}`,
},
method: "GET",
}