quasarr 1.26.6__py3-none-any.whl → 1.27.0__py3-none-any.whl

quasarr/storage/setup.py

@@ -4,6 +4,7 @@
 
 import os
 import sys
+from urllib.parse import urlparse
 
 import requests
 from bottle import Bottle, request, response
@@ -14,11 +15,70 @@ import quasarr.providers.sessions.al
 import quasarr.providers.sessions.dd
 import quasarr.providers.sessions.dl
 import quasarr.providers.sessions.nx
-from quasarr.providers.html_templates import render_button, render_form, render_success, render_fail
+from quasarr.providers.html_templates import render_button, render_form, render_success, render_fail, \
+    render_centered_html
 from quasarr.providers.log import info
 from quasarr.providers.shared_state import extract_valid_hostname
+from quasarr.providers.utils import extract_kv_pairs, extract_allowed_keys
 from quasarr.providers.web_server import Server
 from quasarr.storage.config import Config
+from quasarr.storage.sqlite_database import DataBase
+
+
+def render_reconnect_success(message, countdown_seconds=3):
+    """Render a success page that waits, then polls until the server is back online."""
+    button_html = render_button(f"Continuing in {countdown_seconds}...", "secondary",
+                                {"id": "reconnectBtn", "disabled": "true"})
+
+    script = f'''
+        <script>
+            var remaining = {countdown_seconds};
+            var btn = document.getElementById('reconnectBtn');
+
+            var interval = setInterval(function() {{
+                remaining--;
+                btn.innerText = 'Continuing in ' + remaining + '...';
+                if (remaining <= 0) {{
+                    clearInterval(interval);
+                    btn.innerText = 'Reconnecting...';
+                    tryReconnect();
+                }}
+            }}, 1000);
+
+            function tryReconnect() {{
+                var attempts = 0;
+                function attempt() {{
+                    attempts++;
+                    fetch('/', {{ method: 'HEAD', cache: 'no-store' }})
+                    .then(function(response) {{
+                        if (response.ok) {{
+                            btn.innerText = 'Connected! Reloading...';
+                            btn.className = 'btn-primary';
+                            setTimeout(function() {{ window.location.href = '/'; }}, 500);
+                        }} else {{
+                            scheduleRetry();
+                        }}
+                    }})
+                    .catch(function() {{
+                        scheduleRetry();
+                    }});
+                }}
+                function scheduleRetry() {{
+                    btn.innerText = 'Reconnecting... (attempt ' + attempts + ')';
+                    setTimeout(attempt, 1000);
+                }}
+                attempt();
+            }}
+        </script>
+    '''
+
+    content = f'''<h1><img src="{images.logo}" type="image/png" alt="Quasarr logo" class="logo"/>Quasarr</h1>
+    <h2>✓ Success</h2>
+    <p>{message}</p>
+    {button_html}
+    {script}
+    '''
+    return render_centered_html(content)
 
 
 def add_no_cache_headers(app):
@@ -81,22 +141,31 @@ def path_config(shared_state):
         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)
+        return render_reconnect_success(f'Config path set to: "{config_path}"')
 
     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):
+def hostname_form_html(shared_state, message, show_restart_button=False, show_skip_management=False):
     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>
     '''
 
+    skip_indicator = '''
+    <div class="skip-indicator" id="skip-indicator-{id}" style="margin-top:-0.5rem; margin-bottom:0.75rem; padding:0.5rem; background:var(--code-bg, #f8f9fa); border-radius:0.25rem; font-size:0.875rem;">
+        <span style="color:#dc3545;">⚠️ Login skipped</span>
+        <button type="button" class="btn-subtle" style="margin-left:0.5rem; padding:0.25rem 0.5rem; font-size:0.75rem;" onclick="clearSkipLogin('{id}', this)">Clear &amp; require login</button>
+    </div>
+    '''
+
     field_html = []
     hostnames = Config('Hostnames')  # Load once outside the loop
+    skip_login_db = DataBase("skip_login")
+    login_required_sites = ['al', 'dd', 'dl', 'nx']
+
     for label in shared_state.values["sites"]:
         field_id = label.lower()
         img_html = ''
@@ -119,32 +188,116 @@ def hostname_form_html(shared_state, message):
             value=current_value
         ))
 
+        # Add skip indicator for login-required sites if skip management is enabled
+        if show_skip_management and field_id in login_required_sites:
+            if current_value and skip_login_db.retrieve(field_id):
+                field_html.append(skip_indicator.format(id=field_id))
+
     hostname_form_content = "".join(field_html)
     button_html = render_button("Save", "primary", {"type": "submit", "id": "submitBtn"})
 
+    # Get stored hostnames URL if available
+    stored_url = Config('Settings').get("hostnames_url") or ""
+
+    # Build restart button HTML if needed
+    restart_section = ""
+    if show_restart_button:
+        restart_section = f'''
+        <div class="section-divider" style="margin-top:1.5rem; padding-top:1rem; border-top:1px solid var(--divider-color, #dee2e6);">
+            <p style="font-size:0.875rem; color:var(--secondary, #6c757d);">Restart required after changing login-required hostnames (AL, DD, DL, NX)</p>
+            {render_button("Restart Quasarr", "secondary", {"type": "button", "onclick": "confirmRestart()"})}
+        </div>
+        '''
+
     template = """
+<style>
+    .url-import-section {{
+        border: 1px solid var(--divider-color, #dee2e6);
+        border-radius: 0.5rem;
+        padding: 1rem;
+        margin-bottom: 1.5rem;
+        background: var(--code-bg, #f8f9fa);
+    }}
+    .url-import-section h3 {{
+        margin: 0 0 0.75rem 0;
+        font-size: 1rem;
+        font-weight: 600;
+    }}
+    .url-import-row {{
+        display: flex;
+        gap: 0.5rem;
+        align-items: stretch;
+    }}
+    .url-import-row input {{
+        flex: 1;
+        margin-bottom: 0;
+    }}
+    .url-import-row button {{
+        margin-top: 0;
+        white-space: nowrap;
+    }}
+    .import-status {{
+        margin-top: 0.5rem;
+        font-size: 0.875rem;
+        min-height: 1.25rem;
+    }}
+    .import-status.success {{ color: #198754; }}
+    .import-status.error {{ color: #dc3545; }}
+    .import-status.loading {{ color: var(--secondary, #6c757d); }}
+    .btn-subtle {{
+        background: transparent;
+        color: var(--fg-color, #212529);
+        border: 1px solid var(--btn-subtle-border, #ced4da);
+        padding: 0.25rem 0.5rem;
+        border-radius: 0.25rem;
+        cursor: pointer;
+        font-size: 0.875rem;
+    }}
+    .btn-subtle:hover {{
+        background: var(--btn-subtle-bg, #e9ecef);
+    }}
+</style>
+
 <div id="message" style="margin-bottom:0.5em;">{message}</div>
 <div id="error-msg" style="color:red; margin-bottom:1em;"></div>
 
+<div class="url-import-section">
+    <h3>📥 Import from URL</h3>
+    <div class="url-import-row">
+        <input type="text" id="hostnamesUrl" placeholder="https://pastebin.com/raw/..." value="{stored_url}" autocorrect="off" autocomplete="off">
+        <button type="button" class="btn-secondary" id="importBtn" onclick="importHostnames()">Import</button>
+    </div>
+    <div id="importStatus" class="import-status"></div>
+    <p style="font-size:0.75rem; color:var(--secondary, #6c757d); margin:0.5rem 0 0 0;">
+        Paste a URL containing hostname definitions (same format as --hostnames parameter)
+    </p>
+</div>
+
 <form action="/api/hostnames" method="post" onsubmit="return validateHostnames(this)">
+    <input type="hidden" id="hostnamesUrlHidden" name="hostnames_url" value="{stored_url}">
     {hostname_form_content}
     {button}
 </form>
 
+{restart_section}
+
 <script>
   var formSubmitted = false;
+
   function validateHostnames(form) {{
     if (formSubmitted) return false;
 
     var errorDiv = document.getElementById('error-msg');
     errorDiv.textContent = '';
 
-    var inputs = form.querySelectorAll('input[type="text"]');
+    var inputs = form.querySelectorAll('input[type="text"]:not(#hostnamesUrl)');
     for (var i = 0; i < inputs.length; i++) {{
       if (inputs[i].value.trim() !== '') {{
         formSubmitted = true;
         var btn = document.getElementById('submitBtn');
         if (btn) {{ btn.disabled = true; btn.textContent = 'Saving...'; }}
+        // Sync the URL field to hidden input
+        document.getElementById('hostnamesUrlHidden').value = document.getElementById('hostnamesUrl').value.trim();
         return true;
       }}
     }}
@@ -153,12 +306,164 @@ def hostname_form_html(shared_state, message):
     inputs[0].focus();
     return false;
   }}
+
+  function importHostnames() {{
+    var urlInput = document.getElementById('hostnamesUrl');
+    var url = urlInput.value.trim();
+    var statusDiv = document.getElementById('importStatus');
+    var importBtn = document.getElementById('importBtn');
+
+    if (!url) {{
+      statusDiv.className = 'import-status error';
+      statusDiv.textContent = 'Please enter a URL';
+      return;
+    }}
+
+    statusDiv.className = 'import-status loading';
+    statusDiv.textContent = 'Importing...';
+    importBtn.disabled = true;
+    importBtn.textContent = 'Importing...';
+
+    fetch('/api/hostnames/import-url', {{
+      method: 'POST',
+      headers: {{ 'Content-Type': 'application/json' }},
+      body: JSON.stringify({{ url: url }})
+    }})
+    .then(response => response.json())
+    .then(data => {{
+      importBtn.disabled = false;
+      importBtn.textContent = 'Import';
+
+      if (data.success) {{
+        var count = 0;
+        for (var key in data.hostnames) {{
+          var input = document.getElementById(key);
+          if (input) {{
+            input.value = data.hostnames[key];
+            count++;
+          }}
+        }}
+        statusDiv.className = 'import-status success';
+        var msg = 'Imported ' + count + ' hostname(s)';
+        if (data.errors && Object.keys(data.errors).length > 0) {{
+          msg += ' (' + Object.keys(data.errors).length + ' invalid)';
+        }}
+        statusDiv.textContent = msg + '. Review and click Save.';
+      }} else {{
+        statusDiv.className = 'import-status error';
+        statusDiv.textContent = data.error || 'Import failed';
+      }}
+    }})
+    .catch(error => {{
+      importBtn.disabled = false;
+      importBtn.textContent = 'Import';
+      statusDiv.className = 'import-status error';
+      statusDiv.textContent = 'Network error: ' + error.message;
+    }});
+  }}
+
+  function clearSkipLogin(shorthand, btnElement) {{
+    fetch('/api/skip-login/' + shorthand, {{ method: 'DELETE' }})
+    .then(response => response.json())
+    .then(data => {{
+      if (data.success) {{
+        // Remove the skip indicator using the button's parent
+        var indicator = btnElement.closest('.skip-indicator');
+        if (indicator) indicator.remove();
+        alert('Login requirement restored for ' + shorthand.toUpperCase() + '. Restart Quasarr to be prompted for credentials.');
+      }} else {{
+        alert('Failed to clear skip preference');
+      }}
+    }})
+    .catch(error => {{
+      alert('Error: ' + error.message);
+    }});
+  }}
+
+  function confirmRestart() {{
+    if (confirm('Restart Quasarr now? Any unsaved changes will be lost.')) {{
+      fetch('/api/restart', {{ method: 'POST' }})
+      .then(response => response.json())
+      .then(data => {{
+        if (data.success) {{
+          showRestartOverlay();
+        }}
+      }})
+      .catch(error => {{
+        // Expected - connection will be lost during restart
+        showRestartOverlay();
+      }});
+    }}
+  }}
+
+  function showRestartOverlay() {{
+    document.body.innerHTML = `
+      <div style="text-align:center; padding:2rem; font-family:system-ui,-apple-system,sans-serif;">
+        <h2>Restarting Quasarr...</h2>
+        <p id="restartStatus">Waiting <span id="countdown">10</span> seconds...</p>
+        <div id="spinner" style="display:none; margin-top:1rem;">
+          <div style="display:inline-block; width:24px; height:24px; border:3px solid #ccc; border-top-color:#333; border-radius:50%; animation:spin 1s linear infinite;"></div>
+          <style>@keyframes spin {{ to {{ transform: rotate(360deg); }} }}</style>
+        </div>
+      </div>
+    `;
+    startCountdown(10);
+  }}
+
+  function startCountdown(seconds) {{
+    var countdownEl = document.getElementById('countdown');
+    var statusEl = document.getElementById('restartStatus');
+    var spinnerEl = document.getElementById('spinner');
+
+    var remaining = seconds;
+    var interval = setInterval(function() {{
+      remaining--;
+      if (countdownEl) countdownEl.textContent = remaining;
+
+      if (remaining <= 0) {{
+        clearInterval(interval);
+        statusEl.textContent = 'Reconnecting...';
+        spinnerEl.style.display = 'block';
+        tryReconnect();
+      }}
+    }}, 1000);
+  }}
+
+  function tryReconnect() {{
+    var statusEl = document.getElementById('restartStatus');
+    var attempts = 0;
+
+    function attempt() {{
+      attempts++;
+      fetch('/', {{ method: 'HEAD', cache: 'no-store' }})
+      .then(response => {{
+        if (response.ok) {{
+          statusEl.textContent = 'Connected! Reloading...';
+          setTimeout(function() {{ window.location.href = '/'; }}, 500);
+        }} else {{
+          scheduleRetry();
+        }}
+      }})
+      .catch(function() {{
+        scheduleRetry();
+      }});
+    }}
+
+    function scheduleRetry() {{
+      statusEl.textContent = 'Reconnecting... (attempt ' + attempts + ')';
+      setTimeout(attempt, 1000);
+    }}
+
+    attempt();
+  }}
 </script>
 """
     return template.format(
         message=message,
         hostname_form_content=hostname_form_content,
-        button=button_html
+        button=button_html,
+        stored_url=stored_url,
+        restart_section=restart_section
     )
 
 
@@ -212,6 +517,11 @@ def save_hostnames(shared_state, timeout=5, first_run=True):
             if old_val != '':
                 hostnames.save(shorthand, '')
 
+    # Handle hostnames URL storage
+    hostnames_url = request.forms.get('hostnames_url', '').strip()
+    settings_config = Config("Settings")
+    settings_config.save("hostnames_url", hostnames_url)
+
     quasarr.providers.web_server.temp_server_success = True
 
     # Build success message, include any per-site errors
@@ -227,7 +537,8 @@ def save_hostnames(shared_state, timeout=5, first_run=True):
             if site.lower() in {'al', 'dd', 'dl', '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)
+    full_message = f"{success_msg}<br><small>{optional_text}</small>"
+    return render_reconnect_success(full_message)
 
 
 def hostnames_config(shared_state):
@@ -248,6 +559,63 @@ def hostnames_config(shared_state):
     def set_hostnames():
         return save_hostnames(shared_state)
 
+    @app.post("/api/hostnames/import-url")
+    def import_hostnames_from_url():
+        """Fetch URL and parse hostnames, return JSON for JS to populate fields."""
+        response.content_type = 'application/json'
+        try:
+            data = request.json
+            url = data.get('url', '').strip()
+
+            if not url:
+                return {"success": False, "error": "No URL provided"}
+
+            # Validate URL
+            parsed = urlparse(url)
+            if parsed.scheme not in ("http", "https") or not parsed.netloc:
+                return {"success": False, "error": "Invalid URL format"}
+
+            if "/raw/eX4Mpl3" in url:
+                return {"success": False, "error": "Example URL detected. Please provide a real URL."}
+
+            # Fetch content
+            try:
+                resp = requests.get(url, timeout=15)
+                resp.raise_for_status()
+                content = resp.text
+            except requests.RequestException as e:
+                return {"success": False, "error": f"Failed to fetch URL: {str(e)}"}
+
+            # Parse hostnames
+            allowed_keys = extract_allowed_keys(Config._DEFAULT_CONFIG, 'Hostnames')
+            results = extract_kv_pairs(content, allowed_keys)
+
+            if not results:
+                return {"success": False, "error": "No hostnames found in the provided URL"}
+
+            # Validate each hostname
+            valid_hostnames = {}
+            invalid_hostnames = {}
+            for shorthand, hostname in results.items():
+                domain_check = extract_valid_hostname(hostname, shorthand)
+                domain = domain_check.get('domain')
+                if domain:
+                    valid_hostnames[shorthand] = domain
+                else:
+                    invalid_hostnames[shorthand] = domain_check.get('message', 'Invalid')
+
+            if not valid_hostnames:
+                return {"success": False, "error": "No valid hostnames found in the provided URL"}
+
+            return {
+                "success": True,
+                "hostnames": valid_hostnames,
+                "errors": invalid_hostnames
+            }
+
+        except Exception as e:
+            return {"success": False, "error": f"Error: {str(e)}"}
+
     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()
@@ -271,10 +639,39 @@ def hostname_credentials_config(shared_state, shorthand, domain):
         '''
 
         form_html = f'''
+        <style>
+            .button-row {{
+                display: flex;
+                gap: 0.75rem;
+                justify-content: center;
+                flex-wrap: wrap;
+                margin-top: 1rem;
+            }}
+            .btn-warning {{
+                background-color: #ffc107;
+                color: #212529;
+                border: 1.5px solid #d39e00;
+                padding: 0.5rem 1rem;
+                font-size: 1rem;
+                border-radius: 0.5rem;
+                font-weight: 500;
+                cursor: pointer;
+            }}
+            .btn-warning:hover {{
+                background-color: #e0a800;
+                border-color: #c69500;
+            }}
+        </style>
         <form id="credentialsForm" action="/api/credentials/{shorthand}" method="post" onsubmit="return handleSubmit(this)">
             {form_content}
-            {render_button("Save", "primary", {"type": "submit", "id": "submitBtn"})}
+            <div class="button-row">
+                {render_button("Save", "primary", {"type": "submit", "id": "submitBtn"})}
+                <button type="button" class="btn-warning" id="skipBtn" onclick="skipLogin()">Skip for now</button>
+            </div>
         </form>
+        <p style="font-size:0.875rem; color:var(--secondary, #6c757d); margin-top:1rem;">
+            Skipping will allow Quasarr to start, but this site won't work until credentials are provided.
+        </p>
         <script>
         var formSubmitted = false;
         function handleSubmit(form) {{
@@ -282,13 +679,54 @@ def hostname_credentials_config(shared_state, shorthand, domain):
             formSubmitted = true;
             var btn = document.getElementById('submitBtn');
             if (btn) {{ btn.disabled = true; btn.textContent = 'Saving...'; }}
+            document.getElementById('skipBtn').disabled = true;
             return true;
         }}
+        function skipLogin() {{
+            if (formSubmitted) return;
+            formSubmitted = true;
+            var skipBtn = document.getElementById('skipBtn');
+            var submitBtn = document.getElementById('submitBtn');
+            if (skipBtn) {{ skipBtn.disabled = true; skipBtn.textContent = 'Skipping...'; }}
+            if (submitBtn) {{ submitBtn.disabled = true; }}
+
+            fetch('/api/credentials/{shorthand}/skip', {{ method: 'POST' }})
+            .then(response => {{
+                if (response.ok) {{
+                    window.location.href = '/skip-success';
+                }} else {{
+                    alert('Failed to skip login');
+                    formSubmitted = false;
+                    if (skipBtn) {{ skipBtn.disabled = false; skipBtn.textContent = 'Skip for now'; }}
+                    if (submitBtn) {{ submitBtn.disabled = false; }}
+                }}
+            }})
+            .catch(error => {{
+                alert('Error: ' + error.message);
+                formSubmitted = false;
+                if (skipBtn) {{ skipBtn.disabled = false; skipBtn.textContent = 'Skip for now'; }}
+                if (submitBtn) {{ submitBtn.disabled = false; }}
+            }});
+        }}
         </script>
         '''
 
         return render_form(f"Set User and Password for {shorthand}", form_html)
 
+    @app.get('/skip-success')
+    def skip_success():
+        return render_reconnect_success(
+            f"{shorthand} login skipped. You can configure credentials later in the web UI.")
+
+    @app.post("/api/credentials/<sh>/skip")
+    def skip_credentials(sh):
+        """Skip login for this hostname and continue startup."""
+        sh_lower = sh.lower()
+        DataBase("skip_login").update_store(sh_lower, "true")
+        info(f'Login for "{sh}" skipped by user choice')
+        quasarr.providers.web_server.temp_server_success = True
+        return {"success": True}
+
     @app.post("/api/credentials/<sh>")
     def set_credentials(sh):
         # Guard against duplicate submissions (e.g., double-click)
@@ -303,22 +741,25 @@ def hostname_credentials_config(shared_state, shorthand, domain):
             config.save("user", user)
             config.save("password", password)
 
+            # Clear any skip preference since we now have credentials
+            DataBase("skip_login").delete(sh.lower())
+
             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)
+                    return render_reconnect_success(f"{sh} credentials set successfully")
             elif 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)
+                    return render_reconnect_success(f"{sh} credentials set successfully")
             elif sh.lower() == "dl":
                 if quasarr.providers.sessions.dl.create_and_persist_session(shared_state):
                     quasarr.providers.web_server.temp_server_success = True
-                    return render_success(f"{sh} credentials set successfully", 5)
+                    return render_reconnect_success(f"{sh} credentials set successfully")
             elif 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)
+                    return render_reconnect_success(f"{sh} credentials set successfully")
             else:
                 quasarr.providers.web_server.temp_server_success = False
                 return render_fail(f"Unknown site shorthand! ({sh})")
@@ -331,7 +772,7 @@ def hostname_credentials_config(shared_state, shorthand, domain):
         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!")
+    info("Please set your credentials now, or skip to allow Quasarr to launch!")
     return Server(app, listen='0.0.0.0', port=shared_state.values['port']).serve_temporarily()
 
 
@@ -386,7 +827,7 @@ def flaresolverr_config(shared_state):
                     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)
+                    return render_reconnect_success("FlareSolverr URL saved successfully!")
             except requests.RequestException:
                 pass
 
@@ -524,7 +965,7 @@ def jdownloader_config(shared_state):
                 config.save('password', password)
                 config.save('device', device)
                 quasarr.providers.web_server.temp_server_success = True
-                return render_success("Credentials set", 15)
+                return render_reconnect_success("Credentials set")
 
         return render_fail("Could not set credentials!")