97 lines
3.7 KiB
C#
97 lines
3.7 KiB
C#
using Microsoft.ML;
|
|
using Microsoft.ML.Data;
|
|
using Microsoft.ML.Transforms.Onnx;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Runtime.InteropServices;
|
|
using WinFormsApp3;
|
|
|
|
//https://huggingface.co/protectai/gyr66-bert-base-chinese-finetuned-ner-onnx/tree/main
|
|
//model.onnx
|
|
//vocab.txt
|
|
|
|
namespace WinFormsApp3
|
|
{
|
|
class Program
|
|
{
|
|
[STAThread] // ← 加上这个，99% 立刻解决你的错误
|
|
static int Main(string[] args)
|
|
{
|
|
// 命令行下载：WinFormsApp3.exe --download <视频地址> [--out 保存路径] [--referer 页面地址]
|
|
if (args.Length > 0 && string.Equals(args[0], "--download", StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
// 下载丢给线程池线程去跑，不要在 STA 主线程上等待异步，否则 COM 线程模式会冲突
|
|
return Task.Run(() => DownloadCliAsync(args)).GetAwaiter().GetResult();
|
|
}
|
|
|
|
ApplicationConfiguration.Initialize();
|
|
|
|
// 这里不能提前 form.Show()：那会让 Form.Load 在消息循环开始之前执行，
|
|
// WebView2 在没有消息泵时初始化会报 RPC_E_CHANGED_MODE，交给 Run 自己显示窗体
|
|
Application.Run(new Form1());
|
|
return 0;
|
|
}
|
|
|
|
[DllImport("kernel32.dll", SetLastError = true)]
|
|
private static extern bool AttachConsole(int dwProcessId);
|
|
|
|
private const int AttachParentProcess = -1;
|
|
|
|
/// <summary>
|
|
/// 命令行下载入口：不创建窗体、不初始化 WebView2，直接把地址下完就退出，方便脚本化验证。
|
|
/// </summary>
|
|
private static async Task<int> DownloadCliAsync(string[] args)
|
|
{
|
|
// WinExe 默认没有控制台，Attach 到父进程的输出才能在终端里看到结果
|
|
AttachConsole(AttachParentProcess);
|
|
|
|
string url = args.Length > 1 ? args[1] : string.Empty;
|
|
if (string.IsNullOrWhiteSpace(url) || url.StartsWith("--", StringComparison.Ordinal))
|
|
{
|
|
Console.Error.WriteLine("用法: WinFormsApp3.exe --download <视频地址> [--out 保存路径] [--referer 页面地址]");
|
|
return 2;
|
|
}
|
|
|
|
string? referer = GetOption(args, "--referer");
|
|
string? outPath = GetOption(args, "--out");
|
|
if (string.IsNullOrWhiteSpace(outPath))
|
|
{
|
|
outPath = Path.Combine(Path.GetTempPath(), new VideoDownloader(referer).SuggestedFileName(url, null));
|
|
}
|
|
|
|
using CancellationTokenSource cts = new(TimeSpan.FromMinutes(15));
|
|
Progress<DownloadProgress> progress = new(message => Console.WriteLine(message.Message));
|
|
|
|
try
|
|
{
|
|
VideoDownloader downloader = new(referer);
|
|
DownloadResult result = await downloader.DownloadAsync(url, outPath!, progress, cts.Token);
|
|
Console.WriteLine($"OK file={result.FilePath} bytes={result.Bytes} converted={result.Converted}");
|
|
return 0;
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
Console.Error.WriteLine($"FAIL {ex.GetType().Name}: {ex.Message}");
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// 取 --out / --referer 这类"选项 值"形式的参数。
|
|
/// </summary>
|
|
private static string? GetOption(string[] args, string name)
|
|
{
|
|
for (int i = 0; i < args.Length - 1; i++)
|
|
{
|
|
if (string.Equals(args[i], name, StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
return args[i + 1];
|
|
}
|
|
}
|
|
|
|
return null;
|
|
}
|
|
}
|
|
} |