Merge pull request #321 from docat-org/feat/62-hidden-versions

Feat(docat): Add API to hide project versions
docat-org · Oct 26, 2022 · 5160e11 · 5160e11
2 parents 7e1f306 + ba76e2f
commit 5160e11
Show file tree

Hide file tree

Showing 3 changed files with 487 additions and 2 deletions.
diff --git a/docat/docat/app.py b/docat/docat/app.py
@@ -75,7 +75,21 @@ class ProjectDetailResponse(BaseModel):
 def get_projects():
     if not DOCAT_UPLOAD_FOLDER.exists():
         return ProjectsResponse(projects=[])
-    return ProjectsResponse(projects=[str(x.relative_to(DOCAT_UPLOAD_FOLDER)) for x in DOCAT_UPLOAD_FOLDER.iterdir() if x.is_dir()])
+
+    def has_not_hidden_versions(project):
+        path = DOCAT_UPLOAD_FOLDER / project
+        return any(
+            (path / version).is_dir() and not (path / version / ".hidden").exists() for version in (DOCAT_UPLOAD_FOLDER / project).iterdir()
+        )
+
+    return ProjectsResponse(
+        projects=list(
+            filter(
+                has_not_hidden_versions,
+                [str(project.relative_to(DOCAT_UPLOAD_FOLDER)) for project in DOCAT_UPLOAD_FOLDER.iterdir() if project.is_dir()],
+            )
+        )
+    )
 
 
 @app.get(
@@ -106,7 +120,7 @@ def get_project(project):
                     tags=[str(t.relative_to(docs_folder)) for t in tags if t.resolve() == x],
                 )
                 for x in docs_folder.iterdir()
-                if x.is_dir() and not x.is_symlink()
+                if x.is_dir() and not x.is_symlink() and not (docs_folder / x.name / ".hidden").exists()
             ],
             key=lambda k: k.name,
             reverse=True,
@@ -155,6 +169,77 @@ def upload_icon(
     return ApiResponse(message="Icon successfully uploaded")
 
 
+@app.post("/api/{project}/{version}/hide", response_model=ApiResponse, status_code=status.HTTP_200_OK)
+@app.post("/api/{project}/{version}/hide/", response_model=ApiResponse, status_code=status.HTTP_200_OK)
+def hide_version(
+    project: str,
+    version: str,
+    response: Response,
+    docat_api_key: Optional[str] = Header(None),
+    db: TinyDB = Depends(get_db),
+):
+    project_base_path = DOCAT_UPLOAD_FOLDER / project
+    version_path = project_base_path / version
+    hidden_file = version_path / ".hidden"
+
+    if not project_base_path.exists():
+        response.status_code = status.HTTP_404_NOT_FOUND
+        return ApiResponse(message=f"Project {project} not found")
+
+    if not version_path.exists():
+        response.status_code = status.HTTP_404_NOT_FOUND
+        return ApiResponse(message=f"Version {version} not found")
+
+    if hidden_file.exists():
+        response.status_code = status.HTTP_400_BAD_REQUEST
+        return ApiResponse(message=f"Version {version} is already hidden")
+
+    token_status = check_token_for_project(db, docat_api_key, project)
+    if not token_status.valid:
+        response.status_code = status.HTTP_401_UNAUTHORIZED
+        return ApiResponse(message=token_status.reason)
+
+    with open(hidden_file, "w") as f:
+        f.close()
+
+    return ApiResponse(message=f"Version {version} is now hidden")
+
+
+@app.post("/api/{project}/{version}/show", response_model=ApiResponse, status_code=status.HTTP_200_OK)
+@app.post("/api/{project}/{version}/show/", response_model=ApiResponse, status_code=status.HTTP_200_OK)
+def show_version(
+    project: str,
+    version: str,
+    response: Response,
+    docat_api_key: Optional[str] = Header(None),
+    db: TinyDB = Depends(get_db),
+):
+    project_base_path = DOCAT_UPLOAD_FOLDER / project
+    version_path = project_base_path / version
+    hidden_file = version_path / ".hidden"
+
+    if not project_base_path.exists():
+        response.status_code = status.HTTP_404_NOT_FOUND
+        return ApiResponse(message=f"Project {project} not found")
+
+    if not version_path.exists():
+        response.status_code = status.HTTP_404_NOT_FOUND
+        return ApiResponse(message=f"Version {version} not found")
+
+    if not hidden_file.exists():
+        response.status_code = status.HTTP_400_BAD_REQUEST
+        return ApiResponse(message=f"Version {version} is not hidden")
+
+    token_status = check_token_for_project(db, docat_api_key, project)
+    if not token_status.valid:
+        response.status_code = status.HTTP_401_UNAUTHORIZED
+        return ApiResponse(message=token_status.reason)
+
+    os.remove(hidden_file)
+
+    return ApiResponse(message=f"Version {version} is now shown")
+
+
 @app.post("/api/{project}/{version}", response_model=ApiResponse, status_code=status.HTTP_201_CREATED)
 @app.post("/api/{project}/{version}/", response_model=ApiResponse, status_code=status.HTTP_201_CREATED)
 def upload(

diff --git a/docat/tests/conftest.py b/docat/tests/conftest.py
@@ -20,6 +20,11 @@ def client():
     temp_dir.cleanup()
 
 
+@pytest.fixture
+def upload_folder_path():
+    return docat.DOCAT_UPLOAD_FOLDER
+
+
 @pytest.fixture
 def client_with_claimed_project(client):
     table = docat.db.table("claims")