Add 'upload' command to publish places to Roblox for you

server/src/commands/mod.rs

@@ -1,7 +1,9 @@
 mod serve;
 mod init;
 mod build;
+mod upload;
 
 pub use self::serve::*;
 pub use self::init::*;
-pub use self::build::*;
+pub use self::build::*;
+pub use self::upload::*;

server/src/commands/upload.rs

@@ -0,0 +1,89 @@
+use std::{
+    path::PathBuf,
+    io,
+};
+
+use failure::Fail;
+
+use reqwest::header::{USER_AGENT, CONTENT_TYPE, COOKIE};
+
+use crate::{
+    rbx_session::construct_oneoff_tree,
+    project::{Project, ProjectLoadFuzzyError},
+    imfs::Imfs,
+};
+
+#[derive(Debug, Fail)]
+pub enum UploadError {
+    #[fail(display = "Roblox API Error: {}", _0)]
+    RobloxApiError(String),
+
+    #[fail(display = "Project load error: {}", _0)]
+    ProjectLoadError(#[fail(cause)] ProjectLoadFuzzyError),
+
+    #[fail(display = "IO error: {}", _0)]
+    IoError(#[fail(cause)] io::Error),
+
+    #[fail(display = "HTTP error: {}", _0)]
+    HttpError(#[fail(cause)] reqwest::Error),
+}
+
+impl From<ProjectLoadFuzzyError> for UploadError {
+    fn from(error: ProjectLoadFuzzyError) -> UploadError {
+        UploadError::ProjectLoadError(error)
+    }
+}
+
+impl From<io::Error> for UploadError {
+    fn from(error: io::Error) -> UploadError {
+        UploadError::IoError(error)
+    }
+}
+
+impl From<reqwest::Error> for UploadError {
+    fn from(error: reqwest::Error) -> UploadError {
+        UploadError::HttpError(error)
+    }
+}
+
+#[derive(Debug)]
+pub struct UploadOptions {
+    pub fuzzy_project_path: PathBuf,
+    pub security_cookie: String,
+    pub place_id: u64,
+}
+
+pub fn upload(options: &UploadOptions) -> Result<(), UploadError> {
+    info!("Looking for project at {}", options.fuzzy_project_path.display());
+
+    let project = Project::load_fuzzy(&options.fuzzy_project_path)?;
+
+    info!("Found project at {}", project.file_location.display());
+    info!("Using project {:#?}", project);
+
+    let mut imfs = Imfs::new();
+    imfs.add_roots_from_project(&project)?;
+    let tree = construct_oneoff_tree(&project, &imfs);
+
+    let root_id = tree.get_root_id();
+    let top_level_ids = tree.get_instance(root_id).unwrap().get_children_ids();
+    let mut contents = Vec::new();
+    rbx_xml::encode(&tree, top_level_ids, &mut contents);
+
+    let url = format!("https://data.roblox.com/Data/Upload.ashx?json=1&type=Place&assetid={}", options.place_id);
+
+    let client = reqwest::Client::new();
+    let mut response = client.post(&url)
+        .header(COOKIE, format!(".ROBLOSECURITY={}", &options.security_cookie))
+        .header(USER_AGENT, "Roblox/WinInet")
+        .header("Requester", "Client")
+        .header(CONTENT_TYPE, "application/xml")
+        .body(contents)
+        .send()?;
+
+    if !response.status().is_success() {
+        return Err(UploadError::RobloxApiError(response.text()?));
+    }
+
+    Ok(())
+}