added update docbase feature

src/components/AttributeAdder/AttributeAdder.tsx

@@ -1,23 +1,31 @@
 import { useEffect, useState } from 'react';
 import './AttributeAdder.scss';
 import Icon from '../Icon/Icon';
+import { useIsDocbaseTaskRunning } from '../../providers/DocBaseTaskProvider';
 
 interface Props {
-	populateAble: boolean;
 	onListChange: (list: string[]) => void;
+	populateAble?: boolean;
+	rerunAble?: boolean;
+	onRerun?: (list: string[]) => void;
 	initialList?: string[];
 }
 
 /**
  * A component that allows the user to add attributes to there document base
  */
 function AttributeAdder({
-	populateAble = true,
 	onListChange,
+	populateAble = false,
+	rerunAble = false,
 	initialList = [],
+	// eslint-disable-next-line @typescript-eslint/no-unused-vars
+	onRerun = (_list: string[]) => {},
 }: Props) {
 	const [attList, setAttList] = useState<string[]>([]);
 
+	const isDocBaseTaskRunning = useIsDocbaseTaskRunning();
+
 	useEffect(
 		() => {
 			if (
@@ -57,11 +65,28 @@ function AttributeAdder({
 		}
 	};
 
+	const isListEqualToInitialList = () => {
+		if (attList.length !== initialList.length) {
+			return false;
+		}
+		attList.forEach((att) => {
+			if (!initialList.includes(att)) {
+				return false;
+			}
+		});
+		return true;
+	};
+
 	return (
 		<div className="AttributeAdder">
+			{rerunAble && (
+				<p>
+					<i>Modify the attributes to rerun the document base.</i>
+				</p>
+			)}
 			{attList.length === 0 && (
 				<p>
-					<i>Enter a attribute...</i>
+					<i>Enter an attribute...</i>
 				</p>
 			)}
 			{attList.map((att, i) => {
@@ -121,6 +146,17 @@ function AttributeAdder({
 					Attributes
 				</button>
 			)}
+			{rerunAble &&
+				attList.length > 0 &&
+				!isListEqualToInitialList() &&
+				!isDocBaseTaskRunning() && (
+					<button
+						className="btn populateBtn"
+						onClick={() => onRerun(attList)}
+					>
+						<i className="bi bi-play icon"></i>Rerun
+					</button>
+				)}
 		</div>
 	);
 }

src/components/DocbaseViewer/DocbaseViewer.tsx

@@ -1,3 +1,4 @@
+import { useUpdateDocbaseAttributesTask } from '../../providers/DocBaseTaskProvider';
 import DocBase from '../../types/DocBase';
 import AttributeAdder from '../AttributeAdder/AttributeAdder';
 import NuggetDocumentViewer from '../NuggetViewer/NuggetDocumentViewer';
@@ -13,6 +14,8 @@ interface Props {
  * @param docBase The docbase to view
  */
 function DocbaseViewer({ docBase, onClose }: Props) {
+	const updateDocbaseAttributesTask = useUpdateDocbaseAttributesTask();
+
 	return (
 		<>
 			<div
@@ -25,14 +28,24 @@ function DocbaseViewer({ docBase, onClose }: Props) {
 				<i className="bi bi-x-lg"></i>
 			</div>
 			<div className="DocbaseViewer">
-				<h1>{docBase.name}</h1>
+				<h1>
+					DocBase: <i>{docBase.name}</i>
+				</h1>
 				<h2>Attributes</h2>
 				<AttributeAdder
-					populateAble={false}
+					rerunAble={true}
 					onListChange={(list: string[]) => {
 						docBase.attributes = list;
 					}}
 					initialList={docBase.attributes}
+					onRerun={(list: string[]) => {
+						updateDocbaseAttributesTask(
+							docBase.organizationId,
+							docBase.name,
+							list
+						);
+						onClose();
+					}}
 				></AttributeAdder>
 				{/* <ul className="ver">
 					{docBase.attributes.map((attribute, index) => {
@@ -47,7 +60,7 @@ function DocbaseViewer({ docBase, onClose }: Props) {
 				{docBase.docs.map((doc, index) => {
 					return <NuggetDocumentViewer key={index} doc={doc} />;
 				})}
-				<button className="btn mb" onClick={onClose}>
+				<button className="btn mb ml mt" onClick={onClose}>
 					Close
 				</button>
 			</div>

src/components/Navbar/Navbar.tsx

@@ -1,7 +1,7 @@
 import { Link, useNavigate } from 'react-router-dom';
 import './Navbar.scss';
 import { useState } from 'react';
-import { useGetUsername } from '../../providers/UserProvider';
+import { useGetUsername, useLoggedIn } from '../../providers/UserProvider';
 import { useIsDarkTheme, useToggleTheme } from '../../providers/ThemeProvider';
 import Icon from '../Icon/Icon';
 
@@ -12,6 +12,7 @@ function Navbar() {
 	const navigate = useNavigate();
 
 	const getUserName = useGetUsername();
+	const isLoggedIn = useLoggedIn();
 
 	const [username] = useState(getUserName());
 
@@ -70,7 +71,7 @@ function Navbar() {
 					{/* <i className="bi bi-question-circle icon"></i> */}
 					Help
 				</Link>
-				{username !== '' ? (
+				{isLoggedIn() ? (
 					<Link to={'/profile'}>
 						<div className="profilePicture">
 							{username.slice(0, 1).toUpperCase()}

src/providers/DocBaseTaskProvider.tsx

@@ -21,6 +21,11 @@ const DocBaseTaskContext = React.createContext({
 		_documentIDs: number[],
 		_attributes: string[]
 	) => {},
+	updateDocbaseAttributesTask: (
+		_organizationId: number,
+		_baseName: string,
+		_attributes: string[]
+	) => {},
 	loadDocbaseTask: (_organizationId: number, _baseName: string) => {},
 });
 
@@ -46,6 +51,17 @@ export function useCreateDocbaseTask() {
 	return context.createDocbaseTask;
 }
 
+// eslint-disable-next-line react-refresh/only-export-components
+export function useUpdateDocbaseAttributesTask() {
+	const context = React.useContext(DocBaseTaskContext);
+	if (!context) {
+		throw new Error(
+			'useUpdateDocbaseAttributesTask must be used within a DocBaseTaskProvider'
+		);
+	}
+	return context.updateDocbaseAttributesTask;
+}
+
 // eslint-disable-next-line react-refresh/only-export-components
 export function useLoadDocbaseTask() {
 	const context = React.useContext(DocBaseTaskContext);
@@ -146,7 +162,11 @@ export function DocBaseTaskProvider({ children }: Props) {
 					setLoadingScreen(false);
 					playAudio(MyAudio.SUCCESS);
 
-					const docBase = new DocBase(baseName, attributes);
+					const docBase = new DocBase(
+						baseName,
+						organizationId,
+						attributes
+					);
 					for (const nugget of res.meta.document_base_to_ui.msg
 						.nuggets) {
 						try {
@@ -190,6 +210,121 @@ export function DocBaseTaskProvider({ children }: Props) {
 		}, intervalTime);
 	};
 
+	const updateDocbaseAttributesTask = async (
+		organizationId: number,
+		baseName: string,
+		attributes: string[]
+	) => {
+		if (isRunning) {
+			Logger.warn(
+				'Docbase task is already running, cannot start another'
+			);
+			return;
+		}
+
+		// start the task
+		const taskId = await APIService.updateDocumentBaseAttributes(
+			organizationId,
+			baseName,
+			attributes
+		);
+
+		if (taskId == undefined) {
+			showNotification('Error', 'Failed to update Docbase ' + baseName);
+			return;
+		}
+
+		Logger.log('Task: Update Attributes Docbase ' + baseName);
+		Logger.log('Task ID: ' + taskId);
+
+		sessionStorage.setItem('docbaseId', taskId);
+		setLoadingScreen(
+			true,
+			'Updating Docbase ' + baseName + '...',
+			'Please wait...',
+			taskId
+		);
+
+		setLoadingScreenLock(true);
+		setIsRunning(true);
+
+		const updateInterval = setInterval(() => {
+			// TODO use type
+			// eslint-disable-next-line @typescript-eslint/no-explicit-any
+			APIService.getTaskStatus(taskId).then((res): any => {
+				Logger.log(res);
+				if (
+					res == undefined ||
+					res.state.toUpperCase().trim() === 'FAILURE'
+				) {
+					setLoadingScreenLock(false);
+					setLoadingScreen(false);
+					playAudio(MyAudio.ERROR);
+
+					showNotification(
+						'Error',
+						'Failed to update Docbase ' + baseName
+					);
+					sessionStorage.removeItem('docbaseId');
+					setDocBase(undefined);
+					setIsRunning(false);
+					clearInterval(updateInterval);
+					return;
+				}
+
+				if (res.state === 'SUCCESS') {
+					setLoadingScreenLock(false);
+					setLoadingScreen(false);
+					playAudio(MyAudio.SUCCESS);
+
+					const docBase = new DocBase(
+						baseName,
+						organizationId,
+						attributes
+					);
+					for (const nugget of res.meta.document_base_to_ui.msg
+						.nuggets) {
+						try {
+							docBase.addNugget(
+								nugget.document.name,
+								nugget.document.text,
+								nugget.start_char,
+								nugget.end_char
+							);
+						} catch (error) {
+							showNotification(
+								'Error',
+								'Something went wrong translating the nuggets.'
+							);
+						}
+					}
+					sessionStorage.removeItem('docbaseId');
+					setDocBase(docBase);
+					setIsRunning(false);
+					clearInterval(updateInterval);
+					return;
+				}
+
+				let info = res.state;
+
+				if (res.meta.status !== undefined) {
+					info = res.meta.status;
+				}
+
+				info += info.endsWith('...') ? '' : '...';
+
+				// update loading screen
+				setLoadingScreen(
+					true,
+					'Updating Docbase ' + baseName + '...',
+					info,
+					taskId,
+					true
+				);
+			});
+		}, intervalTime);
+	};
+
 	const loadDocbaseTask = async (
 		organizationId: number,
 		baseName: string
@@ -255,6 +390,7 @@ export function DocBaseTaskProvider({ children }: Props) {
 
 					const docBase = new DocBase(
 						baseName,
+						organizationId,
 						res.meta.document_base_to_ui.msg.attributes ?? []
 					);
 					for (const nugget of res.meta.document_base_to_ui.msg
@@ -316,6 +452,7 @@ export function DocBaseTaskProvider({ children }: Props) {
 				isDocbaseTaskRunning: isDocbaseTaskRunning,
 				createDocbaseTask: createDocbaseTask,
 				loadDocbaseTask: loadDocbaseTask,
+				updateDocbaseAttributesTask: updateDocbaseAttributesTask,
 			}}
 		>
 			{docBase && <DocbaseViewer docBase={docBase} onClose={onClose} />}

src/types/DocBase.ts

@@ -5,11 +5,13 @@ import NuggetDocument from './NuggetDocument';
  */
 class DocBase {
 	readonly name: string;
+	readonly organizationId: number;
 	attributes: string[];
 	docs: NuggetDocument[];
 
-	constructor(name: string, attributes: string[]) {
+	constructor(name: string, organizationId: number, attributes: string[]) {
 		this.name = name;
+		this.organizationId = organizationId;
 		this.attributes = attributes;
 		this.docs = [];
 	}