using System;

using System.ComponentModel.DataAnnotations;

using System.IO;

using System.Threading.Tasks;

namespace SourceGit.ViewModels

{

public class AddWorktree : Popup

{

[CustomValidation(typeof(AddWorktree), nameof(ValidateWorktreePath))]

public string WorktreePath

{

get => _worktreePath;

set => SetProperty(ref _worktreePath, value, true);

}

[CustomValidation(typeof(AddWorktree), nameof(ValidateBranchName))]

public string BranchName

{

get => _branchName;

set => SetProperty(ref _branchName, value, true);

}

public AddWorktree(Repository repo)

{

_repo = repo;

_worktreePath = Path.Combine(repo.GetRecommandedWorktreeDir(), "new-worktree");

}

public static ValidationResult ValidateWorktreePath(string path, ValidationContext ctx)

{

if (ctx.ObjectInstance is not AddWorktree creator)

return new ValidationResult("Missing runtime context to add worktree!");

if (string.IsNullOrEmpty(path))

return new ValidationResult("Worktree path is required!");

var fullPath = Path.IsPathRooted(path) ? path : Path.Combine(creator._repo.FullPath, path);

var info = new DirectoryInfo(fullPath);

if (info.Exists)

{

var files = info.GetFiles();

if (files.Length > 0)

return new ValidationResult("Given path is not empty!!!");

var folders = info.GetDirectories();

if (folders.Length > 0)

return new ValidationResult("Given path is not empty!!!");

}

return ValidationResult.Success;

}

public static ValidationResult ValidateBranchName(string name, ValidationContext ctx)

{

if (ctx.ObjectInstance is not AddWorktree creator)

return new ValidationResult("Missing runtime context to add worktree!");

var test = !string.IsNullOrEmpty(name) ? name : Path.GetFileName(creator._worktreePath.TrimEnd('/').TrimEnd('\\'));

if (!Models.RefName.IsValidBranchName(test))

return new ValidationResult($"{test} is not a valid branch name!");

foreach (var b in creator._repo.Branches)

{

if (b.IsLocal && b.Name.Equals(test, StringComparison.Ordinal))

return new ValidationResult($"A branch with name '{test}' already exists!");

}

return ValidationResult.Success;

}

public override async Task<bool> Sure()

{

using var lockWatcher = _repo.LockWatcher();

ProgressDescription = $"Adding a new worktree ...";

var log = _repo.CreateLog("Add Worktree");

Use(log);

var succ = await new Commands.Worktree(_repo.FullPath)

.Use(log)

.AddAsync(_worktreePath, _branchName);

log.Complete();

return succ;

}

private readonly Repository _repo = null;

private string _worktreePath = string.Empty;

private string _branchName = string.Empty;

}

}