quasarr 1.20.6__py3-none-any.whl

quasarr/storage/setup.py

@@ -0,0 +1,458 @@
+# -*- coding: utf-8 -*-
+# Quasarr
+# Project by https://github.com/rix1337
+
+import os
+import sys
+
+import requests
+from bottle import Bottle, request
+
+import quasarr
+import quasarr.providers.html_images as images
+import quasarr.providers.sessions.al
+import quasarr.providers.sessions.dd
+import quasarr.providers.sessions.nx
+from quasarr.providers.html_templates import render_button, render_form, render_success, render_fail
+from quasarr.providers.log import info
+from quasarr.providers.shared_state import extract_valid_hostname
+from quasarr.providers.web_server import Server
+from quasarr.storage.config import Config
+
+
+def path_config(shared_state):
+    app = Bottle()
+
+    current_path = os.path.dirname(os.path.abspath(sys.argv[0]))
+
+    @app.get('/')
+    def config_form():
+        config_form_html = f'''
+            <form action="/api/config" method="post">
+                <label for="config_path">Path</label>
+                <input type="text" id="config_path" name="config_path" placeholder="{current_path}"><br>
+                {render_button("Save", "primary", {"type": "submit"})}
+            </form>
+            '''
+        return render_form("Press 'Save' to set desired path for configuration",
+                           config_form_html)
+
+    def set_config_path(config_path):
+        config_path_file = "Quasarr.conf"
+
+        if not config_path:
+            config_path = current_path
+
+        config_path = config_path.replace("\\", "/")
+        config_path = config_path[:-1] if config_path.endswith('/') else config_path
+
+        if not os.path.exists(config_path):
+            os.makedirs(config_path)
+
+        with open(config_path_file, "w") as f:
+            f.write(config_path)
+
+        return config_path
+
+    @app.post("/api/config")
+    def set_config():
+        config_path = request.forms.get("config_path")
+        config_path = set_config_path(config_path)
+        quasarr.providers.web_server.temp_server_success = True
+        return render_success(f'Config path set to: "{config_path}"',
+                              5)
+
+    info(f'Starting web server for config at: "{shared_state.values['internal_address']}".')
+    info("Please set desired config path there!")
+    return Server(app, listen='0.0.0.0', port=shared_state.values['port']).serve_temporarily()
+
+
+def hostname_form_html(shared_state, message):
+    hostname_fields = '''
+    <label for="{id}" style="display:inline-flex; align-items:center; gap:4px;">{label}{img_html}</label>
+    <input type="text" id="{id}" name="{id}" placeholder="example.com" autocorrect="off" autocomplete="off" value="{value}"><br>
+    '''
+
+    field_html = []
+    hostnames = Config('Hostnames')  # Load once outside the loop
+    for label in shared_state.values["sites"]:
+        field_id = label.lower()
+        img_html = ''
+        try:
+            img_data = getattr(images, field_id)
+            if img_data:
+                img_html = f' <img src="{img_data}" width="16" height="16" style="filter: blur(2px);" alt="{label} icon">'
+        except AttributeError:
+            pass
+
+        # Get the current value (if any and non-empty)
+        current_value = hostnames.get(field_id)
+        if not current_value:
+            current_value = ''  # Ensure it's empty if None or ""
+
+        field_html.append(hostname_fields.format(
+            id=field_id,
+            label=label,
+            img_html=img_html,
+            value=current_value
+        ))
+
+    hostname_form_content = "".join(field_html)
+    button_html = render_button("Save", "primary", {"type": "submit"})
+
+    template = """
+<div id="message" style="margin-bottom:0.5em;">{message}</div>
+<div id="error-msg" style="color:red; margin-bottom:1em;"></div>
+
+<form action="/api/hostnames" method="post" onsubmit="return validateHostnames(this)">
+    {hostname_form_content}
+    {button}
+</form>
+
+<script>
+  function validateHostnames(form) {{
+    var errorDiv = document.getElementById('error-msg');
+    errorDiv.textContent = '';
+
+    var inputs = form.querySelectorAll('input[type="text"]');
+    for (var i = 0; i < inputs.length; i++) {{
+      if (inputs[i].value.trim() !== '') {{
+        return true;
+      }}
+    }}
+
+    errorDiv.textContent = 'Please fill in at least one hostname!';
+    inputs[0].focus();
+    return false;
+  }}
+</script>
+"""
+    return template.format(
+        message=message,
+        hostname_form_content=hostname_form_content,
+        button=button_html
+    )
+
+
+def save_hostnames(shared_state, timeout=5, first_run=True):
+    hostnames = Config('Hostnames')
+
+    # Collect submitted hostnames, validate, and track errors
+    valid_domains = {}
+    errors = {}
+
+    for site_key in shared_state.values['sites']:
+        shorthand = site_key.lower()
+        raw_value = request.forms.get(shorthand)
+        # treat missing or empty string as intentional clear, no validation
+        if raw_value is None or raw_value.strip() == '':
+            continue
+
+        # non-empty submission: must validate
+        result = extract_valid_hostname(raw_value, shorthand)
+        domain = result.get('domain')
+        message = result.get('message', 'Error checking the hostname you provided!')
+        if domain:
+            valid_domains[site_key] = domain
+        else:
+            errors[site_key] = message
+
+    # Filter out any accidental empty domains and require at least one valid hostname overall
+    valid_domains = {k: d for k, d in valid_domains.items() if d}
+    if not valid_domains:
+        # report last or generic message
+        fail_msg = next(iter(errors.values()), 'No valid hostname provided!')
+        return render_fail(fail_msg)
+
+    # Save: valid ones, explicit empty for those omitted cleanly, leave untouched if error
+    changed_sites = []
+    for site_key in shared_state.values['sites']:
+        shorthand = site_key.lower()
+        raw_value = request.forms.get(shorthand)
+        # determine if change applies
+        if site_key in valid_domains:
+            new_val = valid_domains[site_key]
+            old_val = hostnames.get(shorthand) or ''
+            if old_val != new_val:
+                hostnames.save(shorthand, new_val)
+                changed_sites.append(shorthand)
+        elif raw_value is None:
+            # no submission: leave untouched
+            continue
+        elif raw_value.strip() == '':
+            old_val = hostnames.get(shorthand) or ''
+            if old_val != '':
+                hostnames.save(shorthand, '')
+
+    quasarr.providers.web_server.temp_server_success = True
+
+    # Build success message, include any per-site errors
+    success_msg = 'At least one valid hostname set!'
+    if errors:
+        optional_text = "<br>".join(f"{site}: {msg}" for site, msg in errors.items()) + "<br>"
+    else:
+        optional_text = "All provided hostnames are valid.<br>"
+
+    if not first_run:
+        # Append restart notice for specific sites that actually changed
+        for site in changed_sites:
+            if site.lower() in {'al', 'dd', 'nx'}:
+                optional_text += f"{site.upper()}: You must restart Quasarr and follow additional steps to start using this site.<br>"
+
+    return render_success(success_msg, timeout, optional_text=optional_text)
+
+
+def hostnames_config(shared_state):
+    app = Bottle()
+
+    @app.get('/')
+    def hostname_form():
+        message = """<p>
+          If you're having trouble setting this up, take a closer look at 
+          <a href="https://github.com/rix1337/Quasarr?tab=readme-ov-file#instructions" target="_blank" rel="noopener noreferrer">
+            step one of these instructions.
+          </a>
+        </p>"""
+        return render_form("Set at least one valid hostname", hostname_form_html(shared_state, message))
+
+    @app.post("/api/hostnames")
+    def set_hostnames():
+        return save_hostnames(shared_state)
+
+    info(f'Hostnames not set. Starting web server for config at: "{shared_state.values['internal_address']}".')
+    info("Please set at least one valid hostname there!")
+    return Server(app, listen='0.0.0.0', port=shared_state.values['port']).serve_temporarily()
+
+
+def hostname_credentials_config(shared_state, shorthand, domain):
+    app = Bottle()
+
+    shorthand = shorthand.upper()
+
+    @app.get('/')
+    def credentials_form():
+        form_content = f'''
+        <span>If required register account at: <a href="https://{domain}">{domain}</a>!</span><br><br>
+        <label for="user">Username</label>
+        <input type="text" id="user" name="user" placeholder="User" autocorrect="off"><br>
+
+        <label for="password">Password</label>
+        <input type="password" id="password" name="password" placeholder="Password"><br>
+        '''
+
+        form_html = f'''
+        <form action="/api/credentials/{shorthand}" method="post">
+            {form_content}
+            {render_button("Save", "primary", {"type": "submit"})}
+        </form>
+        '''
+
+        return render_form(f"Set User and Password for {shorthand}", form_html)
+
+    @app.post("/api/credentials/<sh>")
+    def set_credentials(sh):
+        user = request.forms.get('user')
+        password = request.forms.get('password')
+        config = Config(shorthand)
+
+        if user and password:
+            config.save("user", user)
+            config.save("password", password)
+
+            if sh.lower() == "al":
+                if quasarr.providers.sessions.al.create_and_persist_session(shared_state):
+                    quasarr.providers.web_server.temp_server_success = True
+                    return render_success(f"{sh} credentials set successfully", 5)
+            if sh.lower() == "dd":
+                if quasarr.providers.sessions.dd.create_and_persist_session(shared_state):
+                    quasarr.providers.web_server.temp_server_success = True
+                    return render_success(f"{sh} credentials set successfully", 5)
+            if sh.lower() == "nx":
+                if quasarr.providers.sessions.nx.create_and_persist_session(shared_state):
+                    quasarr.providers.web_server.temp_server_success = True
+                    return render_success(f"{sh} credentials set successfully", 5)
+
+        config.save("user", "")
+        config.save("password", "")
+        return render_fail("User and Password wrong or empty!")
+
+    info(
+        f'"{shorthand.lower()}" credentials required to access download links. '
+        f'Starting web server for config at: "{shared_state.values['internal_address']}".')
+    info(f"If needed register here: 'https://{domain}'")
+    info("Please set your credentials now, to allow Quasarr to launch!")
+    return Server(app, listen='0.0.0.0', port=shared_state.values['port']).serve_temporarily()
+
+
+def flaresolverr_config(shared_state):
+    app = Bottle()
+
+    @app.get('/')
+    def url_form():
+        form_content = '''
+        <span><a href="https://github.com/FlareSolverr/FlareSolverr?tab=readme-ov-file#installation">A local instance</a>
+        must be running and reachable to Quasarr!</span><br><br>
+        <label for="url">FlareSolverr URL</label>
+        <input type="text" id="url" name="url" placeholder="http://192.168.0.1:8191/v1"><br>
+        '''
+        form_html = f'''
+        <form action="/api/flaresolverr" method="post">
+            {form_content}
+            {render_button("Save", "primary", {"type": "submit"})}
+        </form>
+        '''
+        return render_form("Set FlareSolverr URL", form_html)
+
+    @app.post('/api/flaresolverr')
+    def set_flaresolverr_url():
+        url = request.forms.get('url').strip()
+        config = Config("FlareSolverr")
+
+        if url:
+            try:
+                headers = {"Content-Type": "application/json"}
+                data = {
+                    "cmd": "request.get",
+                    "url": "http://www.google.com/",
+                    "maxTimeout": 30000
+                }
+                response = requests.post(url, headers=headers, json=data, timeout=30)
+                if response.status_code == 200:
+                    config.save("url", url)
+                    print(f'Using Flaresolverr URL: "{url}"')
+                    quasarr.providers.web_server.temp_server_success = True
+                    return render_success("FlareSolverr URL saved successfully!", 5)
+            except requests.RequestException:
+                pass
+
+        # on failure, clear any existing value and notify user
+        config.save("url", "")
+        return render_fail("Could not reach FlareSolverr at that URL (expected HTTP 200).")
+
+    info(
+        '"flaresolverr" URL is required for proper operation. '
+        f'Starting web server for config at: "{shared_state.values["internal_address"]}".'
+    )
+    info("Please enter your FlareSolverr URL now.")
+    return Server(app, listen='0.0.0.0', port=shared_state.values['port']).serve_temporarily()
+
+
+def jdownloader_config(shared_state):
+    app = Bottle()
+
+    @app.get('/')
+    def jd_form():
+        verify_form_html = f'''
+        <span>If required register account at: <a href="https://my.jdownloader.org/login.html#register">
+        my.jdownloader.org</a>!</span><br>
+
+        <p><strong>JDownloader must be running and connected to My JDownloader!</strong></p><br>
+
+        <form id="verifyForm" action="/api/verify_jdownloader" method="post">
+            <label for="user">E-Mail</label>
+            <input type="text" id="user" name="user" placeholder="user@example.org" autocorrect="off"><br>
+            <label for="pass">Password</label>
+            <input type="password" id="pass" name="pass" placeholder="Password"><br>
+            {render_button("Verify Credentials",
+                           "secondary",
+                           {"id": "verifyButton", "type": "button", "onclick": "verifyCredentials()"})}
+        </form>
+
+        <p>Some JDownloader settings will be enforced by Quasarr on startup.</p>
+
+        <form action="/api/store_jdownloader" method="post" id="deviceForm" style="display: none;">
+            <input type="hidden" id="hiddenUser" name="user">
+            <input type="hidden" id="hiddenPass" name="pass">
+            <label for="device">JDownloader</label>
+            <select id="device" name="device"></select><br>
+            {render_button("Save", "primary", {"type": "submit"})}
+        </form>
+        <p><strong>Saving may take a while!</strong></p><br>
+        '''
+
+        verify_script = '''
+        <script>
+        function verifyCredentials() {
+            var user = document.getElementById('user').value;
+            var pass = document.getElementById('pass').value;
+            fetch('/api/verify_jdownloader', {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                },
+                body: JSON.stringify({user: user, pass: pass}),
+            })
+            .then(response => response.json())
+            .then(data => {
+                if (data.success) {
+                    var select = document.getElementById('device');
+                    data.devices.forEach(device => {
+                        var opt = document.createElement('option');
+                        opt.value = device;
+                        opt.innerHTML = device;
+                        select.appendChild(opt);
+                    });
+                    document.getElementById('hiddenUser').value = document.getElementById('user').value;
+                    document.getElementById('hiddenPass').value = document.getElementById('pass').value;
+                    document.getElementById("verifyButton").style.display = "none";
+                    document.getElementById('deviceForm').style.display = 'block';
+                } else {
+                    alert('Fehler! Bitte die Zugangsdaten überprüfen.');
+                }
+            })
+            .catch((error) => {
+                console.error('Error:', error);
+            });
+        }
+        </script>
+        '''
+        return render_form("Set your credentials for My JDownloader", verify_form_html, verify_script)
+
+    @app.post("/api/verify_jdownloader")
+    def verify_jdownloader():
+        data = request.json
+        username = data['user']
+        password = data['pass']
+
+        devices = shared_state.get_devices(username, password)
+        device_names = []
+
+        if devices:
+            for device in devices:
+                device_names.append(device['name'])
+
+        if device_names:
+            return {"success": True, "devices": device_names}
+        else:
+            return {"success": False}
+
+    @app.post("/api/store_jdownloader")
+    def store_jdownloader():
+        username = request.forms.get('user')
+        password = request.forms.get('pass')
+        device = request.forms.get('device')
+
+        config = Config('JDownloader')
+
+        if username and password and device:
+            config.save('user', username)
+            config.save('password', password)
+            config.save('device', device)
+
+            if not shared_state.set_device_from_config():
+                config.save('user', "")
+                config.save('password', "")
+                config.save('device', "")
+            else:
+                quasarr.providers.web_server.temp_server_success = True
+                return render_success("Credentials set",
+                                      15)
+
+        return render_fail("Could not set credentials!")
+
+    info(
+        f'My-JDownloader-Credentials not set. '
+        f'Starting web server for config at: "{shared_state.values['internal_address']}".')
+    info("If needed register here: 'https://my.jdownloader.org/login.html#register'")
+    info("Please set your credentials now, to allow Quasarr to launch!")
+    return Server(app, listen='0.0.0.0', port=shared_state.values['port']).serve_temporarily()

quasarr/storage/sqlite_database.py

@@ -0,0 +1,80 @@
+# -*- coding: utf-8 -*-
+# Quasarr
+# Project by https://github.com/rix1337
+
+import sqlite3
+import time
+
+from quasarr.providers import shared_state
+from quasarr.providers.log import info
+
+
+class DataBase(object):
+    def __init__(self, table):
+        try:
+            self._conn = sqlite3.connect(shared_state.values["dbfile"], check_same_thread=False, timeout=5)
+            self._table = table
+            if not self._conn.execute(
+                    f"SELECT sql FROM sqlite_master WHERE type = 'table' AND name = '{self._table}';").fetchall():
+                self._conn.execute(f"CREATE TABLE {self._table} (key, value)")
+                self._conn.commit()
+        except sqlite3.OperationalError as e:
+            try:
+                time.sleep(5)
+                self._conn = sqlite3.connect(shared_state.values["dbfile"], check_same_thread=False, timeout=10)
+                self._table = table
+                if not self._conn.execute(
+                        f"SELECT sql FROM sqlite_master WHERE type = 'table' AND name = '{self._table}';").fetchall():
+                    self._conn.execute(f"CREATE TABLE {self._table} (key, value)")
+                    self._conn.commit()
+            except sqlite3.OperationalError as e:
+                info(f"Error accessing Quasarr.db: {e}")
+
+    def retrieve(self, key):
+        query = f"SELECT value FROM {self._table} WHERE key=?"
+        # using this parameterized query to prevent SQL injection, which requires a tuple as second argument
+        res = self._conn.execute(query, (key,)).fetchone()
+        return res[0] if res else None
+
+    def retrieve_all(self, key):
+        query = f"SELECT distinct value FROM {self._table} WHERE key=? ORDER BY value"
+        # using this parameterized query to prevent SQL injection, which requires a tuple as second argument
+        res = self._conn.execute(query, (key,))
+        items = [str(r[0]) for r in res]
+        return items
+
+    def retrieve_all_titles(self):
+        query = f"SELECT distinct key, value FROM {self._table} ORDER BY key"
+        # using this parameterized query to prevent SQL injection, which requires a tuple as second argument
+        res = self._conn.execute(query)
+        items = [[str(r[0]), str(r[1])] for r in res]
+        return items if items else None
+
+    def store(self, key, value):
+        query = f"INSERT INTO {self._table} VALUES (?, ?)"
+        # using this parameterized query to prevent SQL injection, which requires a tuple as second argument
+        self._conn.execute(query, (key, value))
+        self._conn.commit()
+        return True
+
+    def update_store(self, key, value):
+        delete_query = f"DELETE FROM {self._table} WHERE key=?"
+        # using this parameterized query to prevent SQL injection, which requires a tuple as second argument
+        self._conn.execute(delete_query, (key,))
+        insert_query = f"INSERT INTO {self._table} VALUES (?, ?)"
+        # using this parameterized query to prevent SQL injection, which requires a tuple as second argument
+        self._conn.execute(insert_query, (key, value))
+        self._conn.commit()
+        return True
+
+    def delete(self, key):
+        query = f"DELETE FROM {self._table} WHERE key=?"
+        # using this parameterized query to prevent SQL injection, which requires a tuple as second argument
+        self._conn.execute(query, (key,))
+        self._conn.commit()
+        return True
+
+    def reset(self):
+        self._conn.execute(f"DROP TABLE IF EXISTS {self._table}")
+        self._conn.commit()
+        return True