Update app.py

app.py

@@ -497,6 +497,106 @@ const content = document.getElementById('content');
 let active = "";
 let currentPage = 1;
 
+// LocalStorage functionality for URL persistence
+function saveToLocalStorage(urls) {
+  try {
+    localStorage.setItem('favoriteUrls', JSON.stringify(urls));
+    console.log('Saved URLs to localStorage:', urls.length);
+    return true;
+  } catch (e) {
+    console.error('Error saving to localStorage:', e);
+    return false;
+  }
+}
+
+function loadFromLocalStorage() {
+  try {
+    const data = localStorage.getItem('favoriteUrls');
+    if (data) {
+      const urls = JSON.parse(data);
+      console.log('Loaded URLs from localStorage:', urls.length);
+      return urls;
+    }
+  } catch (e) {
+    console.error('Error loading from localStorage:', e);
+  }
+  return null;
+}
+
+// Export/Import functionality
+function exportUrls() {
+  makeRequest('/api/favorites?per_page=1000', 'GET', null, function(data) {
+    if (data.items && data.items.length > 0) {
+      const urls = data.items.map(item => item.url);
+      
+      // Save to localStorage as backup
+      saveToLocalStorage(urls);
+      
+      // Create file for download
+      const blob = new Blob([JSON.stringify(urls, null, 2)], { type: 'application/json' });
+      const a = document.createElement('a');
+      a.href = URL.createObjectURL(blob);
+      a.download = 'favorite_urls.json';
+      document.body.appendChild(a);
+      a.click();
+      document.body.removeChild(a);
+    } else {
+      alert('No URLs to export');
+    }
+  });
+}
+
+function importUrls() {
+  document.getElementById('import-file').click();
+}
+
+function handleImportFile(files) {
+  if (files.length === 0) return;
+  
+  const file = files[0];
+  const reader = new FileReader();
+  
+  reader.onload = function(e) {
+    try {
+      const urls = JSON.parse(e.target.result);
+      if (Array.isArray(urls)) {
+        // Save to localStorage
+        saveToLocalStorage(urls);
+        
+        // Add each URL to the server
+        let processed = 0;
+        
+        function addNextUrl(index) {
+          if (index >= urls.length) {
+            alert(`Import complete. Added ${processed} URLs.`);
+            loadUrlList();
+            if (active === 'Favorites') {
+              loadFavorites(currentPage);
+            }
+            return;
+          }
+          
+          const formData = new FormData();
+          formData.append('url', urls[index]);
+          
+          makeRequest('/api/url/add', 'POST', formData, function(data) {
+            if (data.success) processed++;
+            addNextUrl(index + 1);
+          });
+        }
+        
+        addNextUrl(0);
+      } else {
+        alert('Invalid format. File must contain a JSON array of URLs.');
+      }
+    } catch (e) {
+      alert('Error parsing file: ' + e.message);
+    }
+  };
+  
+  reader.readAsText(file);
+}
+
 // Simple utility functions
 function loadHTML(url, callback) {
   const xhr = new XMLHttpRequest();
@@ -635,6 +735,14 @@ function loadManage() {
       
       <h2>Manage Saved URLs</h2>
       <div id="url-list" class="url-list">Loading...</div>
+      
+      <div style="margin-top: 30px;">
+        <h3>Backup & Restore</h3>
+        <p>Server storage may not persist across restarts. Use these options to save your data:</p>
+        <button onclick="exportUrls()" class="btn btn-success">Export URLs</button>
+        <button onclick="importUrls()" class="btn btn-primary">Import URLs</button>
+        <input type="file" id="import-file" style="display:none" onchange="handleImportFile(this.files)">
+      </div>
     </div>
   `;
   
@@ -643,14 +751,53 @@ function loadManage() {
 
 // URL management functions
 function loadUrlList() {
+  // First try to load from localStorage as a fallback
+  const localUrls = loadFromLocalStorage();
+  
   makeRequest('/api/favorites?per_page=100', 'GET', null, function(data) {
     const urlList = document.getElementById('url-list');
     
+    // If server has no URLs but localStorage does, restore from localStorage
+    if (data.items.length === 0 && localUrls && localUrls.length > 0) {
+      showStatus('add-status', 'Restoring URLs from local backup...', true);
+      
+      // Add each URL from localStorage to the server
+      let restored = 0;
+      
+      function restoreNextUrl(index) {
+        if (index >= localUrls.length) {
+          showStatus('add-status', `Restored ${restored} URLs from local backup`, true);
+          // Reload the list after restoration
+          setTimeout(() => {
+            loadUrlList();
+            if (active === 'Favorites') {
+              loadFavorites(currentPage);
+            }
+          }, 1000);
+          return;
+        }
+        
+        const formData = new FormData();
+        formData.append('url', localUrls[index]);
+        
+        makeRequest('/api/url/add', 'POST', formData, function(data) {
+          if (data.success) restored++;
+          restoreNextUrl(index + 1);
+        });
+      }
+      
+      restoreNextUrl(0);
+      return;
+    }
+    
     if(data.items.length === 0) {
       urlList.innerHTML = '<p style="text-align:center;padding:20px">No URLs saved yet.</p>';
       return;
     }
     
+    // Update localStorage with server data
+    saveToLocalStorage(data.items.map(item => item.url));
+    
     let html = '';
     data.items.forEach(item => {
       // Escape the URL to prevent JavaScript injection when used in onclick handlers
@@ -670,6 +817,7 @@ function loadUrlList() {
     urlList.innerHTML = html;
   });
 }
+}
 
 function addUrl() {
   const url = document.getElementById('new-url').value.trim();
@@ -686,6 +834,14 @@ function addUrl() {
     showStatus('add-status', data.message, data.success);
     if(data.success) {
       document.getElementById('new-url').value = '';
+      
+      // Update localStorage
+      const localUrls = loadFromLocalStorage() || [];
+      if (!localUrls.includes(url)) {
+        localUrls.unshift(url); // Add to beginning
+        saveToLocalStorage(localUrls);
+      }
+      
       loadUrlList();
       // If currently in Favorites tab, reload to see changes immediately
       if(active === 'Favorites') {
@@ -708,6 +864,14 @@ function editUrl(url) {
   
   makeRequest('/api/url/update', 'POST', formData, function(data) {
     if(data.success) {
+      // Update localStorage
+      let localUrls = loadFromLocalStorage() || [];
+      const index = localUrls.indexOf(decodedUrl);
+      if (index !== -1) {
+        localUrls[index] = newUrl;
+        saveToLocalStorage(localUrls);
+      }
+      
       loadUrlList();
       // If currently in Favorites tab, reload to see changes immediately
       if(active === 'Favorites') {
@@ -729,6 +893,14 @@ function deleteUrl(url) {
   
   makeRequest('/api/url/delete', 'POST', formData, function(data) {
     if(data.success) {
+      // Update localStorage
+      let localUrls = loadFromLocalStorage() || [];
+      const index = localUrls.indexOf(decodedUrl);
+      if (index !== -1) {
+        localUrls.splice(index, 1);
+        saveToLocalStorage(localUrls);
+      }
+      
       loadUrlList();
       // If currently in Favorites tab, reload to see changes immediately
       if(active === 'Favorites') {
@@ -778,6 +950,52 @@ tabs.appendChild(manageTab);
 
 // Start with Favorites tab
 loadFavorites(1);
+
+// Check for localStorage on page load 
+window.addEventListener('load', function() {
+  // Try to load URLs from localStorage
+  const localUrls = loadFromLocalStorage();
+  
+  // If we have URLs in localStorage, make sure they're on the server
+  if (localUrls && localUrls.length > 0) {
+    console.log(`Found ${localUrls.length} URLs in localStorage`);
+    
+    // First get server URLs to compare
+    makeRequest('/api/favorites?per_page=1000', 'GET', null, function(data) {
+      const serverUrls = data.items.map(item => item.url);
+      
+      // Find URLs that are in localStorage but not on server
+      const missingUrls = localUrls.filter(url => !serverUrls.includes(url));
+      
+      if (missingUrls.length > 0) {
+        console.log(`Found ${missingUrls.length} URLs missing from server, restoring...`);
+        
+        // Add missing URLs to server
+        let restored = 0;
+        
+        function restoreNextUrl(index) {
+          if (index >= missingUrls.length) {
+            console.log(`Restored ${restored} URLs from localStorage`);
+            if (active === 'Favorites') {
+              loadFavorites(currentPage);
+            }
+            return;
+          }
+          
+          const formData = new FormData();
+          formData.append('url', missingUrls[index]);
+          
+          makeRequest('/api/url/add', 'POST', formData, function(data) {
+            if (data.success) restored++;
+            restoreNextUrl(index + 1);
+          });
+        }
+        
+        restoreNextUrl(0);
+      }
+    });
+  }
+});
 </script>
 </body>
 </html>''')