You've already forked AstralRinth
forked from didirus/AstralRinth
File types (#506)
* File types * Run prepare + fmt * Switch to struct * Update docker version
This commit is contained in:
Geometrically
@@ -1,276 +0,0 @@
|
||||
use crate::database::models::categories::GameVersion;
|
||||
use crate::file_hosting::FileHostingError;
|
||||
use crate::routes::project_creation::CreateError;
|
||||
use chrono::{DateTime, Duration, Utc};
|
||||
use serde::{Deserialize, Serialize};
|
||||
use std::collections::HashMap;
|
||||
|
||||
#[derive(Deserialize, Debug, Clone)]
|
||||
struct FlameGameVersionType {
|
||||
id: i32,
|
||||
slug: String,
|
||||
}
|
||||
|
||||
#[derive(Deserialize, Debug, Clone)]
|
||||
struct FlameGameVersion {
|
||||
id: i32,
|
||||
#[serde(rename = "gameVersionTypeID")]
|
||||
game_version_type_id: i32,
|
||||
name: String,
|
||||
slug: String,
|
||||
}
|
||||
|
||||
#[derive(Serialize, Debug)]
|
||||
#[serde(rename_all = "camelCase")]
|
||||
struct FlameUploadFile {
|
||||
changelog: String,
|
||||
// always "markdown"
|
||||
changelog_type: String,
|
||||
display_name: String,
|
||||
game_versions: Vec<i32>,
|
||||
release_type: String,
|
||||
// TODO: relations?
|
||||
}
|
||||
|
||||
pub struct FlameAnvilQueue {
|
||||
mod_loaders: Vec<FlameGameVersion>,
|
||||
minecraft_versions: Vec<FlameGameVersion>,
|
||||
bukkit_versions: Vec<FlameGameVersion>,
|
||||
last_updated: DateTime<Utc>,
|
||||
}
|
||||
|
||||
pub struct UploadFile {
|
||||
pub loaders: Vec<String>,
|
||||
pub game_versions: Vec<String>,
|
||||
pub display_name: String,
|
||||
pub changelog: String,
|
||||
pub version_type: String,
|
||||
}
|
||||
|
||||
// Batches download transactions every thirty seconds
|
||||
impl FlameAnvilQueue {
|
||||
pub fn new() -> Self {
|
||||
FlameAnvilQueue {
|
||||
mod_loaders: vec![],
|
||||
minecraft_versions: vec![],
|
||||
bukkit_versions: vec![],
|
||||
last_updated: Utc::now() - Duration::days(365),
|
||||
}
|
||||
}
|
||||
|
||||
fn convert_game_versions_to_flame(
|
||||
&self,
|
||||
original: Vec<String>,
|
||||
flame_game_versions: &[FlameGameVersion],
|
||||
game_versions: &[GameVersion],
|
||||
) -> Vec<i32> {
|
||||
let mut og_to_flame = HashMap::new();
|
||||
let mut last_visited = if flame_game_versions
|
||||
.last()
|
||||
.map(|x| x.name.ends_with("-Snapshot"))
|
||||
.unwrap_or_default()
|
||||
{
|
||||
None
|
||||
} else {
|
||||
flame_game_versions
|
||||
.iter()
|
||||
.rfind(|x| !x.name.ends_with("-Snapshot"))
|
||||
.cloned()
|
||||
};
|
||||
|
||||
for game_version in game_versions {
|
||||
if let Some(flame_game_version) =
|
||||
flame_game_versions.iter().find(|x| {
|
||||
x.name
|
||||
== if game_version.version.starts_with('b') {
|
||||
game_version.version.replace('b', "Beta ")
|
||||
} else {
|
||||
game_version.version.clone()
|
||||
}
|
||||
})
|
||||
{
|
||||
last_visited = Some(flame_game_version.clone());
|
||||
og_to_flame
|
||||
.insert(&game_version.version, flame_game_version.id);
|
||||
} else if let Some(last_visited) = &last_visited {
|
||||
if game_version.major {
|
||||
og_to_flame.insert(&game_version.version, last_visited.id);
|
||||
} else {
|
||||
let mut splits = last_visited.name.split('.');
|
||||
let new_str = format!(
|
||||
"{}.{}-Snapshot",
|
||||
splits.next().unwrap_or_default(),
|
||||
splits.next().unwrap_or_default()
|
||||
);
|
||||
|
||||
if let Some(flame_game_version) =
|
||||
flame_game_versions.iter().find(|x| x.name == new_str)
|
||||
{
|
||||
og_to_flame.insert(
|
||||
&game_version.version,
|
||||
flame_game_version.id,
|
||||
);
|
||||
} else {
|
||||
og_to_flame
|
||||
.insert(&game_version.version, last_visited.id);
|
||||
}
|
||||
}
|
||||
} else if let Some(last) = flame_game_versions.last() {
|
||||
og_to_flame.insert(&game_version.version, last.id);
|
||||
}
|
||||
}
|
||||
|
||||
let mut new = Vec::new();
|
||||
|
||||
for x in original {
|
||||
if let Some(value) = og_to_flame.get(&&x) {
|
||||
new.push(*value);
|
||||
}
|
||||
}
|
||||
|
||||
new
|
||||
}
|
||||
|
||||
#[allow(clippy::too_many_arguments)]
|
||||
pub async fn upload_file(
|
||||
&mut self,
|
||||
api_token: &str,
|
||||
project_id: i32,
|
||||
upload_file: UploadFile,
|
||||
game_versions: &[GameVersion],
|
||||
file: Vec<u8>,
|
||||
file_name: String,
|
||||
mime_type: String,
|
||||
is_plugin: bool,
|
||||
) -> Result<i32, CreateError> {
|
||||
if self.last_updated < (Utc::now() - Duration::minutes(30)) {
|
||||
self.index(api_token).await.map_err(|_| {
|
||||
CreateError::InvalidInput(
|
||||
"Indexing metadata from FlameAnvil failed!".to_string(),
|
||||
)
|
||||
})?;
|
||||
}
|
||||
|
||||
let mut loaders_converted = upload_file
|
||||
.loaders
|
||||
.into_iter()
|
||||
.filter_map(|x| self.mod_loaders.iter().find(|y| y.slug == x))
|
||||
.map(|x| x.id)
|
||||
.collect::<Vec<i32>>();
|
||||
|
||||
let mut game_versions_converted = self.convert_game_versions_to_flame(
|
||||
upload_file.game_versions,
|
||||
if is_plugin {
|
||||
&self.bukkit_versions
|
||||
} else {
|
||||
&self.minecraft_versions
|
||||
},
|
||||
game_versions,
|
||||
);
|
||||
|
||||
loaders_converted.append(&mut game_versions_converted);
|
||||
|
||||
let file = reqwest::multipart::Part::bytes(file)
|
||||
.file_name(file_name)
|
||||
.mime_str(&mime_type)
|
||||
.map_err(|_| {
|
||||
CreateError::InvalidInput(
|
||||
"Error while converting inputted file to multipart payload"
|
||||
.to_string(),
|
||||
)
|
||||
})?;
|
||||
|
||||
let form = reqwest::multipart::Form::new()
|
||||
.text(
|
||||
"metadata",
|
||||
serde_json::to_string(&FlameUploadFile {
|
||||
changelog: upload_file.changelog,
|
||||
changelog_type: "markdown".to_string(),
|
||||
display_name: upload_file.display_name,
|
||||
game_versions: loaders_converted,
|
||||
release_type: upload_file.version_type,
|
||||
})
|
||||
.unwrap(),
|
||||
)
|
||||
.part("file", file);
|
||||
|
||||
#[derive(Deserialize)]
|
||||
struct FileResponse {
|
||||
id: i32,
|
||||
}
|
||||
|
||||
let client = reqwest::Client::new();
|
||||
|
||||
let id = client.post(&*format!("https://minecraft.curseforge.com/api/projects/{project_id}/upload-file?token={api_token}"))
|
||||
.multipart(form)
|
||||
.send()
|
||||
.await.map_err(|_| CreateError::FileHostingError(FileHostingError::S3Error("Error uploading file to FlameAnvil!".to_string())))?
|
||||
.json::<FileResponse>()
|
||||
.await.map_err(|_| CreateError::FileHostingError(FileHostingError::S3Error("Error deserializing uploaded file response from FlameAnvil!".to_string())))?;
|
||||
|
||||
Ok(id.id)
|
||||
}
|
||||
|
||||
pub async fn index(
|
||||
&mut self,
|
||||
api_token: &str,
|
||||
) -> Result<(), reqwest::Error> {
|
||||
let (game_versions, game_version_types, bukkit_game_versions, bukkit_game_versions_types) = futures::future::try_join4(
|
||||
reqwest::get(format!("https://minecraft.curseforge.com/api/game/versions?token={api_token}")),
|
||||
reqwest::get(format!("https://minecraft.curseforge.com/api/game/version-types?token={api_token}")),
|
||||
reqwest::get(format!("https://dev.bukkit.org/api/game/versions?token={api_token}")),
|
||||
reqwest::get(format!("https://dev.bukkit.org/api/game/version-types?token={api_token}"))
|
||||
).await?;
|
||||
|
||||
let (
|
||||
game_versions,
|
||||
game_version_types,
|
||||
bukkit_game_versions,
|
||||
bukkit_game_versions_types,
|
||||
) = futures::future::try_join4(
|
||||
game_versions.json::<Vec<FlameGameVersion>>(),
|
||||
game_version_types.json::<Vec<FlameGameVersionType>>(),
|
||||
bukkit_game_versions.json::<Vec<FlameGameVersion>>(),
|
||||
bukkit_game_versions_types.json::<Vec<FlameGameVersionType>>(),
|
||||
)
|
||||
.await?;
|
||||
|
||||
let mod_loader_types = game_version_types
|
||||
.iter()
|
||||
.filter(|x| x.slug == *"modloader")
|
||||
.map(|x| x.id)
|
||||
.collect::<Vec<_>>();
|
||||
let minecraft_types = game_version_types
|
||||
.iter()
|
||||
.filter(|x| x.slug.starts_with("minecraft"))
|
||||
.map(|x| x.id)
|
||||
.collect::<Vec<_>>();
|
||||
let bukkit_types = bukkit_game_versions_types
|
||||
.iter()
|
||||
.filter(|x| x.slug.starts_with("bukkit"))
|
||||
.map(|x| x.id)
|
||||
.collect::<Vec<_>>();
|
||||
|
||||
let mod_loaders = game_versions
|
||||
.iter()
|
||||
.filter(|x| mod_loader_types.contains(&x.game_version_type_id))
|
||||
.cloned()
|
||||
.collect::<Vec<_>>();
|
||||
let minecraft_versions = game_versions
|
||||
.iter()
|
||||
.filter(|x| minecraft_types.contains(&x.game_version_type_id))
|
||||
.cloned()
|
||||
.collect::<Vec<_>>();
|
||||
let bukkit_versions = bukkit_game_versions
|
||||
.iter()
|
||||
.filter(|x| bukkit_types.contains(&x.game_version_type_id))
|
||||
.cloned()
|
||||
.collect::<Vec<_>>();
|
||||
|
||||
self.mod_loaders = mod_loaders;
|
||||
self.minecraft_versions = minecraft_versions;
|
||||
self.bukkit_versions = bukkit_versions;
|
||||
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
@@ -1,3 +1,2 @@
|
||||
pub mod download;
|
||||
pub mod flameanvil;
|
||||
pub mod payouts;
|
||||
|
||||
Reference in New Issue
Block a user