quick setup command

1 files changed, 77 insertions, 2 deletions

diff --git a/import/src/main.rs b/import/src/main.rs
index 2d66221..26899bb 100644
--- a/import/src/main.rs
+++ b/import/src/main.rs
@@ -3,19 +3,23 @@
     which is licensed under the GNU Affero General Public License (version 3); see /COPYING.
     Copyright (C) 2023 metamuffin <metamuffin.org>
 */
+#![feature(file_create_new)]
+
 pub mod infojson;
 pub mod tmdb;
 
 use anyhow::Context;
+use base64::Engine;
 use clap::{Parser, Subcommand};
 use infojson::YVideo;
 use jellycommon::{
-    AssetLocation, LocalTrack, MediaInfo, MediaSource, Node, NodeKind, NodePrivate, NodePublic,
-    Rating,
+    config::GlobalConfig, AssetLocation, LocalTrack, MediaInfo, MediaSource, Node, NodeKind,
+    NodePrivate, NodePublic, Rating,
 };
 use jellymatroska::read::EbmlReader;
 use jellyremuxer::import::{import_metadata, seek_index};
 use log::{info, warn};
+use rand::random;
 use std::{
     fs::{remove_file, File},
     io::{stdin, Write},
@@ -34,22 +38,39 @@ struct Args {
 
 #[derive(Subcommand)]
 enum Action {
+    /// Initialize a new jellything instance
+    Init {
+        /// Base path of the instance, must either be absolute or relative to the servers pwd
+        base_path: PathBuf,
+        #[arg(short, long)]
+        brand: String,
+    },
+    /// Imports a movie, video or series given media and metadata sources
     New {
+        /// Relative path to the node's parent(!).
         path: PathBuf,
+        /// Search the node by title on TMDB
         #[arg(short = 't', long)]
         tmdb_search: Option<String>,
+        /// Search the node by id on TMDB
         #[arg(short = 'T', long)]
         tmdb_id: Option<String>,
         #[arg(long)]
+        /// Prefix the inferred id with something to avoid collisions
         ident_prefix: Option<String>,
+        /// Copies media into the library
         #[arg(long)]
         copy: bool,
+        /// Moves media into the library (potentially destructive operation)
         #[arg(long)]
         r#move: bool,
+        /// Marks node as a video
         #[arg(long)]
         video: bool,
+        /// Path to the media of the node, required for non-series
         #[arg(short, long)]
         input: Option<PathBuf>,
+        /// Marks node as a series
         #[arg(short, long)]
         series: bool,
     },
@@ -63,6 +84,56 @@ fn main() -> anyhow::Result<()> {
     let args = Args::parse();
 
     match args.action {
+        Action::Init {
+            base_path: path,
+            brand,
+        } => {
+            info!("creating new instance...");
+            std::fs::create_dir_all(path.join("library"))?;
+            std::fs::create_dir_all(path.join("cache"))?;
+            std::fs::create_dir_all(path.join("assets"))?;
+            File::create_new(path.join("assets/front.htm"))?
+                .write_fmt(format_args!("<h1>My very own jellything instance</h1>"))?;
+            serde_json::to_writer(
+                File::create_new(path.join("config.json"))?,
+                &GlobalConfig {
+                    brand: brand.clone(),
+                    slogan: "Creative slogan here".to_string(),
+                    asset_path: path.join("assets"),
+                    cache_path: path.join("cache"),
+                    library_path: path.join("library"),
+                    database_path: path.join("database"),
+                    cookie_key: Some(
+                        base64::engine::general_purpose::STANDARD
+                            .encode([(); 32].map(|_| random())),
+                    ),
+                    session_key: Some(
+                        base64::engine::general_purpose::STANDARD
+                            .encode([(); 32].map(|_| random())),
+                    ),
+                    admin_username: "admin".to_string(),
+                    admin_password: "hackme".to_string(),
+                    login_expire: 10,
+                    ..Default::default()
+                },
+            )?;
+            serde_json::to_writer(
+                File::create_new(path.join("library/directory.json"))?,
+                &Node {
+                    public: NodePublic {
+                        kind: NodeKind::Collection,
+                        title: "My Library".to_string(),
+                        ..Default::default()
+                    },
+                    private: NodePrivate {
+                        ..Default::default()
+                    },
+                },
+            )?;
+            info!("{brand:?} is ready!");
+            warn!("please change the admin password.");
+            Ok(())
+        }
         Action::New {
             path,
             tmdb_id,
@@ -74,6 +145,10 @@ fn main() -> anyhow::Result<()> {
             video,
             r#move,
         } => {
+            if std::env::current_dir().unwrap().file_name().unwrap() != "library" {
+                warn!("new command can only be used in the library directory; what you are doing right now probably wont work.")
+            }
+
             let tmdb_kind = if series { "tv" } else { "movie" };
             let tmdb_id = if let Some(id) = tmdb_id {
                 Some(id.parse().unwrap())