Flow.Launcher/Plugins/Flow.Launcher.Plugin.Explorer/Views/QuickAccessLinkSettings.xaml.cs

using System;
using System.Collections.ObjectModel;
using System.ComponentModel;
using System.IO;
using System.Linq;
using System.Windows;
using System.Windows.Forms;
using Flow.Launcher.Plugin.Explorer.Helper;
using Flow.Launcher.Plugin.Explorer.Search;
using Flow.Launcher.Plugin.Explorer.Search.QuickAccessLinks;
using CommunityToolkit.Mvvm.ComponentModel;

namespace Flow.Launcher.Plugin.Explorer.Views;

[INotifyPropertyChanged]
public partial class QuickAccessLinkSettings
{
    private static readonly string ClassName = nameof(QuickAccessLinkSettings);

    private string _selectedPath;
    public string SelectedPath
    {
        get => _selectedPath;
        set
        {
            if (_selectedPath != value)
            {
                _selectedPath = value;
                OnPropertyChanged();
                if (string.IsNullOrEmpty(_selectedName))
                {
                    SelectedName = _selectedPath.GetPathName();
                }
                if (!string.IsNullOrEmpty(_selectedPath))
                {
                    _accessLinkType = GetResultType(_selectedPath);
                }
            }
        }
    }

    private string _selectedName;
    public string SelectedName
    {
        get
        {
            return string.IsNullOrEmpty(_selectedName) ? _selectedPath.GetPathName() : _selectedName;
        }
        set
        {
            if (_selectedName != value)
            {
                _selectedName = value;
                OnPropertyChanged();
            }
        }
    }

    public bool IsFileSelected { get; set; }
    public bool IsFolderSelected { get; set; } = true; // Default to Folder

    private bool IsEdit { get; }
    private AccessLink SelectedAccessLink { get; }

    public ObservableCollection<AccessLink> QuickAccessLinks { get; }

    private ResultType _accessLinkType = ResultType.Folder; // Default to Folder

    public QuickAccessLinkSettings(ObservableCollection<AccessLink> quickAccessLinks)
    {
        IsEdit = false;
        QuickAccessLinks = quickAccessLinks;
        InitializeComponent();
    }

    public QuickAccessLinkSettings(ObservableCollection<AccessLink> quickAccessLinks, AccessLink selectedAccessLink)
    {
        IsEdit = true;
        _selectedName = selectedAccessLink.Name;
        _selectedPath = selectedAccessLink.Path;
        _accessLinkType = GetResultType(_selectedPath); // Initialize link type
        IsFileSelected = selectedAccessLink.Type == ResultType.File; // Initialize default selection
        IsFolderSelected = !IsFileSelected;
        SelectedAccessLink = selectedAccessLink;
        QuickAccessLinks = quickAccessLinks;
        InitializeComponent();
    }

    private void BtnCancel_OnClick(object sender, RoutedEventArgs e)
    {
        DialogResult = false;
        Close();
    }

    private void OnDoneButtonClick(object sender, RoutedEventArgs e)
    {
        // Validate the input before proceeding
        if (string.IsNullOrEmpty(SelectedName) || string.IsNullOrEmpty(SelectedPath))
        {
            var warning = Localize.plugin_explorer_quick_access_link_no_folder_selected();
            Main.Context.API.ShowMsgBox(warning);
            return;
        }

        // Check if the path already exists in the quick access links
        if (QuickAccessLinks.Any(x =>
                x.Path.Equals(SelectedPath, StringComparison.OrdinalIgnoreCase) &&
                x.Name.Equals(SelectedName, StringComparison.OrdinalIgnoreCase)))
        {
            var warning = Localize.plugin_explorer_quick_access_link_path_already_exists();
            Main.Context.API.ShowMsgBox(warning);
            return;
        }

        // If editing, update the existing link
        if (IsEdit)
        {
            if (SelectedAccessLink != null)
            {
                var index = QuickAccessLinks.IndexOf(SelectedAccessLink);
                if (index >= 0)
                {
                    var updatedLink = new AccessLink
                    {
                        Name = SelectedName,
                        Type = _accessLinkType,
                        Path = SelectedPath
                    };
                    QuickAccessLinks[index] = updatedLink;
                }
                DialogResult = true;
                Close();
            }
            // Add a new one if the selected access link is null (should not happen in edit mode, but just in case)
            else
            {
                AddNewAccessLink();
            }
        }
        // Otherwise, add a new one
        else
        {
            AddNewAccessLink();
        }

        void AddNewAccessLink()
        {
            var newAccessLink = new AccessLink
            {
                Name = SelectedName,
                Type = _accessLinkType,
                Path = SelectedPath
            };
            QuickAccessLinks.Add(newAccessLink);
            DialogResult = true;
            Close();
        }
    }

    private void SelectPath_OnClick(object commandParameter, RoutedEventArgs e)
    {
        // Open file or folder selection dialog based on the selected radio button
        if (IsFileSelected)
        {
            var openFileDialog = new OpenFileDialog
            {
                Multiselect = false,
                CheckFileExists = true,
                CheckPathExists = true
            };

            if (openFileDialog.ShowDialog() != System.Windows.Forms.DialogResult.OK ||
                string.IsNullOrEmpty(openFileDialog.FileName))
                return;

            SelectedPath = openFileDialog.FileName;
        }
        else // Folder selection
        {
            var folderBrowserDialog = new FolderBrowserDialog
            {
                ShowNewFolderButton = true
            };

            if (folderBrowserDialog.ShowDialog() != System.Windows.Forms.DialogResult.OK ||
                string.IsNullOrEmpty(folderBrowserDialog.SelectedPath))
                return;

            SelectedPath = folderBrowserDialog.SelectedPath;
        }
    }

    private static ResultType GetResultType(string path)
    {
        // Check if the path is a file or folder
        if (File.Exists(path))
        {
            return ResultType.File;
        }
        else if (Directory.Exists(path))
        {
            if (string.Equals(Path.GetPathRoot(path), path, StringComparison.OrdinalIgnoreCase))
            {
                return ResultType.Volume;
            }
            else
            {
                return ResultType.Folder;
            }
        }
        else
        {
            // This should not happen, but just in case, we assume it's a folder
            Main.Context.API.LogError(ClassName, $"The path '{path}' does not exist or is invalid. Defaulting to Folder type.");
            return ResultType.Folder;
        }
    }
}