use super::Command;
use crate::error::{CommandError, ExecutionError};
use crate::git::GitCommandExecuterBuilder;
use crate::utils;
use clap::{Args, Subcommand};
use regex::Regex;
use std::fs::File;
use std::io::{Read, Write};
use std::path::Path;
use thiserror::Error;
#[derive(Debug, Error)]
pub enum Error {
#[error(
"Remote repository already added if you want to change that you need to reset your dfmn"
)]
AlreadyAdded,
#[error("Remote repository not setted yet")]
NotSetted,
#[error("Not a ssh link")]
NotSSH,
#[error("Invalid link for the remote repository")]
InvalidLink,
#[error("No internet connection")]
NoInternetConnection,
}
impl From<Error> for CommandError {
fn from(err: Error) -> Self {
CommandError::Usage(err.to_string())
}
}
#[derive(Debug, Args)]
pub struct Remote {
#[command(subcommand)]
subcommands: Subcommands,
}
#[derive(Debug, Args)]
pub struct Add {
link: String,
}
#[derive(Debug, Subcommand)]
pub enum Subcommands {
Show,
Add(Add),
}
fn add_remote_link(
storage_folder_path: &Path,
git_storage_folder_path: &Path,
link: &str,
) -> Result<String, CommandError> {
if online::check(None).is_err() {
return Err(Error::NoInternetConnection.into());
}
if storage_folder_path.join("remote.txt").exists() {
return Err(Error::AlreadyAdded.into());
}
let regex = match Regex::new(r"git@git((hub)|(lab))\.com:\S*/\S*\.git") {
Ok(regex) => regex,
Err(err) => {
return Err(ExecutionError::Regex(err.to_string()).into());
}
};
if !regex.is_match(link) {
return Err(Error::NotSSH.into());
}
if let Err(err) = GitCommandExecuterBuilder::new(git_storage_folder_path)
.run_remote_add(link)
.run_pull()
.build()
.run()
{
GitCommandExecuterBuilder::new(git_storage_folder_path)
.run_remote_remove()
.build()
.run()?;
if matches!(err, ExecutionError::RepositoryNotFound) {
return Err(Error::InvalidLink.into());
}
return Err(err.into());
}
let mut file = match File::create(storage_folder_path.join("remote.txt")) {
Ok(file) => file,
Err(err) => {
return Err(ExecutionError::CreateFile(err.to_string()).into());
}
};
if let Err(err) = write!(file, "{link}") {
return Err(ExecutionError::WriteToFile(err.to_string()).into());
}
Ok("Successfully setted the remote repository and synchronized the local repository with the remote repository".to_string())
}
fn show_remote_link(storage_folder_path: &Path) -> Result<String, CommandError> {
if !storage_folder_path.join("remote.txt").exists() {
return Err(Error::NotSetted.into());
}
let mut file = match File::open(storage_folder_path.join("remote.txt")) {
Ok(file) => file,
Err(err) => {
return Err(ExecutionError::OpenFile(err.to_string()).into());
}
};
let mut content = Vec::new();
if let Err(err) = file.read_to_end(&mut content) {
return Err(ExecutionError::ReadFile(err.to_string()).into());
}
let Ok(content) = String::from_utf8(content) else {
return Err(ExecutionError::InvalidUTF8("Convert Vec<u8> to String").into());
};
Ok(content)
}
impl Command for Remote {
fn execute(self) -> Result<String, CommandError> {
let storage_folder_path = match utils::get_dfm_folder_path() {
Ok(path) => path,
Err(err) => {
return Err(ExecutionError::GetStorageFolderPath(err.to_string()).into());
}
};
let storage_folder_path = match storage_folder_path.canonicalize() {
Ok(path) => path,
Err(err) => {
return Err(ExecutionError::CanonicalizePath(err.to_string()).into());
}
};
match self.subcommands {
Subcommands::Show => show_remote_link(&storage_folder_path),
Subcommands::Add(add) => add_remote_link(
&storage_folder_path,
&storage_folder_path.join("dotfiles"),
&add.link,
),
}
}
}