tk/dot
1
0
Fork 0
mirror of https://github.com/nsnail/dot.git synced 2025-06-17 21:13:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

<feat> + git 输出编码参数

Browse Source
This commit is contained in:
nsnail 2022-12-06 21:26:17 +08:00
parent 6bd37be483
commit db83230391
9 changed files with 105 additions and 78 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
src/Color/Win32.cs
View File

@ -2,48 +2,47 @@ using System.Runtime.InteropServices;
namespace Dot.Color;
public static class Win32
public static partial class Win32
{
public delegate nint LowLevelMouseProc(int nCode, nint wParam, nint lParam);
private const string _GDI32_DLL = "gdi32.dll";
private const string _KERNEL32_DLL = "kernel32.dll";
private const string _USER32_DLL = "user32.dll";
public const int SW_HIDE = 0;
public const int SW_HIDE = 0;
[LibraryImport(_USER32_DLL)]
internal static partial nint CallNextHookEx(nint hhk, int nCode, nint wParam, nint lParam);
[LibraryImport(_KERNEL32_DLL)]
internal static partial nint GetConsoleWindow();
[DllImport(_USER32_DLL)]
public static extern nint CallNextHookEx(nint hhk, int nCode, nint wParam, nint lParam);
[DllImport(_KERNEL32_DLL)]
public static extern nint GetConsoleWindow();
[LibraryImport(_USER32_DLL)]
internal static partial nint GetDesktopWindow();
[DllImport(_USER32_DLL)]
public static extern nint GetDesktopWindow();
[LibraryImport(_KERNEL32_DLL, StringMarshalling = StringMarshalling.Utf16)]
internal static partial nint GetModuleHandle(string lpModuleName);
[LibraryImport(_GDI32_DLL)]
internal static partial uint GetPixel(nint dc, int x, int y);
[DllImport(_KERNEL32_DLL, CharSet = CharSet.Unicode)]
public static extern nint GetModuleHandle(string lpModuleName);
[LibraryImport(_USER32_DLL)]
internal static partial nint GetWindowDC(nint hWnd);
[DllImport(_GDI32_DLL)]
public static extern uint GetPixel(nint dc, int x, int y);
[LibraryImport(_USER32_DLL)]
internal static partial int ReleaseDC(nint hWnd, nint dc);
[DllImport(_USER32_DLL)]
public static extern nint GetWindowDC(nint hWnd);
[LibraryImport(_USER32_DLL)]
internal static partial nint SetWindowsHookEx(int idHook, LowLevelMouseProc lpfn, nint hMod, uint dwThreadId);
[DllImport(_USER32_DLL)]
public static extern int ReleaseDC(nint hWnd, nint dc);
[DllImport(_USER32_DLL)]
public static extern nint SetWindowsHookEx(int idHook, LowLevelMouseProc lpfn, nint hMod, uint dwThreadId);
[DllImport(_USER32_DLL)]
public static extern bool ShowWindow(nint hWnd, int nCmdShow);
[DllImport(_USER32_DLL)]
[LibraryImport(_USER32_DLL)]
[return: MarshalAs(UnmanagedType.Bool)]
public static extern bool UnhookWindowsHookEx(nint hhk);
internal static partial bool ShowWindow(nint hWnd, int nCmdShow);
[LibraryImport(_USER32_DLL)]
[return: MarshalAs(UnmanagedType.Bool)]
internal static partial bool UnhookWindowsHookEx(nint hhk);
}

99
src/Git/Main.cs
View File

@ -6,43 +6,48 @@ namespace Dot.Git;
public class Main : ToolBase<Option>
{
private const int _POS_Y_MSG = 74;
private const int _POST_Y_LOADING = 70;
private const int _REP_MAX_LENGTH = 32;
private (int x, int y) _cursorInitPos;
private List<string> _dirList;
private Encoding _encGbk;
private const int _POS_Y_MSG = 74; //git command rsp 显示的位置 y
private const int _POST_Y_LOADING = 70; //loading 动画显示的位置 y
private const int _REP_PATH_LENGTH_LIMIT = 32; //仓库路径长度显示截断阈值
private (int x, int y) _cursorPosBackup; //光标位置备份
private readonly Encoding _gitOutputEnc; //git command rsp 编码
private List<string> _repoPathList; //仓库目录列表
public Main(Option opt) : base(opt) { }
public Main(Option opt) : base(opt)
{
_gitOutputEnc = Encoding.GetEncoding(Opt.GitOutputEncoding);
if (!Directory.Exists(Opt.Path))
throw new ArgumentException(nameof(Opt.Path), string.Format(Str.PathNotFound, Opt.Path));
}
private async ValueTask DirHandle(string dir, CancellationToken cancelToken)
{
var index = _dirList.FindIndex(x => x == dir);
var tAnimate = LoadingAnimate(_POST_Y_LOADING, _cursorInitPos.y + index, out var cts);
var index = _repoPathList.FindIndex(x => x == dir);
var tAnimate = LoadingAnimate(_POST_Y_LOADING, _cursorPosBackup.y + index, out var cts);
void Write(object sender, DataReceivedEventArgs e)
void ExecRspReceived(object sender, DataReceivedEventArgs e)
{
if (e.Data is null) return;
var msg = Encoding.UTF8.GetString(_encGbk.GetBytes(e.Data));
ConcurrentWrite(_POS_Y_MSG, _cursorInitPos.y + index, new string(' ', Console.WindowWidth - _POS_Y_MSG));
ConcurrentWrite(_POS_Y_MSG, _cursorInitPos.y + index, msg);
var msg = Encoding.UTF8.GetString(_gitOutputEnc.GetBytes(e.Data));
ConcurrentWrite(_POS_Y_MSG, _cursorPosBackup.y + index, new string(' ', Console.WindowWidth - _POS_Y_MSG));
ConcurrentWrite(_POS_Y_MSG, _cursorPosBackup.y + index, msg);
}
var gitStartInfo = new ProcessStartInfo {
CreateNoWindow = true
, WorkingDirectory = dir
, FileName = "git"
, Arguments = Opt.Args
, UseShellExecute = false
, RedirectStandardOutput = true
, RedirectStandardError = true
};
using var p = Process.Start(gitStartInfo);
p.OutputDataReceived += Write;
p.ErrorDataReceived += Write;
var startInfo = new ProcessStartInfo {
CreateNoWindow = true
, WorkingDirectory = dir
, FileName = "git"
, Arguments = Opt.Args
, UseShellExecute = false
, RedirectStandardOutput = true
, RedirectStandardError = true
};
using var p = Process.Start(startInfo);
p!.OutputDataReceived += ExecRspReceived;
p.ErrorDataReceived += ExecRspReceived;
p.BeginOutputReadLine();
p.BeginErrorReadLine();
await p.WaitForExitAsync();
@ -53,37 +58,45 @@ public class Main : ToolBase<Option>
cts.Dispose();
}
private void StashCurorPos()
{
_cursorPosBackup = Console.GetCursorPosition();
}
public override async Task Run()
{
if (!Directory.Exists(Opt.Path))
throw new ArgumentException(nameof(Opt.Path), string.Format(Str.PathNotFound, Opt.Path));
Encoding.RegisterProvider(CodePagesEncodingProvider.Instance);
_encGbk = Encoding.GetEncoding("gbk");
Console.Write(Str.FindGitReps, Opt.Path);
_cursorInitPos = Console.GetCursorPosition();
var tAnimate = LoadingAnimate(_cursorInitPos.x, _cursorInitPos.y, out var cts);
StashCurorPos();
var tAnimate = LoadingAnimate(_cursorPosBackup.x, _cursorPosBackup.y, out var cts);
_repoPathList = Directory.GetDirectories(Opt.Path, ".git", SearchOption.AllDirectories)
.Select(x => Directory.GetParent(x)!.FullName)
.ToList();
_dirList = Directory.GetDirectories(Opt.Path, ".git", SearchOption.AllDirectories)
.Select(x => Directory.GetParent(x)!.FullName)
.ToList();
cts.Cancel();
await tAnimate;
cts.Dispose();
Console.WriteLine(Str.Ok);
_cursorInitPos = Console.GetCursorPosition();
var i = 0;
Console.WriteLine(string.Join(Environment.NewLine
, _dirList.Select(
x => $"{++i}: {new DirectoryInfo(x).Name.Sub(0, _REP_MAX_LENGTH)}")));
StashCurorPos();
await Parallel.ForEachAsync(_dirList, DirHandle);
{
var i = 0;
Console.WriteLine( //
string.Join(Environment.NewLine
, _repoPathList.Select(
x => $"{++i}: {new DirectoryInfo(x).Name.Sub(0, _REP_PATH_LENGTH_LIMIT)}"))
//
);
}
Console.SetCursorPosition(_cursorInitPos.x, _cursorInitPos.y + _dirList.Count);
await Parallel.ForEachAsync(_repoPathList, DirHandle);
Console.SetCursorPosition(_cursorPosBackup.x, _cursorPosBackup.y + _repoPathList.Count);
}
}

4
src/Git/Option.cs
View File

@ -6,6 +6,10 @@ public class Option : OptionBase
[Option('a', "args", HelpText = nameof(Str.GitArgs), Default = "status", ResourceType = typeof(Str))]
public string Args { get; set; }
[Option('e', "git-output-encoding", HelpText = nameof(Str.GitOutputEncoding), Default = "utf-8"
, ResourceType = typeof(Str))]
public string GitOutputEncoding { get; set; }
[Value(0, HelpText = nameof(Str.FolderPath), Default = ".", ResourceType = typeof(Str))]
public string Path { get; set; }
}

5
src/Lang/Str.en-US.resx
View File

@ -1,4 +1,4 @@
<?xml version="1.0" encoding="utf-8"?>
<?xml version="1.0" encoding="utf-8"?>
<root>
<xsd:schema xmlns:xsd="http://www.w3.org/2001/XMLSchema" xmlns:msdata="urn:schemas-microsoft-com:xml-msdata"
id="root" xmlns="">
@ -149,4 +149,7 @@
<data name="FindGitReps" xml:space="preserve">
<value>Find all git repository directories under "{0}"...</value>
</data>
<data name="GitOutputEncoding" xml:space="preserve">
<value>Git output encoding</value>
</data>
</root>

3
src/Lang/Str.resx
View File

@ -57,6 +57,9 @@
<data name="Copied" xml:space="preserve">
<value>{0}（已复制到剪贴板）</value>
</data>
<data name="GitOutputEncoding" xml:space="preserve">
<value>Git输出编码</value>
</data>
<data name="FileSearchPattern" xml:space="preserve">
<value>文件通配符</value>
</data>

5
src/Program.cs
View File

@ -1,4 +1,5 @@
using System.Reflection;
using System.Text;
using Dot;
Type[] LoadVerbs()
@ -24,8 +25,8 @@ async Task Run(object args)
}
//Entry Point
// Entry Point
Encoding.RegisterProvider(CodePagesEncodingProvider.Instance);
var types = LoadVerbs();
try {

2
src/Text/Main.cs
View File

@ -66,8 +66,6 @@ public sealed class Main : ToolBase<Option>
private static void ParseAndShow(string text)
{
Encoding.RegisterProvider(CodePagesEncodingProvider.Instance);
var ansi = BuildOutput(text, Encoding.GetEncoding("gbk"));
var utf8 = BuildOutput(text, Encoding.UTF8);
var unicodeLittleEndian = BuildOutput(text, Encoding.Unicode);

10
src/ToolBase.cs
View File

@ -2,7 +2,8 @@ namespace Dot;
public abstract class ToolBase<TOption> : ITool where TOption : OptionBase
{
private static readonly object _lockObj = new();
// ReSharper disable once StaticMemberInGenericType
private static SpinLock _spinlock;
protected readonly ProgressBarOptions //
DefaultProgressBarOptions = new() {
@ -25,10 +26,15 @@ public abstract class ToolBase<TOption> : ITool where TOption : OptionBase
protected static void ConcurrentWrite(int x, int y, string text)
{
lock (_lockObj) {
var lockTaken = false;
try {
_spinlock.Enter(ref lockTaken);
Console.SetCursorPosition(x, y);
Console.Write(text);
}
finally {
if (lockTaken) _spinlock.Exit(false);
}
}
protected static IEnumerable<string> EnumerateFiles(string path, string searchPattern)

4
src/dot.csproj
View File

@ -50,8 +50,8 @@
<ItemGroup Condition="!Exists('Lang\Str.Designer.cs')">
<Compile Include="Lang\Str.Designer.cs"/>
</ItemGroup>
<Exec Command="dotnet tool restore"/>
<Exec WorkingDirectory="$(ProjectDir)\Lang" Command="dotnet t4 Str.tt"/>
<Exec Command="dotnet tool restore" StdOutEncoding="utf-8"/>
<Exec WorkingDirectory="$(ProjectDir)\Lang" Command="dotnet t4 Str.tt" StdOutEncoding="utf-8"/>
</Target>
</Project>