aswi20220x00-gitlab/webapp/pages/documents/annotator/index.tsx @ 5004f8b6

import 'antd/dist/antd.css';
import React, { useContext, useEffect, useState } from 'react';

import { useUnauthRedirect } from '../../../hooks';
import { useRouter } from 'next/router';
import { Button, Table, Tag, Typography } from 'antd';
import { faFileLines } from '@fortawesome/free-solid-svg-icons';
import { FontAwesomeIcon } from '@fortawesome/react-fontawesome';
import { LoggedUserContext } from '../../../contexts/LoggedUserContext';
import { MainLayout } from '../../../layouts/MainLayout';
import { userController } from '../../../controllers';
import { AnnotationListInfo } from '../../../api';

function UserDocumentPage() {
    const redirecting = useUnauthRedirect('/login');
    const { logout, role } = useContext(LoggedUserContext);
    const router = useRouter();
    const [documents, setDocuments] = useState<AnnotationListInfo[]>([]);

    useEffect(() => {
        async function fetchData() {
            let docs = (await userController.userAnnotationsGet()).data.annotations;
            if (!docs) {
                setDocuments([]);
            } else {
                setDocuments(docs);
            }
        }

        if (!redirecting /* && role === 'ANNOTATOR'*/) {
            // admin can also fetch data if he likes so
            fetchData();
        }
    }, [logout, redirecting, role, router]);

    const columns = [
        {
            title: 'Název dokumentu',
            dataIndex: 'documentName',
            key: 'documentName',
        },
        {
            title: 'Stav anotace',
            key: 'state',
            dataIndex: 'state',
            render: (state: string) => {
                let color = 'green';
                let label = 'Hotovo';
                if (state === 'NEW') {
                    color = 'volcano';
                    label = 'Nový';
                }
                if (state === 'IN_PROGRESS') {
                    color = 'orange';
                    label = 'Rozpracováno';
                }
                return (
                    <Tag color={color} key={label}>
                        {label.toUpperCase()}
                    </Tag>
                );
            },
        },
        {
            title: '',
            key: 'action',
            dataIndex: 'annotationId',
            render: (anotationId: number) => (
                <Button
                    key={anotationId}
                    type={'primary'}
                    onClick={() => router.push('/annotation/' + anotationId)}
                >
                    Anotovat
                </Button>
            ),
        },
    ];

    return redirecting /*|| role !== 'ANNOTATOR' */ ? null : (
        <MainLayout>
            <Typography.Title level={2}>
                <FontAwesomeIcon icon={faFileLines} /> Dokumenty
            </Typography.Title>
            <Table
                columns={columns}
                dataSource={documents}
                size="small"
                scroll={{ y: 500 }}
            />
        </MainLayout>
    );
}

export default UserDocumentPage;