Compare commits

..

4 Commits

Author SHA1 Message Date
b1d4f3693b move interfaces to separate folder 2025-12-10 22:32:48 +05:00
30256a86c2 iteration 2 2025-12-10 22:25:30 +05:00
5479da163e refactor mvvm, add datacontext 2025-12-10 21:33:06 +05:00
b6ddadc6d3 iteration 1 2025-12-10 18:59:27 +05:00
9 changed files with 301 additions and 453 deletions

View File

@@ -1,8 +1,8 @@
using CodeContextGenerator.Interfaces;
using System.Windows;
using CodeContextGenerator.Interfaces;
using CodeContextGenerator.Services;
using CodeContextGenerator.ViewModels;
using Microsoft.Extensions.DependencyInjection;
using System.Windows;
namespace CodeContextGenerator;

View File

@@ -1,26 +1,24 @@
using System.Globalization;
using System;
using System.Globalization;
using System.Windows;
using System.Windows.Data;
namespace CodeContextGenerator.Converters;
namespace CodeContextGenerator.Converters
{
public class BooleanToVisibilityConverter : IValueConverter
{
public object Convert(object value, Type targetType, object parameter, CultureInfo culture)
{
bool boolValue = value is true;
// Проверяем параметр для инвертирования
if (parameter is string paramStr && paramStr.Equals("Inverted", StringComparison.OrdinalIgnoreCase))
if (value is bool boolValue)
{
boolValue = !boolValue;
}
return boolValue ? Visibility.Visible : Visibility.Collapsed;
}
return Visibility.Collapsed;
}
public object ConvertBack(object value, Type targetType, object parameter, CultureInfo culture)
{
throw new NotImplementedException();
}
}
}

View File

@@ -15,7 +15,6 @@ public partial class FileItem : ObservableObject
private bool isDirectory;
[ObservableProperty]
[NotifyPropertyChangedFor(nameof(HasSelectedChildren))]
private bool? isSelected;
[ObservableProperty]
@@ -23,11 +22,6 @@ public partial class FileItem : ObservableObject
public ObservableCollection<FileItem> Children { get; } = new ObservableCollection<FileItem>();
// Событие, вызываемое при изменении состояния выбора
public event EventHandler SelectionChanged;
public bool HasSelectedChildren => Children.Any(c => c.IsSelected == true || c.HasSelectedChildren);
partial void OnIsSelectedChanged(bool? oldValue, bool? newValue)
{
if (newValue.HasValue)
@@ -35,7 +29,6 @@ public partial class FileItem : ObservableObject
UpdateChildrenSelection(newValue.Value);
}
UpdateParentSelection();
NotifySelectionChanged();
}
private void UpdateChildrenSelection(bool value)
@@ -44,13 +37,7 @@ public partial class FileItem : ObservableObject
foreach (var child in Children)
{
// Гарантируем вызов события для каждого ребенка
var oldValue = child.IsSelected;
child.IsSelected = value;
if (oldValue != child.IsSelected)
{
child.RaiseSelectionChanged();
}
}
}
@@ -63,41 +50,7 @@ public partial class FileItem : ObservableObject
var noneSelected = children.All(c => c.IsSelected == false);
var hasIndeterminate = children.Any(c => c.IsSelected == null);
bool? newParentState = hasIndeterminate ? null : (allSelected ? true : (noneSelected ? false : null));
/*bool? newParentState;
if (hasIndeterminate)
{
newParentState = null;
}
else if (allSelected)
{
newParentState = true;
}
else if (noneSelected)
{
newParentState = false;
}
else
{
newParentState = null;
}*/
if (Parent.IsSelected != newParentState)
{
Parent.IsSelected = newParentState;
Parent.RaiseSelectionChanged();
}
}
private void NotifySelectionChanged()
{
RaiseSelectionChanged();
}
// Публичный метод для гарантии вызова события
public void RaiseSelectionChanged()
{
// Вызываем событие для текущего элемента
SelectionChanged?.Invoke(this, EventArgs.Empty);
Parent.IsSelected = hasIndeterminate ? null : (allSelected ? true : (noneSelected ? false : null));
Parent.UpdateParentSelection();
}
}

View File

@@ -9,7 +9,7 @@ public class FileScannerService : IFileScannerService
private static readonly string[] ExcludedDirectories = {
"bin", "obj", ".git", "packages", ".vs", "Properties",
"node_modules", ".vscode", ".idea", "Debug", "Release",
"wwwroot", "dist", "build", ".gitignore", ".dockerignore"
"wwwroot", "dist", "build", "node_modules"
};
private static readonly string[] IncludedExtensions = { ".cs", ".xaml" };
@@ -25,13 +25,13 @@ public class FileScannerService : IFileScannerService
var directories = Directory.GetDirectories(path)
.Where(d => !ExcludedDirectories.Any(ex =>
d.EndsWith(ex, StringComparison.OrdinalIgnoreCase) ||
Path.GetFileName(d).Equals(ex, StringComparison.OrdinalIgnoreCase)))
d.EndsWith(ex, System.StringComparison.OrdinalIgnoreCase) ||
Path.GetFileName(d).Equals(ex, System.StringComparison.OrdinalIgnoreCase)))
.ToList();
var files = Directory.GetFiles(path)
.Where(f => IncludedExtensions.Any(ext =>
f.EndsWith(ext, StringComparison.OrdinalIgnoreCase)))
f.EndsWith(ext, System.StringComparison.OrdinalIgnoreCase)))
.ToList();
int totalItems = directories.Count + files.Count;
@@ -55,7 +55,7 @@ public class FileScannerService : IFileScannerService
await BuildDirectoryTreeAsync(dir, dirItem, progress, cancellationToken);
// Добавляем директорию только если в ней есть файлы или поддиректории с файлами
if (dirItem.Children.Count > 0 || files.Count > 0)
if (dirItem.Children.Any())
{
parentItem.Children.Add(dirItem);
}

View File

@@ -1,7 +1,6 @@
using CodeContextGenerator.Interfaces;
using CodeContextGenerator.Models;
using System.IO;
using System.Text.RegularExpressions;
using System.Windows;
namespace CodeContextGenerator.Services;
@@ -64,7 +63,7 @@ public class ProjectLoaderService : IProjectLoaderService
try
{
var projectPaths = ParseSolutionProjects(solutionPath, solutionDir);
var projectPaths = ParseSolutionProjects(solutionPath);
int totalProjects = projectPaths.Count;
int processedProjects = 0;
@@ -95,10 +94,7 @@ public class ProjectLoaderService : IProjectLoaderService
}
processedProjects++;
if (totalProjects > 0 && progress != null)
{
progress.Report((int)((processedProjects * 100.0) / totalProjects));
}
progress?.Report((int)((processedProjects * 100.0) / totalProjects));
}
}
catch (Exception ex)
@@ -109,40 +105,50 @@ public class ProjectLoaderService : IProjectLoaderService
return solutionItem;
}
// Улучшенный парсинг .sln файлов
private List<string> ParseSolutionProjects(string solutionPath, string solutionDir)
private async Task<FileItem> LoadCsprojAsync(string csprojPath, IProgress<int> progress, CancellationToken cancellationToken)
{
string projectDir = Path.GetDirectoryName(csprojPath);
string projectName = Path.GetFileNameWithoutExtension(csprojPath);
var projectItem = new FileItem
{
Name = projectName,
FullName = projectDir,
IsDirectory = true,
IsSelected = false
};
await _fileScannerService.BuildDirectoryTreeAsync(projectDir, projectItem, progress, cancellationToken);
return projectItem;
}
private List<string> ParseSolutionProjects(string solutionPath)
{
var projects = new List<string>();
var projectRegex = new Regex(@"Project\(""[^""]*""\)\s*=\s*""([^""]*)"",\s*""([^""]*)""", RegexOptions.IgnoreCase);
string solutionDir = Path.GetDirectoryName(solutionPath);
try
{
foreach (string line in File.ReadAllLines(solutionPath))
{
var match = projectRegex.Match(line);
if (match.Success && match.Groups.Count >= 3)
if (line.Trim().StartsWith("Project(", StringComparison.OrdinalIgnoreCase))
{
string projectName = match.Groups[1].Value;
string relativePath = match.Groups[2].Value;
// Пропускаем служебные проекты
if (projectName.Contains("Solution Items") ||
relativePath.EndsWith(".sln", StringComparison.OrdinalIgnoreCase) ||
relativePath.EndsWith(".suo", StringComparison.OrdinalIgnoreCase))
var parts = line.Split(new[] { '"' }, StringSplitOptions.RemoveEmptyEntries);
if (parts.Length >= 3)
{
string relativePath = parts[2].Trim();
if (relativePath.EndsWith(".csproj", StringComparison.OrdinalIgnoreCase))
{
continue;
}
string absolutePath = Path.GetFullPath(Path.Combine(solutionDir, relativePath));
// Проверяем, что это .csproj файл
if (absolutePath.EndsWith(".csproj", StringComparison.OrdinalIgnoreCase) && File.Exists(absolutePath))
if (File.Exists(absolutePath))
{
projects.Add(absolutePath);
}
}
}
}
}
}
catch (Exception ex)
{
throw new Exception($"Ошибка парсинга файла решения: {ex.Message}", ex);
@@ -151,23 +157,6 @@ public class ProjectLoaderService : IProjectLoaderService
return projects;
}
private async Task<FileItem> LoadCsprojAsync(string csprojPath, IProgress<int> progress, CancellationToken cancellationToken)
{
string projectDir = Path.GetDirectoryName(csprojPath);
string projectName = Path.GetFileNameWithoutExtension(csprojPath);
var projectItem = new FileItem
{
Name = $"{projectName} (Проект)",
FullName = projectDir,
IsDirectory = true,
IsSelected = false
};
await _fileScannerService.BuildDirectoryTreeAsync(projectDir, projectItem, progress, cancellationToken);
return projectItem;
}
public string GetDefaultOutputFileName(string projectPath)
{
if (Directory.Exists(projectPath))

View File

@@ -7,32 +7,20 @@ namespace CodeContextGenerator.Services;
public class UIService : IUIService
{
private readonly ISettingsService _settingsService;
public UIService(ISettingsService settingsService)
{
_settingsService = settingsService;
}
public string ShowFolderBrowserDialog(string initialDirectory = null)
{
// Используем специальный трюк для выбора папки в WPF
// Используем OpenFileDialog для выбора папки - это стандартный способ в WPF
var dialog = new OpenFileDialog
{
Title = "Выберите папку с проектом",
Filter = "Папки|*.dummy", // Фильтр для отображения только папок
FileName = "выберите_папку", // Специальное имя файла
Filter = "Папки|*.folder", // Фильтр для отображения только папок
FileName = "select_folder", // Имя файла для обхода проверки существования файла
CheckFileExists = false,
CheckPathExists = true,
ValidateNames = false,
ValidateNames = false, // Отключаем валидацию имен для выбора папок
DereferenceLinks = true
};
if (string.IsNullOrEmpty(initialDirectory) || !Directory.Exists(initialDirectory))
{
initialDirectory = _settingsService.GetLastProjectPath();
}
if (!string.IsNullOrEmpty(initialDirectory) && Directory.Exists(initialDirectory))
{
dialog.InitialDirectory = initialDirectory;
@@ -41,9 +29,8 @@ public class UIService : IUIService
var result = dialog.ShowDialog();
if (result == true)
{
// Возвращаем директорию, а не путь к файлу
string selectedFolder = Path.GetDirectoryName(dialog.FileName);
return selectedFolder;
// Возвращаем папку, а не файл
return Path.GetDirectoryName(dialog.FileName);
}
return null;
@@ -55,17 +42,11 @@ public class UIService : IUIService
var dialog = new OpenFileDialog
{
Title = "Выберите файл решения или проекта",
Filter = "Все поддерживаемые файлы (*.sln;*.csproj)|*.sln;*.csproj|Файлы решений Visual Studio (*.sln)|*.sln|Файлы проектов C# (*.csproj)|*.csproj|Все файлы (*.*)|*.*",
Title = "Выберите файл решения, проекта или папку",
Filter = "Все поддерживаемые файлы (*.sln;*.csproj;*.razor)|*.sln;*.csproj;*.razor|Файлы решений (*.sln)|*.sln|Файлы проектов (*.csproj)|*.csproj|Файлы Razor (*.razor)|*.razor|Все файлы (*.*)|*.*",
Multiselect = false
};
var lastPath = _settingsService.GetLastProjectPath();
if (!string.IsNullOrEmpty(lastPath) && Directory.Exists(lastPath))
{
dialog.InitialDirectory = lastPath;
}
bool? result = dialog.ShowDialog();
if (result == true)
{

View File

@@ -1,17 +1,12 @@
using CodeContextGenerator.Interfaces;
using CodeContextGenerator.Models;
using CodeContextGenerator.Services;
using CommunityToolkit.Mvvm.ComponentModel;
using CommunityToolkit.Mvvm.Input;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
using System.Windows;
namespace CodeContextGenerator.ViewModels
{
namespace CodeContextGenerator.ViewModels;
public partial class MainViewModel : ObservableObject
{
private readonly IProjectLoaderService _projectLoaderService;
@@ -40,12 +35,6 @@ namespace CodeContextGenerator.ViewModels
[ObservableProperty]
private bool isProjectLoaded;
// Явное объявление команд для гарантии их создания
public IRelayCommand SelectProjectCommand { get; }
public IAsyncRelayCommand GenerateContextFileCommand { get; }
public IRelayCommand CancelProcessingCommand { get; }
public IRelayCommand ExitApplicationCommand { get; }
public MainViewModel(
IProjectLoaderService projectLoaderService,
IFileScannerService fileScannerService,
@@ -58,41 +47,24 @@ namespace CodeContextGenerator.ViewModels
_contextFileGenerator = contextFileGenerator;
_uiService = uiService;
_settingsService = settingsService;
// Явная инициализация команд
SelectProjectCommand = new RelayCommand(SelectProject, CanSelectProject);
GenerateContextFileCommand = new AsyncRelayCommand(GenerateContextFileAsync, CanGenerate);
CancelProcessingCommand = new RelayCommand(CancelProcessing);
ExitApplicationCommand = new RelayCommand(ExitApplication);
}
private bool CanSelectProject() => !IsProcessing;
private bool CanGenerate()
{
bool result = !IsProcessing &&
IsProjectLoaded &&
RootDirectory != null &&
HasSelectedFiles(RootDirectory);
// Отладочная информация
System.Diagnostics.Debug.WriteLine($"CanGenerate: {result}, IsProcessing: {IsProcessing}, IsProjectLoaded: {IsProjectLoaded}, RootDirectory: {(RootDirectory != null)}, HasSelectedFiles: {HasSelectedFiles(RootDirectory)}");
return result;
}
public bool CanSelectProject => !IsProcessing;
public bool CanGenerate => !IsProcessing && IsProjectLoaded && HasSelectedFiles(RootDirectory);
[RelayCommand(CanExecute = nameof(CanSelectProject))]
private void SelectProject()
{
var initialDir = !string.IsNullOrEmpty(SelectedProjectPath) && Directory.Exists(SelectedProjectPath)
? Path.GetDirectoryName(SelectedProjectPath)
: _settingsService.GetLastProjectPath();
var initialDir = _settingsService.GetLastProjectPath();
// Сначала пробуем выбрать файл проекта/решения
if (_uiService.ShowOpenProjectFileDialog(out var filePath))
{
LoadProject(filePath);
return;
}
// Если отменили выбор файла - предлагаем выбрать папку
var folderPath = _uiService.ShowFolderBrowserDialog(initialDir);
if (!string.IsNullOrEmpty(folderPath))
{
@@ -100,6 +72,7 @@ namespace CodeContextGenerator.ViewModels
}
}
[RelayCommand(CanExecute = nameof(CanGenerate))]
private async Task GenerateContextFileAsync()
{
var selectedFiles = _fileScannerService.GetSelectedFiles(RootDirectory);
@@ -110,7 +83,7 @@ namespace CodeContextGenerator.ViewModels
}
var defaultFileName = _projectLoaderService.GetDefaultOutputFileName(SelectedProjectPath);
var initialDir = _settingsService.GetLastProjectPath() ?? Path.GetDirectoryName(SelectedProjectPath);
var initialDir = _settingsService.GetLastProjectPath();
if (!_uiService.ShowSaveFileDialog(defaultFileName, initialDir, out var savePath))
return;
@@ -152,17 +125,16 @@ namespace CodeContextGenerator.ViewModels
finally
{
IsProcessing = false;
// Принудительно обновляем команды после завершения операции
UpdateCommandsCanExecute();
}
}
[RelayCommand]
private void CancelProcessing()
{
_cancellationTokenSource?.Cancel();
ProgressText = "Операция отменена";
}
[RelayCommand]
private void ExitApplication()
{
Application.Current.Shutdown();
@@ -174,14 +146,12 @@ namespace CodeContextGenerator.ViewModels
_settingsService.SaveLastProjectPath(Path.GetDirectoryName(projectPath));
IsProjectLoaded = false;
RootDirectory = null;
UpdateCommandsCanExecute();
LoadProjectAsync(projectPath);
}
private async void LoadProjectAsync(string projectPath)
{
IsProcessing = true;
ProgressText = "Загрузка проекта...";
ProgressValue = 0;
@@ -199,18 +169,9 @@ namespace CodeContextGenerator.ViewModels
IsProjectLoaded = true;
ProgressText = "Проект загружен успешно";
// Подписываемся на события изменения выбора
SubscribeToSelectionChanges(RootDirectory);
// Сбрасываем выделение после загрузки
if (RootDirectory != null)
{
ClearSelections(RootDirectory);
}
// Принудительно обновляем команды
UpdateCommandsCanExecute();
}
catch (OperationCanceledException)
{
ProgressText = "Загрузка отменена";
@@ -222,30 +183,11 @@ namespace CodeContextGenerator.ViewModels
finally
{
IsProcessing = false;
UpdateCommandsCanExecute();
}
}
// Рекурсивная подписка на события изменения выбора
private void SubscribeToSelectionChanges(FileItem item)
{
if (item == null) return;
item.SelectionChanged += (sender, args) =>
{
UpdateCommandsCanExecute();
};
foreach (var child in item.Children)
{
SubscribeToSelectionChanges(child);
}
}
private void ClearSelections(FileItem item)
{
if (item == null) return;
item.IsSelected = false;
foreach (var child in item.Children)
{
@@ -256,27 +198,11 @@ namespace CodeContextGenerator.ViewModels
private bool HasSelectedFiles(FileItem item)
{
if (item == null) return false;
if (!item.IsDirectory && item.IsSelected == true)
return true;
if (item.IsDirectory)
{
if (item.IsSelected == true && !item.IsDirectory) return true;
foreach (var child in item.Children)
{
if (HasSelectedFiles(child))
return true;
if (HasSelectedFiles(child)) return true;
}
}
return false;
}
// Метод для принудительного обновления всех команд
private void UpdateCommandsCanExecute()
{
(SelectProjectCommand as RelayCommand)?.NotifyCanExecuteChanged();
(GenerateContextFileCommand as AsyncRelayCommand)?.NotifyCanExecuteChanged();
}
}
}

View File

@@ -59,9 +59,8 @@
TextWrapping="Wrap" />
</StackPanel>
<!-- Контейнер для дерева и сообщения -->
<Grid Grid.Row="2">
<Border
Grid.Row="2"
Padding="5"
BorderBrush="Gray"
BorderThickness="1"
@@ -97,13 +96,13 @@
</Border>
<TextBlock
Grid.Row="2"
HorizontalAlignment="Center"
VerticalAlignment="Center"
FontSize="14"
Foreground="Gray"
Text="Проект еще не загружен. Выберите файл решения или проекта."
Visibility="{Binding IsProjectLoaded, Converter={StaticResource BooleanToVisibilityConverter}, ConverterParameter=Inverted}" />
</Grid>
Visibility="{Binding IsProjectLoaded, Converter={StaticResource BooleanToVisibilityConverter}, ConverterParameter=Collapsed}" />
<ProgressBar
Grid.Row="3"
@@ -131,12 +130,14 @@
Content="Отмена"
Foreground="White"
Visibility="{Binding IsProcessing, Converter={StaticResource BooleanToVisibilityConverter}}" />
<Button Command="{Binding ExitApplicationCommand}" Content="В главное меню" />
<Button Command="{Binding ExitApplicationCommand}" Content="Закрыть" />
<Button
Background="#FF28A745"
Command="{Binding GenerateContextFileCommand}"
Content="Сформировать"
Foreground="White" />
Foreground="White"
IsEnabled="{Binding CanGenerate}"
Visibility="{Binding IsProjectLoaded, Converter={StaticResource BooleanToVisibilityConverter}}" />
</StackPanel>
</Grid>
</Window>

View File

@@ -1,3 +1,3 @@
# CodeContextGenerator
Генератор контекста для нейросети из проекта Visual Studio / VS Code
Генератор контекста из проекта для нейросети