From fdcee29a0d591f43417e8c26c3280a0585be235f Mon Sep 17 00:00:00 2001
From: swve <bdswve@gmail.com>
Date: Sun, 19 Feb 2023 19:28:38 +0100
Subject: [PATCH] feat: add pdf extension to editor

---
 front/components/Editor/Editor.tsx            |   6 +
 .../Editor/Extensions/PDF/PDFBlock.ts         |  35 ++++++
 .../Extensions/PDF/PDFBlockComponent.tsx      |  88 ++++++++++++++
 .../Editor/Toolbar/ToolbarButtons.tsx         |  15 ++-
 front/services/files/documents.ts             |  20 ++++
 src/routers/files.py                          |  27 ++++-
 src/services/files/documents.py               | 112 ++++++++++++++++++
 7 files changed, 297 insertions(+), 6 deletions(-)
 create mode 100644 front/components/Editor/Extensions/PDF/PDFBlock.ts
 create mode 100644 front/components/Editor/Extensions/PDF/PDFBlockComponent.tsx
 create mode 100644 front/services/files/documents.ts
 create mode 100644 src/services/files/documents.py

diff --git a/front/components/Editor/Editor.tsx b/front/components/Editor/Editor.tsx
index d90d5951..da0e0da5 100644
--- a/front/components/Editor/Editor.tsx
+++ b/front/components/Editor/Editor.tsx
@@ -20,6 +20,8 @@ import Youtube from "@tiptap/extension-youtube";
 import VideoBlock from "./Extensions/Video/VideoBlock";
 import { Save } from "lucide-react";
 import MathEquationBlock from "./Extensions/MathEquation/MathEquationBlock";
+import PDFBlockComponent from "./Extensions/PDF/PDFBlockComponent";
+import PDFBlock from "./Extensions/PDF/PDFBlock";
 
 interface Editor {
   content: string;
@@ -59,6 +61,10 @@ function Editor(props: Editor) {
         editable: true,
         lecture: props.lecture,
       }),
+      PDFBlock.configure({
+        editable: true,
+        lecture: props.lecture,
+      }),
       Youtube.configure({
         controls: true,
         modestBranding: true,
diff --git a/front/components/Editor/Extensions/PDF/PDFBlock.ts b/front/components/Editor/Extensions/PDF/PDFBlock.ts
new file mode 100644
index 00000000..d329b79e
--- /dev/null
+++ b/front/components/Editor/Extensions/PDF/PDFBlock.ts
@@ -0,0 +1,35 @@
+import { mergeAttributes, Node } from "@tiptap/core";
+import { ReactNodeViewRenderer } from "@tiptap/react";
+
+import PDFBlockComponent from "./PDFBlockComponent";
+
+export default Node.create({
+  name: "blockPDF",
+  group: "block",
+
+  atom: true,
+
+  addAttributes() {
+    return {
+      fileObject: {
+        default: null,
+      },
+    };
+  },
+
+  parseHTML() {
+    return [
+      {
+        tag: "block-pdf",
+      },
+    ];
+  },
+
+  renderHTML({ HTMLAttributes }) {
+    return ["block-pdf", mergeAttributes(HTMLAttributes), 0];
+  },
+
+  addNodeView() {
+    return ReactNodeViewRenderer(PDFBlockComponent);
+  },
+});
diff --git a/front/components/Editor/Extensions/PDF/PDFBlockComponent.tsx b/front/components/Editor/Extensions/PDF/PDFBlockComponent.tsx
new file mode 100644
index 00000000..57099699
--- /dev/null
+++ b/front/components/Editor/Extensions/PDF/PDFBlockComponent.tsx
@@ -0,0 +1,88 @@
+import { NodeViewWrapper } from "@tiptap/react";
+import React from "react";
+import styled from "styled-components";
+import { AlertCircle, AlertTriangle, FileText, Image, ImagePlus, Info } from "lucide-react";
+import { getPDFFile, uploadNewPDFFile } from "../../../../services/files/documents";
+import { getBackendUrl } from "../../../../services/config";
+
+function PDFBlockComponent(props: any) {
+  const [pdf, setPDF] = React.useState(null);
+  const [isLoading, setIsLoading] = React.useState(false);
+  const [fileObject, setfileObject] = React.useState(props.node.attrs.fileObject);
+
+  const handlePDFChange = (event: React.ChangeEvent<any>) => {
+    setPDF(event.target.files[0]);
+  };
+
+  const handleSubmit = async (e: any) => {
+    e.preventDefault();
+    setIsLoading(true);
+    let object = await uploadNewPDFFile(pdf, props.extension.options.lecture.lecture_id);
+    setIsLoading(false);
+    setfileObject(object);
+    props.updateAttributes({
+      fileObject: object,
+    });
+  };
+
+  return (
+    <NodeViewWrapper className="block-pdf">
+      {!fileObject && (
+        <BlockPDFWrapper contentEditable={props.extension.options.editable}>
+          <div>
+            <FileText color="#e1e0e0" size={50} />
+            <br />
+          </div>
+          <input onChange={handlePDFChange} type="file" name="" id="" />
+          <br />
+          <button onClick={handleSubmit}>Submit</button>
+        </BlockPDFWrapper>
+      )}
+      {fileObject && (
+        <BlockPDF>
+          <iframe
+            src={`${getBackendUrl()}content/uploads/files/documents/${props.extension.options.lecture.lecture_id}/${fileObject.file_id}.${
+              fileObject.file_format
+            }`}
+          />
+        </BlockPDF>
+      )}
+      {isLoading && (
+        <div>
+          <AlertTriangle color="#e1e0e0" size={50} />
+        </div>
+      )}
+    </NodeViewWrapper>
+  );
+}
+
+export default PDFBlockComponent;
+
+const BlockPDFWrapper = styled.div`
+  display: flex;
+  flex-direction: column;
+  background: #f9f9f9;
+  border-radius: 3px;
+  padding: 30px;
+  min-height: 74px;
+  border: ${(props) => (props.contentEditable ? "2px dashed #713f1117" : "none")};
+
+  // center
+  align-items: center;
+  justify-content: center;
+  text-align: center;
+  font-size: 14px;
+`;
+
+const BlockPDF = styled.div`
+  display: flex;
+  flex-direction: column;
+  img {
+    width: 100%;
+    border-radius: 6px;
+    height: 300px;
+    // cover
+    object-fit: cover;
+  }
+`;
+const PDFNotFound = styled.div``;
diff --git a/front/components/Editor/Toolbar/ToolbarButtons.tsx b/front/components/Editor/Toolbar/ToolbarButtons.tsx
index 24c58d98..17af1615 100644
--- a/front/components/Editor/Toolbar/ToolbarButtons.tsx
+++ b/front/components/Editor/Toolbar/ToolbarButtons.tsx
@@ -1,6 +1,6 @@
 import styled from "styled-components";
 import { FontBoldIcon, FontItalicIcon, StrikethroughIcon, ArrowLeftIcon, ArrowRightIcon, OpacityIcon } from "@radix-ui/react-icons";
-import { AlertCircle, AlertTriangle, ImagePlus, Info, Sigma, Video, Youtube } from "lucide-react";
+import { AlertCircle, AlertTriangle, FileText, ImagePlus, Info, Sigma, Video, Youtube } from "lucide-react";
 
 export const ToolbarButtons = ({ editor }: any) => {
   if (!editor) {
@@ -103,6 +103,19 @@ export const ToolbarButtons = ({ editor }: any) => {
       >
         <Sigma size={15} />
       </ToolBtn>
+      <ToolBtn
+        onClick={() =>
+          editor
+            .chain()
+            .focus()
+            .insertContent({
+              type: "blockPDF",
+            })
+            .run()
+        }
+      >
+        <FileText size={15} />
+      </ToolBtn>
     </ToolButtonsWrapper>
   );
 };
diff --git a/front/services/files/documents.ts b/front/services/files/documents.ts
new file mode 100644
index 00000000..1faa3546
--- /dev/null
+++ b/front/services/files/documents.ts
@@ -0,0 +1,20 @@
+import { getAPIUrl } from "@services/config";
+import { RequestBody, RequestBodyForm } from "@services/utils/requests";
+
+export async function uploadNewPDFFile(file: any, lecture_id: string) {
+  // Send file thumbnail as form data
+  const formData = new FormData();
+  formData.append("file_object", file);
+  formData.append("lecture_id", lecture_id);
+
+  return fetch(`${getAPIUrl()}files/document`, RequestBodyForm("POST", formData))
+    .then((result) => result.json())
+    .catch((error) => console.log("error", error));
+}
+
+export async function getPDFFile(file_id: string) {
+  // todo : add course id to url
+  return fetch(`${getAPIUrl()}files/document?file_id=${file_id}`, RequestBody("GET", null))
+    .then((result) => result.json())
+    .catch((error) => console.log("error", error));
+}
diff --git a/src/routers/files.py b/src/routers/files.py
index d2cfb89d..d520c196 100644
--- a/src/routers/files.py
+++ b/src/routers/files.py
@@ -1,6 +1,7 @@
 from fastapi import APIRouter, Depends, UploadFile, Form, Request
 from src.dependencies.auth import get_current_user
 from fastapi import HTTPException, status, UploadFile
+from src.services.files.documents import create_document_file, get_document_file
 
 from src.services.files.pictures import create_picture_file, get_picture_file
 from src.services.files.videos import create_video_file, get_video_file
@@ -10,15 +11,15 @@ router = APIRouter()
 
 
 @router.post("/picture")
-async def api_create_picture_file(request: Request,file_object: UploadFile, lecture_id: str = Form(),  current_user: PublicUser = Depends(get_current_user)):
+async def api_create_picture_file(request: Request, file_object: UploadFile, lecture_id: str = Form(),  current_user: PublicUser = Depends(get_current_user)):
     """
     Create new picture file
     """
-    return await create_picture_file(request,file_object, lecture_id)
+    return await create_picture_file(request, file_object, lecture_id)
 
 
 @router.post("/video")
-async def api_create_video_file(request: Request,file_object: UploadFile,lecture_id: str = Form(), current_user: PublicUser = Depends(get_current_user)):
+async def api_create_video_file(request: Request, file_object: UploadFile, lecture_id: str = Form(), current_user: PublicUser = Depends(get_current_user)):
     """
     Create new video file
     """
@@ -26,7 +27,7 @@ async def api_create_video_file(request: Request,file_object: UploadFile,lecture
 
 
 @router.get("/picture")
-async def api_get_picture_file(request: Request,file_id: str, current_user: PublicUser = Depends(get_current_user)):
+async def api_get_picture_file(request: Request, file_id: str, current_user: PublicUser = Depends(get_current_user)):
     """
     Get picture file
     """
@@ -34,8 +35,24 @@ async def api_get_picture_file(request: Request,file_id: str, current_user: Publ
 
 
 @router.get("/video")
-async def api_get_video_file(request: Request,file_id: str, current_user: PublicUser = Depends(get_current_user)):
+async def api_get_video_file(request: Request, file_id: str, current_user: PublicUser = Depends(get_current_user)):
     """
     Get video file
     """
     return await get_video_file(request, file_id, current_user)
+
+
+@router.get("/document")
+async def api_get_document_file(request: Request, file_id: str, current_user: PublicUser = Depends(get_current_user)):
+    """
+    Get document file
+    """
+    return await get_document_file(request, file_id, current_user)
+
+
+@router.post("/document")
+async def api_create_document_file(request: Request, file_object: UploadFile, lecture_id: str = Form(), current_user: PublicUser = Depends(get_current_user)):
+    """
+    Create new document file
+    """
+    return await create_document_file(request, file_object, lecture_id)
diff --git a/src/services/files/documents.py b/src/services/files/documents.py
new file mode 100644
index 00000000..9fbdf718
--- /dev/null
+++ b/src/services/files/documents.py
@@ -0,0 +1,112 @@
+from uuid import uuid4
+from pydantic import BaseModel
+from fastapi import HTTPException, status, UploadFile, Request
+from fastapi.responses import StreamingResponse
+import os
+
+from src.services.users import PublicUser
+
+
+class DocumentFile(BaseModel):
+    file_id: str
+    file_format: str
+    file_name: str
+    file_size: int
+    file_type: str
+    lecture_id: str
+
+
+async def create_document_file(request: Request, document_file: UploadFile, lecture_id: str):
+    documents = request.app.db["files"]
+
+    # generate file_id
+    file_id = str(f"file_{uuid4()}")
+
+    # get file format
+    file_format = document_file.filename.split(".")[-1]
+
+    # validate file format
+    if file_format not in ["pdf"]:
+        raise HTTPException(
+            status_code=status.HTTP_409_CONFLICT, detail="Document file format not supported")
+
+    # create file
+    file = await document_file.read()
+
+    # get file size
+    file_size = len(file)
+
+    # get file type
+    file_type = document_file.content_type
+
+    # get file name
+    file_name = document_file.filename
+
+    # create file object
+    uploadable_file = DocumentFile(
+        file_id=file_id,
+        file_format=file_format,
+        file_name=file_name,
+        file_size=file_size,
+        file_type=file_type,
+        lecture_id=lecture_id
+    )
+    # TODO : this is probably not working as intended
+    # create folder for lecture
+    if not os.path.exists(f"content/uploads/files/documents/{lecture_id}"):
+        os.mkdir(f"content/uploads/files/documents/{lecture_id}")
+
+    # upload file to server
+    with open(f"content/uploads/files/documents/{lecture_id}/{file_id}.{file_format}", 'wb') as f:
+        f.write(file)
+        f.close()
+
+    # insert file object into database
+    document_file_in_db = documents.insert_one(uploadable_file.dict())
+
+    if not document_file_in_db:
+        raise HTTPException(
+            status_code=status.HTTP_409_CONFLICT, detail="Document file could not be created")
+
+    return uploadable_file
+
+
+async def get_document_object(request: Request, file_id: str):
+    documents = request.app.db["files"]
+
+    document_file = documents.find_one({"file_id": file_id})
+
+    if document_file:
+        document_file = DocumentFile(**document_file)
+        return document_file
+
+    else:
+        raise HTTPException(
+            status_code=status.HTTP_409_CONFLICT, detail="Document file does not exist")
+
+
+async def get_document_file(request: Request, file_id: str, current_user: PublicUser):
+    documents = request.app.db["files"]
+
+    document_file = documents.find_one({"file_id": file_id})
+
+    # TODO : check if user has access to file
+
+    if document_file:
+
+        # check media type
+        if document_file.format not in ["jpg", "jpeg", "png", "gif"]:
+            raise HTTPException(
+                status_code=status.HTTP_409_CONFLICT, detail="Document file format not supported")
+
+        # stream file
+        document_file = DocumentFile(**document_file)
+        file_format = document_file.file_format
+        lecture_id = document_file.lecture_id
+        file = open(
+            f"content/uploads/files/documents/{lecture_id}/{file_id}.{file_format}", 'rb')
+        return StreamingResponse(file, media_type=document_file.file_type)
+
+    else:
+        raise HTTPException(
+            status_code=status.HTTP_409_CONFLICT, detail="Document file does not exist")