using CloudBuilder.AI.Data; using CloudBuilder.AI.Entity; using CloudBuilder.AI.Policy; using CloudBuilder.AI.SearchCriteria; using CloudBuilder.AI.Service.Utility; using CloudBuilder.Core.DatabaseAccessor; using CloudBuilder.Core.DatabaseAccessor.Entity; using CloudBuilder.Core.Service; using CloudBuilder.Security.Data; using CloudBuilder.Security.Entity; using CloudBuilder.Security.Policy; using CloudBuilder.Security.Service; using CloudBuilder.Security.Utility; using DocumentFormat.OpenXml.ExtendedProperties; using Microsoft.AspNetCore.Authorization; using Microsoft.Extensions.DependencyInjection; using System.Diagnostics.Contracts; using System.Text; using System.Transactions; namespace CloudBuilder.AI.Service { public class AiBookService : IAiBookService { private readonly IApplicationService service; private IRepository repository; public AiBookService(IApplicationService service) { this.service = service; repository = service.GetRepository>(); } [PermissionOperate(name: AIPermissionPolicy.AiBook, operate: Operate.Find)] public AiBookData FindByKey(string key) { AiBookData data = new AiBookData(); data.AiBook = repository.DetachedEntities.Where(x => x.Guid == key).FirstOrDefault()!; if (data.AiBook.FileId > 0) { IRepository repositoryUploadDmsFile = service.GetRepository>(); BookDmsFileViewEntity dms; dms = repositoryUploadDmsFile.DetachedEntities.Where(x => x.FileId == data.AiBook.FileId).FirstOrDefault()!; data.FileName = dms.OrgFileName; } return data; } [PermissionOperate(name: AIPermissionPolicy.AiBook, operate: Operate.Find)] public AiBookEntity FindByTitle(string title) { return repository.DetachedEntities.Where(x => x.Title == title).FirstOrDefault()!; } [PermissionOperate(name: AIPermissionPolicy.AiBook, operate: Operate.Find)] public AiBookResultData FindBySearchCriteria(AiBookSearchCriteria criteria) { AiBookResultData data = new AiBookResultData(); AiBookEntity[] list = null; int count = 0; string sqlWhere = null; if (criteria.Filter != null && criteria.Filter.Length > 0) { sqlWhere = SearchFilter.ComposeWhereClause(criteria.Filter); } AiBookEntity entity = AiBookEntity.GetInstance(); string sql = entity.GetSelectTopFrom(criteria.limit, criteria.page, criteria.orderby.StringToDbField(), criteria.orderbytag, sqlWhere); string err = null; try { list = repository.FromSql(sql).ToArray(); } catch (System.Exception ex) { err = ex.Message; } if (list != null && list.Length > 0) { sql = entity.GetSelectCountFrom(sqlWhere); count = (int)repository.ExecuteScalar(sql); } data.Limit = criteria.limit; data.Page = criteria.page; data.Datas = list; data.Total = count; //data.Sql = sql; data.ErrorMessage = err; return data; } //[PermissionOperate(name: AIPermissionPolicy.AiBook, operate: Operate.Print)] public dynamic PrintBySearchCriteria(AiBookSearchCriteria criteria) { throw new NotImplementedException(); } //[PermissionOperate(name: AIPermissionPolicy.AiBook, operate: Operate.Insert)] //[UnitOfWork] //public AiBookEntity Insert(AiBookEntity data) //{ // throw new NotImplementedException(); //} [PermissionOperate(name: AIPermissionPolicy.AiBook, operate: Operate.Insert)] [UnitOfWork] public AiBookData Insert(AiBookData data) { AiBookEntity entity = repository.DetachedEntities.Where(x => x.Title == data.AiBook.Title).FirstOrDefault(); if (entity != null) { throw new ValidatedException("记录已存在."); } ISecurityGeneratorNumberService securityGeneratorNumberService = service.ServiceProvider.GetService()!; data.AiBook.Guid = securityGeneratorNumberService.GetNo(AIPrefixPolicy.AI_BOOK, AIPrefixPolicy.GUID, AIPrefixPolicy.AI_BOOK_PREFIX, YesNoPolicy.NO, AIPrefixPolicy.AI_BOOK_GUID_FORMAT); repository.InsertNow(data.AiBook); IRepository repositoryAiChapterEntity = repository.GetRepository>(); if (data.AiChapters != null && data.AiChapters.Length > 0) repositoryAiChapterEntity.InsertNow(data.AiChapters); return data; } [PermissionOperate(name: AIPermissionPolicy.AiBook, operate: Operate.Update)] [UnitOfWork] public AiBookData Update(AiBookData data) { AiBookEntity aiBook = repository.DetachedEntities.Where(x => x.Guid == data.AiBook.Guid).FirstOrDefault(); if (aiBook == null) { throw new ValidatedException("记录不存在."); } ValidateHelper.Validate(service, data.AiBook, AiBookEntity.DB_NAME_AI_BOOK); ObjectCopyHelper.Copy(data.AiBook, aiBook); repository.UpdateNow(aiBook); return data; } [PermissionOperate(name: AIPermissionPolicy.AiBook, operate: Operate.Delete)] [UnitOfWork] public void Delete(string key) { var ent = repository.DetachedEntities.Where(x => x.Guid == key).FirstOrDefault()!; if (ent != null) { repository.DeleteNow(ent); } } [PermissionOperate(name: AIPermissionPolicy.AiBook, operate: Operate.Delete)] [UnitOfWork] public void Deletes(string[] keys) { foreach (string key in keys) { Delete(key); } } [PermissionOperate(name: AIPermissionPolicy.AiBook, operate: Operate.Update)] [UnitOfWork] public AiBookData ProcessBook(string key, string encoding, string pattern) { AiBookData data = FindByKey(key); ChapterSplitter chapterSplitter = new ChapterSplitter(); ParagraphSplitter paragraphSplitter = new ParagraphSplitter(); IRepository repositoryDmsFile = repository.GetRepository>(); DmsFileEntity entity = repositoryDmsFile.DetachedEntities.Where(x => x.FileId == data.AiBook.FileId && x.CreatedBy == service.UserId).FirstOrDefault()!; if (entity == null) throw new ValidatedException("文件不存在."); var fileExtension = Path.GetExtension(entity.OrderKey);//获取文件格式,拓展名 string file = FileTypePolicy.PackFilePath(service, entity.CreatedBy, entity.OrderKey); if (!File.Exists(file)) throw new ValidatedException("文件不存在."); System.Text.Encoding.RegisterProvider(System.Text.CodePagesEncodingProvider.Instance); string context = File.ReadAllText(file, Encoding.GetEncoding(encoding)); List aiChapterEntities = chapterSplitter.SplitNovelIntoChapters(context, pattern); if (aiChapterEntities == null || aiChapterEntities.Count() == 0) return data; aiChapterEntities.ToLookup(x => x.BookGuid = data.AiBook.Guid); aiChapterEntities.ToLookup(x => x.Status = ChapterStatusPolicy.PROCESS); data.AiChapters = aiChapterEntities.ToArray(); paragraphSplitter.PackParagraphs(data); IRepository repositoryAiChapterEntity = repository.GetRepository>(); IRepository repositoryAiParagraphEntity = repository.GetRepository>(); IRepository repositoryAiSentenceEntity = repository.GetRepository>(); AiChapterEntity[] aiChapters = repositoryAiChapterEntity.DetachedEntities.Where(x => x.BookGuid == data.AiBook.Guid).ToArray(); if (aiChapters != null && aiChapters.Length > 0) { foreach (AiChapterEntity aiChapter in aiChapters) { AiParagraphEntity[] aiParagraphs = repositoryAiParagraphEntity.DetachedEntities.Where(x => x.ChapterGuid == aiChapter.Guid).ToArray(); if (aiParagraphs != null && aiParagraphs.Length > 0) { foreach (AiParagraphEntity aiParagraph in aiParagraphs) { AiSentenceEntity[] aiSentences = repositoryAiSentenceEntity.DetachedEntities.Where(x => x.ParagraphGuid == aiParagraph.Guid).ToArray(); if (aiSentences != null && aiSentences.Length > 0) { repositoryAiSentenceEntity.DeleteNow(aiSentences); } } repositoryAiParagraphEntity.DeleteNow(aiParagraphs); } } repositoryAiChapterEntity.DeleteNow(aiChapters); } repositoryAiChapterEntity.InsertNow(data.AiChapters); if (data.AiParagraphs == null || data.AiParagraphs.Count() == 0) return data; repositoryAiParagraphEntity.InsertNow(data.AiParagraphs); if (data.AiSentences == null || data.AiSentences.Count() == 0) return data; repositoryAiSentenceEntity.InsertNow(data.AiSentences); return FindByKey(key); } } }