using System;

using System.Collections.Generic;

using System.Collections.Specialized;

using System.Reflection;

using System.Text;

using System.Threading.Tasks;

using NDesk.Options;

using Quartz;

using Quartz.Impl;

using SiteServer.Cli.Core;

using SiteServer.Cli.Jobs;

using SiteServer.CMS.Plugin;

using SiteServer.Plugin;

using SiteServer.Utils;

namespace SiteServer.Cli

{

internal static class Program

{

private static bool IsHelp { get; set; }

private static string Repeat { get; set; }

private static Dictionary<string, Func<IJobContext, Task>> Jobs { get; set; }

public static string CommandName { get; private set; }

public static string[] CommandArgs { get; private set; }

private static readonly OptionSet Options = new OptionSet {

{ "r|repeat=", "schedule CRON expression",

v => Repeat = v },

{ "h|help", "命令说明",

v => IsHelp = v != null }

};

private static void Main(string[] args)

{

//Console.OutputEncoding = Encoding.UTF8;

Console.OutputEncoding = Encoding.GetEncoding(936);

if (!CliUtils.ParseArgs(Options, args)) return;

var commandNames = new List<string>();

var commandArgs = new List<string>();

if (args.Length >= 1)

{

var isCommand = true;

foreach (var arg in args)

{

if (isCommand && !StringUtils.StartsWith(arg, "-"))

{

commandNames.Add(StringUtils.Trim(arg));

}

else

{

isCommand = false;

commandArgs.Add(StringUtils.Trim(arg));

}

}

}

CommandName = string.Join(" ", commandNames);

CommandArgs = commandArgs.ToArray();

Console.WriteLine("欢迎使用 SiteServer Cli 命令行工具");

Console.WriteLine();

Jobs = new Dictionary<string, Func<IJobContext, Task>>(StringComparer.CurrentCultureIgnoreCase)

{

{BackupJob.CommandName, BackupJob.Execute},

{InstallJob.CommandName, InstallJob.Execute},

{RestoreJob.CommandName, RestoreJob.Execute},

{UpdateJob.CommandName, UpdateJob.Execute},

{VersionJob.CommandName, VersionJob.Execute},

{TestJob.CommandName, TestJob.Execute}

};

PluginManager.LoadPlugins(CliUtils.PhysicalApplicationPath);

var pluginJobs = PluginJobManager.GetJobs();

if (pluginJobs != null && pluginJobs.Count > 0)

{

foreach (var command in pluginJobs.Keys)

{

if (!Jobs.ContainsKey(command))

{

Jobs.Add(command, pluginJobs[command]);

}

}

}

if (!Jobs.ContainsKey(CommandName))

{

RunHelpAsync(IsHelp, CommandName).GetAwaiter().GetResult();

}

else if (!string.IsNullOrEmpty(Repeat))

{

RunRepeatAsync(Repeat).GetAwaiter().GetResult();

}

else

{

RunExecuteAsync(CommandName, CommandArgs, null).GetAwaiter().GetResult();

}

}

private static async Task RunHelpAsync(bool isHelp, string commandName)

{

if (isHelp || string.IsNullOrEmpty(commandName))

{

var version = Assembly.GetExecutingAssembly().GetName().Version.ToString();

await Console.Out.WriteLineAsync($"Cli 命令行版本: {version.Substring(0, version.Length - 2)}");

await Console.Out.WriteLineAsync($"当前文件夹: {CliUtils.PhysicalApplicationPath}");

await Console.Out.WriteLineAsync($"Cli 命令行文件夹: {Assembly.GetExecutingAssembly().Location}");

await Console.Out.WriteLineAsync();

await CliUtils.PrintRowLine();

await CliUtils.PrintRow("Usage");

await CliUtils.PrintRowLine();

BackupJob.PrintUsage();

InstallJob.PrintUsage();

RestoreJob.PrintUsage();

UpdateJob.PrintUsage();

VersionJob.PrintUsage();

await CliUtils.PrintRowLine();

await CliUtils.PrintRow("https://www.siteserver.cn/docs/cli");

await CliUtils.PrintRowLine();

Console.ReadLine();

}

else

{

Console.WriteLine($"'{commandName}' is not a siteserver command. See 'sitserver --help'");

}

}

private static async Task RunRepeatAsync(string schedule)

{

try

{

var factory = new StdSchedulerFactory(new NameValueCollection

{

{ "quartz.serializer.type", "binary" }

});

var sched = await factory.GetScheduler();

await sched.Start();

var job = JobBuilder.Create<SchedJob>()

.WithIdentity("job1", "group1")

.Build();

var trigger = TriggerBuilder.Create()

.WithIdentity("trigger1", "group1")

.StartNow()

.WithCronSchedule(schedule)

.WithPriority(1)

.Build();

await sched.ScheduleJob(job, trigger);

await Task.Delay(-1);

await sched.Shutdown();

}

catch (Exception ex)

{

await CliUtils.PrintErrorAsync(ex.Message);

}

}

public static async Task RunExecuteAsync(string commandName, string[] commandArgs, IJobExecutionContext jobContext)

{

try

{

Func<IJobContext, Task> job;

if (Jobs.TryGetValue(commandName, out job))

{

if (job != null)

{

var context = new JobContextImpl(commandName, commandArgs, jobContext);

await job(context);

}

}

}

catch (Exception ex)

{

await CliUtils.PrintErrorAsync(ex.Message);

var errorLogFilePath = CliUtils.CreateErrorLogFile("siteserver");

await CliUtils.AppendErrorLogsAsync(errorLogFilePath, new List<TextLogInfo>

{

new TextLogInfo

{

DateTime = DateTime.Now,

Detail = "Console Error",

Exception = ex

}

});

}

}

}

}