path: root/src/main.rs

use clap::{Parser, Subcommand};
use std::path::{Path, PathBuf};
use std::fs::write;
use std::process::exit;
use std::fs;
use serde::{Serialize, Deserialize};

mod content;
mod utils;

#[derive(Parser)]
struct Cli {
		#[clap(subcommand)]
		command: Commands,
}

#[derive(Subcommand)]
enum Commands {
		/// Initialize Godot project for GLAM
		Init {
		},

		/// Install new GLAM package
		Install {
				/// Package project git
				git_repo: String,
				/// Verbose (output subshell commands)
				#[clap(short, long, takes_value = false)]
				verbose: bool
		},

		/// Update all GLAM packages
		Update {
		},

		/// Remove a GLAM package
		Remove {
		},
}

#[derive(Serialize, Deserialize)]
struct GlamObject {
		packages: Vec<GlamPackage>
}

// TODO: Add commit hash
#[derive(Clone, Debug, Serialize, Deserialize)]
struct GlamPackage {
		name : String,
		git_repo : String,
		commit : String,
}

fn main() {
		let cli = Cli::parse();

		match &cli.command {
				Commands::Init {} => {
						initialize();
				},

				Commands::Install {git_repo, verbose} => {
						check_initialization();
						install_addon(git_repo, *verbose);
				},

				Commands::Update {} => {
						check_initialization();
						update_addon();
				},

				Commands::Remove {} => {
						check_initialization();
						remove_addon();
				},
		}
}

fn initialize() {
		let root = search_project_root();

		let git_ignore = &format!("{}/.gitignore", &root);
		if !Path::new(git_ignore).exists() {
				match write(git_ignore, content::create_gitignore_file()) {
						Ok(_v) => (),
						Err(_e) => {
								utils::log_error("There was a problem creating the .gitignore file!");
								exit(1);
						}
				}
		}

		let gd_ignore = &format!("{}/.gdignore", &root);
		if !Path::new(gd_ignore).exists() {
				match write(gd_ignore, content::create_gdignore_file()) {
						Ok(_v) => (),
						Err(_e) => {
								utils::log_error("There was a problem creating the .gdignore file!");
								exit(1);
						}
				}
		}
}

fn check_initialization() {
		let root = search_project_root();

		let git_ignore = &format!("{}/.gitignore", &root);
		if !Path::new(git_ignore).exists() {
				utils::log_warning(".gitignore file does not exist!");
		}

		let gd_ignore = &format!("{}/.gdignore", &root);
		if !Path::new(gd_ignore).exists() {
				utils::log_warning(".gdignore file does not exist!");
		}
}

fn install_addon(git_repo : &str, verbose : bool) {
		// Search for project root folder
		let root = search_project_root();
		utils::log_check(&format!("Found root project in: {}", root));

		// Create glam.d/ folder if it doesn't exist
		if !Path::new(&format!("{}/.glam.d/", root)).exists() {
				utils::run_shell_command(
						&format!("mkdir -p {}/.glam.d/", root),
						&root,
						false
				);
				utils::log_info("Created .glam.d/ folder");
		}

		// Create .glam file if it doesn't exist
		if !Path::new(&format!("{}/.glam", root)).exists() {
				fs::write(&format!("{}/.glam",root),
									content::create_glam_file())
						.expect("Couldn't create .glam file!");
				utils::log_info("Created .glam file");
		}

		// Find glam object or create one with default configuration
		let mut glam_object = read_glam_file();
		let mut glam_packages = glam_object.packages;
		let mut package_index = 0;
		let mut found_package = false;

//		let mut target_package : Option<GlamPackage> = None;
		let name = utils::get_repo_name(git_repo);

		for (i, package) in glam_packages.iter().enumerate() {
				if package.name == name {
						package_index = i;
						found_package = true;
				}
		}

		if !found_package {
				let package = GlamPackage{
						git_repo : git_repo.to_string(),
						name : name.to_string(),
						commit : "".to_string(),
				};

				glam_packages.push(package);
				package_index = glam_packages.len() - 1;
		}

		let target_package = &mut glam_packages[package_index];

		// If glam package folder doesn't exist, clone project
		if !Path::new(&format!("{}/.glam.d/{}", root, target_package.name)).exists() {
				utils::run_shell_command(
						&format!("cd .glam.d/ && git clone {} {} --progress", target_package.git_repo, target_package.name),
						&root,
						verbose
				);
				utils::log_check("Created package folder on .glam.d");
		} else {
				utils::log_info("Glam package folder already exists");
		}

		// Update package folder to commit hash
		if target_package.commit == "" {
				let res = utils::run_shell_command(
						&format!("cd .glam.d/{} && git rev-parse HEAD",
										 target_package.name),
						&root,
						verbose).unwrap();
				target_package.commit = res.trim().to_string();
		} else {
				utils::log_info("Git checkout to package commit");
				utils::run_shell_command(
						&format!("cd .glam.d/{} && git checkout {}",
										 target_package.name,
										 target_package.commit),
						&root,
						verbose);
		}

		// If project addon folder doesn't exist, create it
		utils::run_shell_command(
				&format!("mkdir -p {}/addons/{}", root, name),
				&root,
				verbose
		);

		// TODO: use source_folder to copy files from (default: /addons/)
		// TODO: use target_folder to copy files to (defautl: (root)/)
		// Copy addon repository content to target folder
		utils::run_shell_command(
				&format!("cp -f -r .glam.d/{}/addons/* -t {}/addons/", name, root),
				&root,
				verbose
		);

		// Write .glam file
		glam_object.packages = glam_packages;
		write_glam_file(&glam_object);
}

// TODO: Use root folder
fn read_glam_file() -> GlamObject {
		if !Path::new("./.glam").exists() {
				fs::write("./.glam", content::create_glam_file()).expect("Couldn't create .glam file!");
		}

		let glam_content = fs::read_to_string("./.glam").expect("Couldn't read .glam file!");
		let glam_obj : GlamObject = serde_json::from_str(&glam_content).unwrap();

		return glam_obj;
}

// TODO: Use root folder
fn write_glam_file(glam_object : &GlamObject) {
		let json_string = serde_json::to_string_pretty(glam_object).unwrap();
		fs::write("./.glam", json_string).expect("Couldn't create .glam file!");
}

fn update_addon() {
}

fn remove_addon() {
}

fn search_project_root() -> String{
		let path = PathBuf::from("./");
		let mut dir = path.canonicalize().unwrap();

		loop {
				let dir_path = dir.to_str().unwrap();
				let proj_path = format!("{}/project.godot", dir_path);

				let godot_project = Path::new(&proj_path);

				if godot_project.exists() {
						break;
				}
				else {
						let parent = dir.parent();
						if parent.is_none() {
								utils::log_error("Godot project not found!");
								exit(1);
						}
						dir = dir.parent().unwrap().to_path_buf();
				}
		}

		return dir.to_str().unwrap().to_string();
}