main.py

import datetime
import io
import json
import logging
import os
import sys
import threading

import apscheduler.schedulers.background
import bs4
import colorama
import flask
import flask_cors
import googleapiclient
import requests

import src.functions.config
import src.functions.credentials
import src.functions.metadata
import src.functions.tests

colorama.init()
print(
    "====================================================\n\033[96m               libDrive - v1.4.7\033[94m\n                   @eliasbenb\033[0m\n====================================================\n"
)

print("\033[32mREADING CONFIG...\033[0m")
if os.getenv("LIBDRIVE_CONFIG"):
    config_str = os.getenv("LIBDRIVE_CONFIG")
    with open("config.json", "w+") as w:
        json.dump(obj=json.loads(config_str), fp=w, sort_keys=True, indent=4)
config = src.functions.config.readConfig()
print("DONE.\n")

print("\033[32mREADING METADATA...\033[0m")
metadata = src.functions.metadata.readMetadata(config)
if os.getenv("LIBDRIVE_CLOUD") and config.get("refresh_token"):
    config, drive = src.functions.credentials.refreshCredentials(config)
    params = {
        "supportsAllDrives": True,
        "includeItemsFromAllDrives": True,
        "fields": "files(id,name)",
        "q": "'%s' in parents and trashed = false and mimeType = 'application/json'"
        % (os.getenv("LIBDRIVE_CLOUD")),
    }
    files = drive.files().list(**params).execute()["files"]
    config_file = next((i for i in files if i["name"] == "config.json"), None)
    metadata_file = next((i for i in files if i["name"] == "metadata.json"), None)
    if config_file:
        request = drive.files().get_media(fileId=config_file["id"])
        fh = io.BytesIO()
        downloader = googleapiclient.http.MediaIoBaseDownload(fh, request)
        done = False
        while done is False:
            status, done = downloader.next_chunk()
        config = json.loads(fh.getvalue())
        config, drive = src.functions.credentials.refreshCredentials(config)
        src.functions.config.updateConfig(config)
    if metadata_file:
        request = drive.files().get_media(fileId=metadata_file["id"])
        fh = io.BytesIO()
        downloader = googleapiclient.http.MediaIoBaseDownload(fh, request)
        done = False
        while done is False:
            status, done = downloader.next_chunk()
        metadata = json.loads(fh.getvalue())
        with open("metadata.json", "w+") as w:
            json.dump(metadata, w)
print("DONE.\n")

if not config.get("account_list"):
    config["account_list"] = []
if config.get("account_list") == [] and config.get("signup") == False:
    config["auth"] = False
if not config.get("auth"):
    config["auth"] = False
if not config.get("build_interval"):
    config["build_interval"] = 360
if not config.get("build_type"):
    config["build_type"] = "hybrid"
if not config.get("category_list"):
    config["category_list"] = []
if not config.get("cloudflare"):
    config["cloudflare"] = ""
if not config.get("prefer_mkv"):
    config["prefer_mkv"] = False
if not config.get("prefer_mp4"):
    config["prefer_mp4"] = True
if not config.get("service_accounts"):
    config["service_accounts"] = []
if not config.get("signup"):
    config["signup"] = False
if not config.get("subtitles"):
    config["subtitles"] = False
if not config.get("transcoded"):
    config["transcoded"] = False
if not config.get("ui_config"):
    config["ui_config"] = {}

with open("config.json", "w+") as w:
    json.dump(obj=config, fp=w, sort_keys=True, indent=4)

print("\033[32mTESTING YOUR CONFIG...\033[0m")
src.functions.tests.tmdb_test(config)
src.functions.tests.category_list_test(config)
src.functions.tests.account_list_test(config)
src.functions.tests.cloudflare_test(config)
print("DONE.\n")


def threaded_metadata():
    for thread in threading.enumerate():
        if thread.name == "metadata_thread":
            print("DONE.\n")
            return (
                {
                    "code": 500,
                    "content": None,
                    "message": "Drive is already building metadata, please wait.",
                    "success": False,
                },
                500,
            )
    config = src.functions.config.readConfig()
    if len(config.get("category_list")) > 0:
        metadata_thread = threading.Thread(
            target=src.functions.metadata.writeMetadata,
            args=(config,),
            daemon=True,
            name="metadata_thread",
        )
        metadata_thread.start()
    else:
        with open("./metadata.json", "w+") as w:
            w.write(json.dumps([]))
    return (
        {
            "code": 200,
            "content": None,
            "message": "Drive is building your new metadata.",
            "success": True,
        },
        200,
    )


def create_app():
    if os.path.exists("./build"):
        LIBDRIVE_DEBUG = os.getenv("LIBDRIVE_DEBUG")
        if LIBDRIVE_DEBUG:
            if LIBDRIVE_DEBUG.lower() == "true":
                LIBDRIVE_DEBUG = True
            else:
                LIBDRIVE_DEBUG = False
        else:
            LIBDRIVE_DEBUG = False
        r = open("./build/index.html", "r")
        soup = bs4.BeautifulSoup(r.read(), features="html.parser")
        if config.get("ui_config", {}).get("icon"):
            try:
                soup.find("meta", {"id": "@ld-meta-og-image"})["content"] = config.get(
                    "ui_config", {}
                ).get("icon")
            except:
                pass
            try:
                soup.find("link", {"id": "@ld-link-icon"})["href"] = config.get(
                    "ui_config", {}
                ).get("icon")
            except:
                pass
        else:
            try:
                soup.find("meta", {"id": "@ld-meta-og-image"})[
                    "content"
                ] = "/images/icons/icon-512x512.png"
            except:
                pass
            try:
                soup.find("link", {"id": "@ld-link-icon"})["href"] = "/favicon.ico"
            except:
                pass
        if config.get("ui_config", {}).get("title"):
            try:
                soup.find("meta", {"id": "@ld-meta-og-title"})["content"] = config.get(
                    "ui_config", {}
                ).get("title")
            except:
                pass
            try:
                soup.find("meta", {"id": "@ld-meta-og-site_name"})[
                    "content"
                ] = config.get("ui_config", {}).get("title")
            except:
                pass
            try:
                soup.find("title", {"id": "@ld-title"}).string = config.get(
                    "ui_config", {}
                ).get("title")
            except:
                pass
        else:
            try:
                soup.find("meta", {"id": "@ld-meta-og-title"})["content"] = "libDrive"
            except:
                pass
            try:
                soup.find("meta", {"id": "@ld-meta-og-site_name"})[
                    "content"
                ] = "libDrive"
            except:
                pass
            try:
                soup.find("title", {"id": "@ld-title"}).string = "libDrive"
            except:
                pass
        if (
            config.get("arcio")
            and config.get("arcio") != ""
            and LIBDRIVE_DEBUG == False
        ):
            req = requests.get("https://arc.io/arc-sw.js")
            with open("./build/arc-sw.js", "wb") as wb:
                wb.write(req.content)
            code = config.get("arcio")
            if code == "dev":
                code = "tUUqUjhw"
            soup.find("script", {"id": "@ld-script-arcio"})[
                "src"
            ] = "//arc.io/widget.min.js#%s" % (code)
        else:
            if os.path.exists("./build/arc-sw.js"):
                os.remove("./build/arc-sw.js")
            soup.find("script", {"id": "@ld-script-arcio"})["src"] = ""
        with open("./build/index.html", "w+") as w:
            w.write(str(soup))
        r.close()

    app = flask.Flask(__name__, static_folder="build")

    build_interval = config.get("build_interval")
    if not build_interval:
        build_interval = 360
    if build_interval != 0:
        print("\033[32mCREATING CRON JOB...\033[0m")
        sched = apscheduler.schedulers.background.BackgroundScheduler(daemon=True)
        sched.add_job(
            threaded_metadata,
            "interval",
            minutes=build_interval,
        )
        sched.start()
        print("DONE.\n")

    config_categories = [d["id"] for d in config["category_list"]]
    metadata_categories = [d["id"] for d in metadata]
    if len(metadata) > 0 and sorted(config_categories) == sorted(metadata_categories):
        if build_interval == 0:
            return app
        elif datetime.datetime.utcnow() <= datetime.datetime.strptime(
            metadata[-1]["buildTime"], "%Y-%m-%d %H:%M:%S.%f"
        ) + datetime.timedelta(minutes=build_interval):
            return app
        else:
            threaded_metadata()
    else:
        threaded_metadata()

    return app


app = create_app()
flask_cors.CORS(app)
app.secret_key = config.get("secret_key")

from src.routes.auth import authBP
from src.routes.config import configBP
from src.routes.debug import debugBP
from src.routes.download import downloadBP
from src.routes.environment import environmentBP
from src.routes.image import imageBP
from src.routes.metadata import metadataBP
from src.routes.ping import pingBP
from src.routes.rebuild import rebuildBP
from src.routes.redirectdownload import redirectdownloadBP
from src.routes.restart import restartBP
from src.routes.signup import signupBP
from src.routes.streammap import streammapBP
from src.routes.subtitledownload import subtitledownloadBP
from src.routes.trailer import trailerBP

app.register_blueprint(authBP)
app.register_blueprint(configBP)
app.register_blueprint(debugBP)
app.register_blueprint(downloadBP)
app.register_blueprint(environmentBP)
app.register_blueprint(imageBP)
app.register_blueprint(metadataBP)
app.register_blueprint(pingBP)
app.register_blueprint(rebuildBP)
app.register_blueprint(redirectdownloadBP)
app.register_blueprint(restartBP)
app.register_blueprint(signupBP)
app.register_blueprint(streammapBP)
app.register_blueprint(subtitledownloadBP)
app.register_blueprint(trailerBP)


@app.route("/", defaults={"path": ""})
@app.route("/<path:path>")
async def serve(path):
    if (path != "") and os.path.exists("%s/%s" % (app.static_folder, path)):
        return flask.send_from_directory(app.static_folder, path)
    else:
        return flask.send_from_directory(app.static_folder, "index.html")


if __name__ == "__main__":
    print("\033[32mSERVING SERVER...\033[0m")
    LIBDRIVE_DEBUG = os.getenv("LIBDRIVE_DEBUG")
    if LIBDRIVE_DEBUG:
        if LIBDRIVE_DEBUG.lower() == "true":
            LIBDRIVE_DEBUG = True
        else:
            LIBDRIVE_DEBUG = False
    else:
        LIBDRIVE_DEBUG = False
    print("DONE.\n")
    app.run(
        host="0.0.0.0",
        port=31145,
        threaded=True,
        debug=LIBDRIVE_DEBUG,
    )
else:
    print("\033[32mINITIALIZING LOGGER...\033[0m")
    if not os.path.exists("./logs"):
        os.mkdir("./logs")
    logs_path = os.path.abspath("./logs")
    logs_max_files = 5

    def sorted_ls(path):
        mtime = lambda f: os.stat(os.path.join(path, f)).st_mtime
        return list(sorted(os.listdir(path), key=mtime))

    del_list = sorted_ls(logs_path)[0 : (len(sorted_ls(logs_path)) - logs_max_files)]
    for del_file in del_list:
        try:
            os.remove(os.path.join(logs_path, del_file))
        except:
            pass
    logging.getLogger("googleapiclient").setLevel(logging.WARNING)
    logging.getLogger("oauth2client").setLevel(logging.WARNING)
    logging.getLogger("waitress").setLevel(logging.INFO)
    logging.basicConfig(
        filename="./logs/%s.log"
        % (datetime.datetime.utcnow().strftime("%Y%m%d-%H%M%S")),
        level=logging.INFO,
    )
    console_logger = logging.getLogger()
    console_logger.setLevel(logging.INFO)
    console_handler = logging.StreamHandler(sys.stdout)
    console_handler.setLevel(logging.INFO)
    console_logger.addHandler(console_handler)
    print("DONE.\n")