"""Website for uploading files, creating .HTMLs, and thumbnails.
This was created for Discord. You can use this to embed videos in Discord.
"""
import sys
from datetime import datetime
from pathlib import Path
from typing import Dict
import ffmpeg
from dhooks import Webhook
from fastapi import FastAPI, File, UploadFile
from fastapi.responses import HTMLResponse
from settings import Settings
hook = Webhook(Settings.webhook_url)
app = FastAPI(
title="discord-nice-embed",
description=Settings.description,
version="0.0.1",
contact={
"name": "Joakim Hellsén",
"url": "https://github.com/TheLovinator1",
"email": "tlovinator@gmail.com",
},
license_info={
"name": "GPL-3.0",
"url": "https://www.gnu.org/licenses/gpl-3.0.txt",
},
)
@app.post("/uploadfiles/")
async def upload_file(file: UploadFile = File(...)) -> Dict[str, str]:
"""Page for uploading files.
It starts by checking if the file is a video.
Then it checks if the directory exists, if not it creates it.
Then it saves the file to the /video folder.
Then it finds the resolution of the video and creates a thumbnail.
Then it generates a HTML file for the video. This is what you will send to Discord.
Finally it returns a URL for the HTML file to the user.
If something goes wrong, we will send a message to Discord.
Args:
file (UploadFile): Our uploaded file. Defaults to File(...).
Returns:
Dict[str, str]: Returns a dict with the filename or a link to the .html if it was a video.
"""
# TODO: Add syntax highlighting for text.
try:
# Make custom html for video files.
if file.content_type.startswith("video/"):
# Create folder if it doesn't exist.
Path(f"{Settings.upload_folder}/video").mkdir(parents=True, exist_ok=True)
# Save file to disk.
with open(f"{Settings.upload_folder}/video/{file.filename}", "wb+") as file_object:
file_object.write(file.file.read())
file_url = f"{Settings.domain}/video/{file.filename}"
file_location = f"{Settings.upload_folder}/video/{file.filename}"
height, width = find_video_resolution(file_location)
screenshot_url = make_thumbnail_from_video(file_location, file.filename)
html_url = generate_html(
filename=file.filename, url=file_url, width=width, height=height, screenshot=screenshot_url
)
return {"html_url": f"{html_url}"}
# Save file to disk.
with open(f"{Settings.upload_folder}/{file.filename}", "wb+") as file_object:
file_object.write(file.file.read())
hook.send(f"{Settings.domain}/{file.filename} was uploaded.")
return {"html_url": f"{Settings.domain}/{file.filename}"}
except Exception as e:
# TODO: Change response code to 400.
hook.send(f"Something went wrong for {Settings.domain}/{file.filename}:\n{e}")
return {"error": f"Something went wrong: {e}"}
@app.get("/", response_class=HTMLResponse)
async def main():
"""Our index view.
You can upload files here.
Returns:
HTMLResponse: Returns HTML for site.
"""
return """
discord-nice-embed
Swagger UI - API documentation
ReDoc - Alternative API documentation
"""
def generate_html(url: str, width: int, height: int, screenshot: str, filename: str) -> str:
"""Generate HTML for video files.
This is what we will send to other people on Discord.
You can remove the .html with your web server so the link will look normal.
For example, with nginx, you can do this(note the $uri.html):
location / {
try_files $uri $uri/ $uri.html;
}
Args:
url (str): URL for the video. This is accessible from the browser.
width (int): This is the width of the video.
height (int): This is the height of the video.
screenshot (str): URL for screenshot. This is what you will see in Discord.
filename (str): Original video filename. We will append .html to the filename.
Returns:
str: [description]
"""
video_html = f"""
"""
html_url = f"{Settings.domain}/{filename}"
# Take the filename and append .html to it.
filename += ".html"
with open(f"{Settings.upload_folder}/{filename}", "w", encoding="utf-8") as file:
file.write(video_html)
return html_url
def find_video_resolution(path_to_video: str) -> tuple[int, int]:
"""Find video resolution.
Args:
path_to_video (str): Path to video file.
Returns:
tuple[int, int]: Returns height and width.
"""
probe = ffmpeg.probe(path_to_video)
video_stream = next(
(stream for stream in probe["streams"] if stream["codec_type"] == "video"),
None,
)
if video_stream is None:
print("No video stream found", file=sys.stderr)
sys.exit(1)
width = int(video_stream["width"])
height = int(video_stream["height"])
return height, width
def make_thumbnail_from_video(path_video: str, file_filename: str) -> str:
"""Make thumbnail for Discord. This is a screenshot of the video.
Args:
path_video (str): Path where video file is stored.
file_filename (str): File name for URL.
Returns:
str: Returns thumbnail filename.
"""
(
ffmpeg.input(path_video, ss="1") # Take a screenshot at 1 second.
.output(f"{Settings.upload_folder}/{file_filename}.jpg", vframes=1) # Output to file.
.overwrite_output() # Overwrite output.
.run() # Run.
)
# Return URL for thumbnail.
return f"{Settings.domain}/{file_filename}.jpg"