fix duplicates and fix nuke

Author: Fredolx

src-tauri/src/lib.rs

@@ -287,7 +287,7 @@ fn toggle_source(value: bool, source_id: i64) -> Result<(), String> {
 
 #[tauri::command(async)]
 fn delete_database() -> Result<(), String> {
-    sql::delete_database().map_err(map_err_frontend)
+    utils::create_nuke_request().map_err(map_err_frontend)
 }
 
 #[tauri::command(async)]

src-tauri/src/main.rs

@@ -1,7 +1,11 @@
 // Prevents additional console window on Windows in release, DO NOT REMOVE!!
 #![cfg_attr(not(debug_assertions), windows_subsystem = "windows")]
-use anyhow::Result;
+use anyhow::{Context, Result};
+use open_tv_lib::log;
 fn main() -> Result<()> {
+    _ = open_tv_lib::utils::check_nuke()
+        .with_context(|| "Failed to delete db after nuke request")
+        .inspect_err(|e| log::log(format!("{:?}", e)));
     open_tv_lib::sql::create_or_initialize_db()?;
     open_tv_lib::run();
     Ok(())

src-tauri/src/sql.rs

@@ -19,6 +19,7 @@ use rusqlite::{OptionalExtension, Row, Transaction, params, params_from_iter};
 use rusqlite_migration::{M, Migrations};
 
 const PAGE_SIZE: u8 = 36;
+pub const DB_NAME: &str = "db.sqlite";
 static CONN: LazyLock<Pool<SqliteConnectionManager>> = LazyLock::new(|| create_connection_pool());
 
 pub fn get_conn() -> Result<PooledConnection<SqliteConnectionManager>> {
@@ -38,7 +39,7 @@ fn get_and_create_sqlite_db_path() -> String {
     if !path.exists() {
         std::fs::create_dir_all(&path).unwrap();
     }
-    path.push("db.sqlite");
+    path.push(DB_NAME);
     return path.to_string_lossy().to_string();
 }
 
@@ -117,11 +118,6 @@ fn structure_exists() -> Result<bool> {
     Ok(table_exists)
 }
 
-pub fn delete_database() -> Result<()> {
-    std::fs::remove_file(get_and_create_sqlite_db_path())?;
-    std::process::exit(0);
-}
-
 pub fn create_or_initialize_db() -> Result<()> {
     if !structure_exists()? {
         create_structure()?;
@@ -213,6 +209,12 @@ fn apply_migrations() -> Result<()> {
               CREATE UNIQUE INDEX unique_seasons ON seasons(season_number, series_id, source_id);
             "#,
         ),
+        M::up(
+            r#"
+              DROP INDEX IF EXISTS channels_unique;
+              CREATE UNIQUE INDEX channels_unique ON channels(name, source_id, url, series_id, season_id);
+        "#,
+        ),
     ]);
     migrations.to_latest(&mut sql)?;
     Ok(())
@@ -280,7 +282,7 @@ pub fn insert_channel(tx: &Transaction, channel: Channel) -> Result<()> {
         r#"
 INSERT INTO channels (name, group_id, image, url, source_id, media_type, series_id, favorite, stream_id, tv_archive, season_id, episode_num)
 VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
-ON CONFLICT (name, source_id)
+ON CONFLICT (name, source_id, url, series_id, season_id)
 DO UPDATE SET
     url = excluded.url,
     media_type = excluded.media_type,

src-tauri/src/utils.rs

@@ -8,12 +8,14 @@ use crate::{
 };
 use anyhow::{Context, Result, anyhow, bail};
 use chrono::{DateTime, Local, Utc};
+use directories::ProjectDirs;
 use regex::Regex;
 use reqwest::Client;
 use serde::Serialize;
 use std::{
     env::{consts::OS, current_exe},
-    path::Path,
+    fs::File,
+    path::{Path, PathBuf},
     sync::{
         Arc, LazyLock,
         atomic::{AtomicBool, Ordering::Relaxed},
@@ -192,6 +194,34 @@ pub fn is_container() -> bool {
     std::env::var("container").is_ok()
 }
 
+pub fn create_nuke_request() -> Result<()> {
+    let path = get_nuke_path()?;
+    File::create(path)?;
+    std::process::exit(0);
+}
+
+fn get_nuke_path() -> Result<PathBuf> {
+    let path = ProjectDirs::from("dev", "fredol", "open-tv").context("project dir not found")?;
+    let path = path.cache_dir();
+    let path = path.join("nuke.txt");
+    Ok(path)
+}
+
+pub fn check_nuke() -> Result<()> {
+    let path = get_nuke_path()?;
+    if !path.exists() {
+        return Ok(());
+    }
+    std::fs::remove_file(path)?;
+    let path = ProjectDirs::from("dev", "fredol", "open-tv").context("project dir not found")?;
+    let path = path.data_dir();
+    let path = path.join(sql::DB_NAME);
+    if path.exists() {
+        std::fs::remove_file(path)?;
+    }
+    Ok(())
+}
+
 #[cfg(test)]
 mod test_utils {
     use super::sanitize;