Major overhaul + Bare bones web UI

1 files changed, 143 insertions, 0 deletions

diff --git a/api/routes/links_route.py b/api/routes/links_route.py
new file mode 100644
index 0000000..fd0a77a
--- /dev/null
+++ b/api/routes/links_route.py
@@ -0,0 +1,143 @@
+from fastapi import APIRouter, status, Path, Depends, Security, Request
+from fastapi.exception_handlers import HTTPException
+from typing import Annotated
+import string
+import random
+import datetime
+import validators
+
+from api.util.db_dependency import get_db
+from api.util.check_api_key import check_api_key
+from models import Link, Record
+from api.schemas.links_schemas import URLSchema
+
+
+router = APIRouter(prefix="/links", tags=["links"])
+
+
+@router.get("/", summary="Get all of the links associated with your account")
+async def get_links(
+    db=Depends(get_db),
+    api_key: str = Security(check_api_key),
+):
+    links = db.query(Link).filter(Link.owner == api_key["owner"]).all()
+    if not links:
+        raise HTTPException(
+            status_code=status.HTTP_404_NOT_FOUND, detail="No links found"
+        )
+    return links
+
+
+@router.post("/", summary="Create a new link")
+async def create_link(
+    url: URLSchema,
+    db=Depends(get_db),
+    api_key: str = Security(check_api_key),
+):
+    # Check if the URL is valid
+    if not validators.url(url.url):
+        raise HTTPException(
+            status_code=status.HTTP_422_UNPROCESSABLE_ENTITY, detail="Invalid URL"
+        )
+    # Create the new link and add it to the database
+    while True:
+        try:
+            link_path = "".join(
+                random.choices(string.ascii_uppercase + "1234567890", k=5)
+            )
+            new_link = Link(
+                link=link_path,
+                owner=api_key["owner"],
+                redirect_link=url.url,
+                expire_date=datetime.datetime.now() + datetime.timedelta(days=30),
+            )
+            db.add(new_link)
+            db.commit()
+            break
+        except:
+            continue
+
+    return {
+        "response": "Link successfully created",
+        "expire_date": new_link.expire_date,
+        "link": new_link.link,
+    }
+
+
+@router.delete("/{link}", summary="Delete a link")
+async def delete_link(
+    link: Annotated[str, Path(title="Link to delete")],
+    db=Depends(get_db),
+    api_key: str = Security(check_api_key),
+):
+    # Get the link and check the owner
+    link = db.query(Link).filter(Link.link == link).first()
+    if not link:
+        raise HTTPException(
+            status_code=status.HTTP_404_NOT_FOUND, detail="Link not found"
+        )
+    if link.owner != api_key["owner"]:
+        raise HTTPException(
+            status_code=status.HTTP_401_UNAUTHORIZED,
+            detail="Link not associated with your account",
+        )
+
+    # Get and delete all records associated with the link
+    records = db.query(Record).filter(Record.link == link.link).all()
+    for record in records:
+        db.delete(record)
+    # Delete the link
+    db.delete(link)
+    db.commit()
+
+    return {"response": "Link successfully deleted", "link": link.link}
+
+
+@router.get("/{link}/records", summary="Get all of the IP log records associated with a link")
+async def get_link_records(
+    link: Annotated[str, Path(title="Link to get records for")],
+    db=Depends(get_db),
+    api_key: str = Security(check_api_key),
+):
+    # Get the link and check the owner
+    link = db.query(Link).filter(Link.link == link).first()
+    if not link:
+        raise HTTPException(
+            status_code=status.HTTP_404_NOT_FOUND, detail="Link not found"
+        )
+    if link.owner != api_key["owner"]:
+        raise HTTPException(
+            status_code=status.HTTP_401_UNAUTHORIZED,
+            detail="Link not associated with your account",
+        )
+
+    # Get and return all of the records associated with the link
+    records = db.query(Record).filter(Record.link == link.link).all()
+    return records
+
+
+@router.delete("/{link}/records", summary="Delete all of the IP log records associated with a link")
+async def delete_link_records(
+    link: Annotated[str, Path(title="Link to delete records for")],
+    db=Depends(get_db),
+    api_key: str = Security(check_api_key),
+):
+    # Get the link and check the owner
+    link = db.query(Link).filter(Link.link == link).first()
+    if not link:
+        raise HTTPException(
+            status_code=status.HTTP_404_NOT_FOUND, detail="Link not found"
+        )
+    if link.owner != api_key["owner"]:
+        raise HTTPException(
+            status_code=status.HTTP_401_UNAUTHORIZED,
+            detail="Link not associated with your account",
+        )
+
+    # Get all of the records associated with the link and delete them
+    records = db.query(Record).filter(Record.link == link.link).all()
+    for record in records:
+        db.delete(record)
+    db.commit()
+
+    return {"response": "Records successfully deleted", "link": link.link}