using CloudBuilder.Core.DatabaseAccessor.Entity;
using CloudBuilder.Core.DependencyInjection.Task;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Text.RegularExpressions;
using System.Threading.Tasks;

namespace CloudBuilder.Topshelf.Task.Novel
{
    //task:SplitFileChapterTask encoding:GB2312 pattern: @"第([^】]+)章" input:D:\NET8\AI\frxxz.txt output:D:\NET8\AI\NovelSplitter
    //input可以是文件名，也可以是目錄名
    //output輸出目錄
    //pattern可能自定義分割表達式
    public class SplitFileChapterTask : IScheduleTask
    {
        private readonly IApplicationService service;

        public SplitFileChapterTask(IApplicationService service)
        {
            this.service = service;
        }

        public void Run(Dictionary<string, string> bodyDict)
        {
            if (!bodyDict.ContainsKey("input") || string.IsNullOrEmpty(bodyDict["input"])) return;

            string input = bodyDict["input"];
            string output = AppDomain.CurrentDomain.BaseDirectory;

            //如果是文件名，輸出到同一目錄下
            if (File.Exists(input))
            {
                output = Path.GetDirectoryName(input) + "\\";
            }
            else if (Directory.Exists(input))
            {
                output = input;
            }

            if (bodyDict.ContainsKey("output") && !string.IsNullOrEmpty(bodyDict["output"]))
            {
                output = bodyDict["output"];
                //如果目錄不存在，創建目錄
                if (!Directory.Exists(output)) Directory.CreateDirectory(output);
            }


            string pattern = @"第([^】]+)章";
            if (bodyDict.ContainsKey("pattern") && !string.IsNullOrEmpty(bodyDict["pattern"]))
                pattern = bodyDict["pattern"];

            string encoding = "UTF-8";

            if (bodyDict.ContainsKey("encoding") && !string.IsNullOrEmpty(bodyDict["encoding"])) encoding = bodyDict["encoding"];

            List<string> files = PackFiles(input);

            foreach (string file in files)
            {
                SaveFile(encoding, file, output, pattern);
            }
        }

        private void SaveFile(string encoding, string fileName, string output, string pattern)
        {
            System.Text.Encoding.RegisterProvider(System.Text.CodePagesEncodingProvider.Instance);

            string[] lines = File.ReadAllLines(fileName, Encoding.GetEncoding(encoding));
           
            string title = string.Empty;

            int chapterIndex = 1;
            StringBuilder chapter = new StringBuilder();
            foreach (string line in lines)
            {
                Match match = Regex.Match(line, pattern);
                if (match.Success)
                {
                    // 提取第一个分组（即括号内的内容）
                    //string value = match.Groups[1].Value;

                    //第一次找到是空的，下次找到新的保存之前的
                    if (!string.IsNullOrEmpty(title) && chapter.Length > 0)
                    {
                        File.WriteAllText(Path.Combine(output, Path.GetFileNameWithoutExtension(fileName) + string.Format("_{0}_{1}", chapterIndex++,title) + Path.GetExtension(fileName)), chapter.ToString(), Encoding.UTF8);

                        chapter = new StringBuilder();
                    }

                    title = line;
                }

                if (!string.IsNullOrEmpty(title))
                {
                    chapter.AppendLine(line);
                }
            }
            //最后一章
            if (!string.IsNullOrEmpty(title) && chapter.Length > 0)
            {
                File.WriteAllText(Path.Combine(output, Path.GetFileNameWithoutExtension(fileName) + string.Format("_{0}_{1}", chapterIndex++, title) + Path.GetExtension(fileName)), chapter.ToString(), Encoding.UTF8);
            }
        }

        private List<string> PackFiles(string input)
        {
            List<string> files = new List<string>();
            if (Path.HasExtension(input) && !string.IsNullOrEmpty(input))
            {
                if (File.Exists(input))
                {
                    files.Add(input);
                }
            }
            else if (!string.IsNullOrEmpty(input))
            {
                if (Directory.Exists(input))
                {
                    string[] tempFiles = Directory.GetFiles(input);
                    if (tempFiles != null && tempFiles.Length > 0)
                    {
                        files.AddRange(tempFiles);
                    }
                }
            }

            return files;
        }

    }
}