2024-10-30 16:27:32 -04:00
|
|
|
pub mod data;
|
|
|
|
|
pub mod downloader;
|
|
|
|
|
pub mod html;
|
2024-10-30 16:32:50 -04:00
|
|
|
pub mod parser;
|
|
|
|
|
|
|
|
|
|
use serde::Deserialize;
|
|
|
|
|
use std::convert::Infallible;
|
2024-10-31 18:16:42 -04:00
|
|
|
use std::io::Write;
|
2024-10-30 16:32:50 -04:00
|
|
|
use std::sync::{Arc, Mutex};
|
2024-10-31 18:16:42 -04:00
|
|
|
use reqwest::Url;
|
2024-10-30 16:32:50 -04:00
|
|
|
use touche::server::Service;
|
|
|
|
|
use touche::{Body, HttpBody, Request, Response, Server, StatusCode};
|
2024-10-31 18:16:42 -04:00
|
|
|
use zip::write::SimpleFileOptions;
|
|
|
|
|
use zip::ZipWriter;
|
2024-10-30 16:32:50 -04:00
|
|
|
use crate::downloader::download;
|
2024-10-31 18:22:23 -04:00
|
|
|
use crate::html::create_html;
|
2024-10-30 16:32:50 -04:00
|
|
|
use crate::parser::parse_search;
|
2024-10-31 18:16:42 -04:00
|
|
|
#[cfg(target_family = "wasm")]
|
|
|
|
|
use base64::prelude::*;
|
2024-10-31 18:35:50 -04:00
|
|
|
#[cfg(target_family = "wasm")]
|
|
|
|
|
use wasm_bindgen::prelude::wasm_bindgen;
|
2024-10-30 16:32:50 -04:00
|
|
|
|
|
|
|
|
const LODESTONE_HOST: &str = "https://na.finalfantasyxiv.com";
|
|
|
|
|
|
|
|
|
|
#[derive(Default, Deserialize, Clone)]
|
|
|
|
|
struct Package {
|
|
|
|
|
playtime: String,
|
|
|
|
|
height: i32,
|
|
|
|
|
bust_size: i32,
|
|
|
|
|
gil: u32,
|
|
|
|
|
is_battle_mentor: bool,
|
|
|
|
|
is_trade_mentor: bool,
|
|
|
|
|
is_novice: bool,
|
|
|
|
|
is_returner: bool,
|
|
|
|
|
player_commendations: i32,
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[derive(Clone)]
|
|
|
|
|
struct PackageService<'a> {
|
|
|
|
|
wants_stop: Arc<Mutex<bool>>, // TODO: THIS IS TERRIBLE STOP STOP STOP
|
|
|
|
|
package: &'a Arc<Mutex<Package>>,
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl Service for PackageService<'_> {
|
|
|
|
|
type Body = &'static str;
|
|
|
|
|
type Error = Infallible;
|
|
|
|
|
|
|
|
|
|
fn call(&self, req: Request<Body>) -> Result<Response<Self::Body>, Self::Error> {
|
|
|
|
|
*self.package.lock().unwrap() = serde_json::from_str(&String::from_utf8(req.into_body().into_bytes().unwrap()).unwrap()).unwrap();
|
|
|
|
|
|
|
|
|
|
*self.wants_stop.lock().unwrap() = true;
|
|
|
|
|
|
|
|
|
|
Ok(Response::builder()
|
|
|
|
|
.status(StatusCode::OK)
|
|
|
|
|
.body("")
|
|
|
|
|
.unwrap())
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// TODO: NO NO NO NO
|
|
|
|
|
fn wants_stop(&self) -> bool {
|
|
|
|
|
*self.wants_stop.lock().unwrap()
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2024-10-31 18:16:42 -04:00
|
|
|
/// Archives the character named `character_name` and gives a ZIP file as bytes that can be written to disk.
|
2024-10-31 19:33:45 -04:00
|
|
|
pub async fn archive_character(character_name: &str, use_dalamud: bool) -> Vec<u8> {
|
2024-10-31 18:16:42 -04:00
|
|
|
let search_page = download(&Url::parse_with_params(&format!("{LODESTONE_HOST}/lodestone/character?"), &[("q", character_name)]).unwrap())
|
|
|
|
|
.await
|
2024-10-30 16:32:50 -04:00
|
|
|
.expect("Failed to download the search page from the Lodestone.");
|
|
|
|
|
|
2024-10-31 18:16:42 -04:00
|
|
|
let href = parse_search(&String::from_utf8(search_page).unwrap());
|
2024-10-30 16:32:50 -04:00
|
|
|
if href.is_empty() {
|
|
|
|
|
println!("Unable to find character!");
|
|
|
|
|
}
|
|
|
|
|
|
2024-10-31 18:16:42 -04:00
|
|
|
let char_page = download(&Url::parse(&format!("{LODESTONE_HOST}{}", href)).unwrap())
|
|
|
|
|
.await
|
2024-10-30 16:32:50 -04:00
|
|
|
.expect("Failed to download the character page from the Lodestone.");
|
|
|
|
|
|
2024-10-31 18:16:42 -04:00
|
|
|
let mut char_data = crate::parser::parse_lodestone(&String::from_utf8(char_page).unwrap());
|
2024-10-30 16:32:50 -04:00
|
|
|
|
2024-10-31 18:16:42 -04:00
|
|
|
// 2 MiB, for one JSON and two images
|
|
|
|
|
let mut buf = vec![0; 2097152];
|
|
|
|
|
let mut zip = ZipWriter::new(std::io::Cursor::new(&mut buf[..]));
|
|
|
|
|
|
|
|
|
|
let options = SimpleFileOptions::default().compression_method(zip::CompressionMethod::Stored);
|
|
|
|
|
zip.start_file("character.json", options);
|
|
|
|
|
zip.write_all(serde_json::to_string(&char_data).unwrap().as_ref());
|
2024-10-30 16:32:50 -04:00
|
|
|
|
|
|
|
|
if !char_data.portrait_url.is_empty() {
|
2024-10-31 18:16:42 -04:00
|
|
|
let portrait_url = char_data.portrait_url.replace("img2.finalfantasyxiv.com", "img-tunnel.ryne.moe");
|
|
|
|
|
|
|
|
|
|
let portrait = download(&Url::parse(&portrait_url).unwrap())
|
|
|
|
|
.await
|
2024-10-30 16:32:50 -04:00
|
|
|
.expect("Failed to download the character portrait image.");
|
2024-10-31 18:16:42 -04:00
|
|
|
|
|
|
|
|
zip.start_file("portrait.jpg", options);
|
|
|
|
|
zip.write_all(&*portrait);
|
2024-10-30 16:32:50 -04:00
|
|
|
}
|
|
|
|
|
if !char_data.face_url.is_empty() {
|
2024-10-31 18:16:42 -04:00
|
|
|
let face_url = char_data.face_url.replace("img2.finalfantasyxiv.com", "img-tunnel.ryne.moe");
|
|
|
|
|
|
|
|
|
|
let face = download(&Url::parse(&face_url).unwrap())
|
|
|
|
|
.await
|
2024-10-30 16:32:50 -04:00
|
|
|
.expect("Failed to download the character face image.");
|
2024-10-31 18:16:42 -04:00
|
|
|
|
|
|
|
|
zip.start_file("face.jpg", options);
|
|
|
|
|
zip.write_all(&*face);
|
2024-10-30 16:32:50 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if use_dalamud {
|
|
|
|
|
println!("Now waiting for the Dalamud plugin. Type /auracite begin in chat.");
|
|
|
|
|
|
|
|
|
|
let package = Arc::new(Mutex::new(Package::default()));
|
|
|
|
|
|
|
|
|
|
Server::bind("0.0.0.0:8000").serve_single_thread(PackageService { wants_stop: Arc::new(Mutex::new(false)), package: &package }).unwrap();
|
|
|
|
|
|
|
|
|
|
let package = &*package.lock().unwrap();
|
|
|
|
|
|
|
|
|
|
char_data.playtime = package.playtime.parse().unwrap();
|
|
|
|
|
char_data.appearance.height = package.height;
|
|
|
|
|
char_data.appearance.bust_size = package.bust_size;
|
|
|
|
|
char_data.currencies.gil = package.gil; // TODO: also fetch from the lodestone
|
|
|
|
|
char_data.is_battle_mentor = package.is_battle_mentor;
|
|
|
|
|
char_data.is_trade_mentor = package.is_trade_mentor;
|
|
|
|
|
char_data.is_novice = package.is_novice;
|
|
|
|
|
char_data.is_returner = package.is_returner;
|
|
|
|
|
char_data.player_commendations = package.player_commendations; // TODO: fetch from the lodestone?
|
|
|
|
|
}
|
|
|
|
|
|
2024-10-31 18:20:45 -04:00
|
|
|
let html = create_html(
|
|
|
|
|
&char_data
|
|
|
|
|
);
|
|
|
|
|
|
|
|
|
|
zip.start_file("character.html", options);
|
|
|
|
|
zip.write_all(html.as_ref());
|
|
|
|
|
|
2024-10-31 18:16:42 -04:00
|
|
|
zip.finish();
|
2024-10-30 16:32:50 -04:00
|
|
|
|
2024-10-31 18:16:42 -04:00
|
|
|
return buf;
|
2024-10-30 16:32:50 -04:00
|
|
|
}
|
2024-10-31 18:16:42 -04:00
|
|
|
|
|
|
|
|
/// Archives the character named `character_name` and converts the ZIP file to Base64. Useful for downloading via data URIs.
|
|
|
|
|
#[cfg(target_family = "wasm")]
|
|
|
|
|
#[wasm_bindgen]
|
|
|
|
|
pub async extern fn archive_character_base64(character_name: &str, use_dalamud: bool) -> String {
|
|
|
|
|
let buf = archive_character(character_name, use_dalamud).await;
|
|
|
|
|
|
|
|
|
|
let base64 = BASE64_STANDARD.encode(buf);
|
|
|
|
|
return format!("data:application/octet-stream;charset=utf-16le;base64,{base64}").into();
|
|
|
|
|
}
|
