Initial Commit

entry/src/main_loop.rs

@@ -0,0 +1,91 @@
+use crate::data::{FetchResult, State};
+use crate::download::Downloader;
+use crate::github::{fetch_last_hash, sync_data};
+use crate::{CONFIG, PROVIDE, STATE, STORE};
+use anyhow::anyhow;
+use assets_updater::core::export_pipeline::find_files;
+use std::fs;
+use log::debug;
+
+pub async fn main() -> anyhow::Result<()> {
+    let current_hash = fetch_last_hash().await?;
+    let old_state = {
+        let mut l = STATE.read();
+        if l.is_err() {
+            STATE.clear_poison();
+            drop(l);
+            let mut s = STATE.write().map_err(|e| anyhow!("{}", e))?;
+            *s = State::new();
+            let _ = s.init();
+            drop(s);
+            l = STATE.read();
+        }
+        if let Ok(state) = l {
+            state.clone()
+        } else {
+            State::new()
+        }
+    };
+    if let Some(old_hash) = &old_state.git_hash
+        && old_hash == &current_hash
+    {
+        return Ok(());
+    }
+    // hash updated
+
+    sync_data().await?;
+    let store = STORE.read().map_err(|e| anyhow!("{}", e))?;
+    let event = store.find_last_event();
+    if event
+        .as_ref()
+        .zip(old_state.event_id.as_ref())
+        .is_none_or(|(e, id)| &e.event_id == id)
+    {
+        let mut state = STATE.write().map_err(|e| anyhow!("{}", e))?;
+        state.git_hash = Some(current_hash);
+        if let Some(event) = event {
+            state.event_id = Some(event.event_id);
+        }
+        state.save();
+        return Ok(());
+    }
+    let event = event.unwrap();
+    // event updated
+
+    let cards = store.find_cards_by_event(event.event_id);
+    drop(store);
+    let downloader = Downloader::new().await?;
+    let mut images = Vec::new();
+    let cache_dir = CONFIG.get().unwrap().cache_dir();
+    for card in cards {
+        debug!("Pulling card {} from bundle {}", card.card_id, card.assetbundle_name);
+        let dir = downloader
+            .pull(&format!("character/member/{}", card.assetbundle_name))
+            .await?;
+        let files = find_files(&dir.0)?;
+        for file in files {
+            if file.is_file() && file.extension().map(|ext| ext == "png").unwrap_or(false) {
+                let new_filename = format!(
+                    "{}_{}",
+                    card.assetbundle_name,
+                    file.file_name().unwrap().to_str().unwrap()
+                );
+                let new_file = cache_dir.join(new_filename);
+                fs::copy(&file, &new_file)?;
+                fs::remove_file(&file)?;
+                images.push(new_file);
+            }
+        }
+    }
+
+    *PROVIDE.write().map_err(|e| anyhow!("{}", e))? = Some(FetchResult {
+        event_id: event.event_id,
+        card_paths: images,
+    });
+    let mut state = STATE.write().map_err(|e| anyhow!("{}", e))?;
+    state.git_hash = Some(current_hash);
+    state.event_id = Some(event.event_id);
+    state.save();
+
+    Ok(())
+}