Пример #1
0
        public VSButton LoadUri(Uri Uri, IHost Host)
        {
            Status = CurrentLanguage.LoadingComic;

            CurrentHost = CreateInstance(Host);

            CurrentInfo     = CurrentHost.LoadUri(Uri);
            TitleLabel.Text = CurrentInfo.Title;
            CoverBox.Image  = CurrentHost.GetDecoder().Decode(CurrentInfo.Cover);

            if (CurrentInfo.Url == null)
            {
                CurrentInfo.Url = Uri;
            }

            CoverBox.Cursor = Cursors.Default;

            string Dir = DataTools.GetRawName(CurrentInfo.Title.Trim(), FileNameMode: true);

            ChapterTools.MatchLibraryPath(ref Dir, Settings.LibraryPath, CurrentInfo.Url, ReplaceMode.UpdateURL, CurrentLanguage);
            RefreshCoverLink(Dir);

            var OnlinePath = Path.Combine(Settings.LibraryPath, Dir, "Online.url");

            if (Directory.Exists(Dir) && !File.Exists(OnlinePath))
            {
                var OnlineData = string.Format(Properties.Resources.UrlFile, CurrentInfo.Url.AbsoluteUri);
                File.WriteAllText(OnlinePath, OnlineData);
            }

            ButtonsContainer.Controls.Clear();

            VSButton DownAll = null;

            var Chapters = new Dictionary <int, string>();

            foreach (var Chapter in CurrentHost.EnumChapters())
            {
                if (Chapters.ContainsValue(Chapter.Value))
                {
                    continue;
                }

                VSButton Button = new VSButton()
                {
                    Size        = new Size(110, 30),
                    Text        = string.Format(CurrentLanguage.ChapterName, Chapter.Value),
                    Indentifier = CurrentHost
                };

                Button.Click += (sender, args) =>
                {
                    try
                    {
                        IHost HostIsnt = (IHost)((VSButton)sender).Indentifier;
                        DownloadChapter(Chapters, Chapter.Key, HostIsnt, CurrentInfo);
                    }
                    catch (Exception ex)
                    {
                        if (Program.Debug)
                        {
                            throw;
                        }
                    }
                    StatusBar.SecondLabelText = string.Empty;
                    Status = CurrentLanguage.IDLE;
                };

                DownAll = Button;

                ButtonsContainer.Controls.Add(Button);
                Chapters.Add(Chapter.Key, Chapter.Value);
            }

            if (Chapters.Count > 1)
            {
                VSButton Button = new VSButton()
                {
                    Size        = new Size(110, 30),
                    Text        = CurrentLanguage.DownloadAll,
                    Indentifier = CurrentHost
                };

                Button.Click += (sender, args) =>
                {
                    foreach (var Chapter in Chapters.Reverse())
                    {
                        try
                        {
                            IHost HostIsnt = (IHost)((VSButton)sender).Indentifier;
                            DownloadChapter(Chapters, Chapter.Key, HostIsnt, CurrentInfo);
                        }
                        catch (Exception ex)
                        {
                            if (Program.Debug)
                            {
                                throw;
                            }
                        }

                        StatusBar.SecondLabelText = string.Empty;
                        Status = CurrentLanguage.IDLE;
                    }
                };
                DownAll = Button;
                ButtonsContainer.Controls.Add(Button);
            }

            var PActions = CurrentHost.GetPluginInfo().Actions;

            if (PActions != null)
            {
                var Actions = (from x in PActions where x.Availability.HasFlag(ActionTo.ChapterList) select x);

                if (Actions.Any())
                {
                    foreach (var Action in Actions)
                    {
                        VSButton Bnt = new VSButton()
                        {
                            Size        = new Size(110, 30),
                            Text        = Action.Name,
                            Indentifier = CurrentHost
                        };
                        Bnt.Click += (sender, args) => Action.Action();
                        ButtonsContainer.Controls.Add(Bnt);
                    }
                    ;
                }
            }

            Status = CurrentLanguage.IDLE;

            return(DownAll);
        }
Пример #2
0
        private void DownloadChapter(Dictionary <int, string> Chapters, int ID, IHost Host, ComicInfo Info)
        {
            string Name = DataTools.GetRawName(Chapters[ID], FileNameMode: true);

            StatusBar.SecondLabelText = $"{Info.Title} - {string.Format(CurrentLanguage.ChapterName, Name)}";
            Status = CurrentLanguage.Loading;

            int KIndex = Chapters.IndexOfKey(ID);

            string LName          = null;
            string LastChapterPah = null;

            if (KIndex + 1 < Chapters.Keys.Count)
            {
                LName = DataTools.GetRawName(Chapters.Values.ElementAt(KIndex + 1), FileNameMode: true);
            }

            string NName           = null;
            string NextChapterPath = null;

            if (KIndex > 0)
            {
                NName = DataTools.GetRawName(Chapters.Values.ElementAt(KIndex - 1), FileNameMode: true);
            }

            string Title = DataTools.GetRawName(Info.Title.Trim(), FileNameMode: true);

            ChapterTools.MatchLibraryPath(ref Title, Settings.LibraryPath, CurrentInfo.Url, (ReplaceMode)Settings.ReplaceMode, CurrentLanguage);
            RefreshCoverLink(Title);

            string TitleDir = Path.Combine(Settings.LibraryPath, Title);

            if (!Directory.Exists(TitleDir))
            {
                Directory.CreateDirectory(TitleDir);
                RefreshLibrary();
            }

            if (LName != null)
            {
                ChapterTools.GetChapterPath(Languages, CurrentLanguage, TitleDir, LName, out LastChapterPah, false);
            }

            ChapterTools.GetChapterPath(Languages, CurrentLanguage, TitleDir, Name, out string ChapterPath, false);

            if (NName != null)
            {
                ChapterTools.GetChapterPath(Languages, CurrentLanguage, TitleDir, NName, out NextChapterPath, false);
            }

            string AbsoluteChapterPath = Path.Combine(TitleDir, ChapterPath);

            if (Settings.SkipDownloaded && File.Exists(AbsoluteChapterPath.TrimEnd('\\', '/') + ".html"))
            {
                return;
            }

            int PageCount = 1;

            if (Info.ContentType == ContentType.Comic)
            {
                PageCount = Host.GetChapterPageCount(ID);

                if (Directory.Exists(AbsoluteChapterPath) && Directory.GetFiles(AbsoluteChapterPath, "*").Length < PageCount)
                {
                    Directory.Delete(AbsoluteChapterPath, true);
                }

                if (Settings.SkipDownloaded && Directory.Exists(AbsoluteChapterPath))
                {
                    var Pages = (from x in Directory.GetFiles(AbsoluteChapterPath) select Path.GetFileName(x)).ToArray();
                    ChapterTools.GenerateComicReader(CurrentLanguage, Pages, LastChapterPah, NextChapterPath, TitleDir, ChapterPath, Name);
                    string OnlineData = string.Format(Properties.Resources.UrlFile, Info.Url.AbsoluteUri);
                    File.WriteAllText(Path.Combine(TitleDir, "Online.url"), OnlineData);
                    return;
                }
            }

            if (Info.ContentType == ContentType.Novel)
            {
                ChapterPath = Path.Combine(AbsoluteChapterPath, string.Format(CurrentLanguage.ChapterName, Name) + ".epub");
                if (Settings.SkipDownloaded && File.Exists(ChapterPath))
                {
                    return;
                }
            }


            if (!Directory.Exists(AbsoluteChapterPath))
            {
                Directory.CreateDirectory(AbsoluteChapterPath);
            }

            string UrlData = string.Format(Properties.Resources.UrlFile, Info.Url.AbsoluteUri);

            File.WriteAllText(Path.Combine(TitleDir, "Online.url"), UrlData);


            switch (Info.ContentType)
            {
            case ContentType.Comic:
                var           Decoder = Host.GetDecoder();
                List <string> Pages   = new List <string>();
                foreach (var Data in Host.DownloadPages(ID).CatchExceptions())
                {
                    Status = string.Format(CurrentLanguage.Downloading, Pages.Count + 1, PageCount);
                    Application.DoEvents();

                    bool IsWebP = Data.Length > 12 && BitConverter.ToUInt32(Data, 8) == 0x50424557;

                    try
                    {
#if NOASYNCSAVE
                        string PageName = $"{Pages.Count:D3}.png";
                        string PagePath = Path.Combine(TitleDir, ChapterPath, PageName);

                        Page OutPage = new Page();
                        OutPage.Data = Data;

                        if ((SaveAs)Settings.SaveAs != SaveAs.RAW)
                        {
                            using (Bitmap Result = Decoder.Decode(Data))
                            {
                                PageName = $"{Pages.Count:D3}.{GetExtension(Result, out ImageFormat Format)}";
                                PagePath = OutPage.Path = Path.Combine(TitleDir, ChapterPath, PageName);

                                if (Result.GetImageExtension() == "png" && Settings.APNGBypass)
                                {
                                    var OutData = BypassAPNG(Data);
                                    OutPage.Data = OutData;
                                }
                                else
                                {
                                    using (MemoryStream tmp = new MemoryStream())
                                    {
                                        Result.Save(tmp, Format);
                                        OutPage.Data = tmp.ToArray();
                                    }
                                }
                            }
                        }

                        if (Settings.ImageClipping)
                        {
                            PostProcessQueue.Enqueue(OutPage);
                        }
                        else
                        {
                            File.WriteAllBytes(OutPage.Path, OutPage.Data);
                        }

                        while (PostProcessQueue.Count > 0)
                        {
                            ThreadTools.Wait(1000, true);
                        }
#else
                        string PageName = $"{Pages.Count:D3}.{(IsWebP ? "webp" : "png")}";
                        string PagePath = Path.Combine(TitleDir, ChapterPath, PageName);

                        Page OutPage = new Page();
                        OutPage.Data = Data;

                        if ((SaveAs)Settings.SaveAs != SaveAs.RAW)
                        {
                            byte[] ModData = null;

                            if (IsWebP)
                            {
                                ModData = DecodeWebP(Data);
                            }

                            using (MemoryStream Buffer = new MemoryStream())
                                using (Bitmap Result = Decoder.Decode(ModData ?? Data))
                                {
                                    PageName = $"{Pages.Count:D3}.{GetExtension(Result, out ImageFormat Format)}";
                                    PagePath = Path.Combine(TitleDir, ChapterPath, PageName);
                                    if (Result.GetImageExtension() == "png" && Settings.APNGBypass)
                                    {
                                        using (MemoryStream OutBuffer = new MemoryStream(BypassAPNG(ModData ?? Data)))
                                            using (Bitmap Img = new Bitmap(OutBuffer)) {
                                                Img.Save(Buffer, Format);
                                                OutPage.Data = Buffer.ToArray();
                                            }
                                    }
                                    else
                                    {
                                        Result.Save(Buffer, Format);
                                        OutPage.Data = Buffer.ToArray();
                                    }
                                }
                        }

                        OutPage.Path = PagePath;

                        if (PostProcessQueue.Count > 5)
                        {
                            const ulong MinMemory = 400000000;
                            while (AvailablePhysicalMemory < MinMemory && PostProcessQueue.Count > 0 || (Settings.MaxPagesBuffer != 0 && PostProcessQueue.Count >= Settings.MaxPagesBuffer))
                            {
                                ThreadTools.Wait(1000, true);
                            }
                        }

                        PostProcessQueue.Enqueue(OutPage);
#endif
                        Pages.Add(PageName);
                    }
                    catch (Exception ex)
                    {
                        if (Program.Debug)
                        {
                            throw;
                        }
                        Console.Error.WriteLine(ex.ToString());
                    }
                }

                if (Settings.ReaderGenerator)
                {
                    ChapterTools.GenerateComicReader(CurrentLanguage, Pages.ToArray(), LastChapterPah, NextChapterPath, TitleDir, ChapterPath, Name);
                    ChapterTools.GenerateReaderIndex(Languages, CurrentLanguage, Info, TitleDir, Name);
                }

                break;

            case ContentType.Novel:
                var Chapter = Host.DownloadChapter(ID);
                AsyncContext.Run(async() =>
                {
                    using (var Epub = await EPubWriter.CreateWriterAsync(File.Create(ChapterPath), Info.Title ?? "Untitled", Chapter.Author ?? "Anon", Chapter.URL ?? "None"))
                    {
                        Chapter.HTML.RemoveNodes("//script");
                        Chapter.HTML.RemoveNodes("//iframe");

                        foreach (var Node in Chapter.HTML.DocumentNode.DescendantsAndSelf())
                        {
                            if (Node.Name == "img" && Node.GetAttributeValue("src", string.Empty) != string.Empty)
                            {
                                string Src   = Node.GetAttributeValue("src", string.Empty);
                                string RName = Path.GetFileName(Src.Substring(null, "?", IgnoreMissmatch: true));
                                string Mime  = ResourceHandler.GetMimeType(Path.GetExtension(RName));

                                if (Node.GetAttributeValue("type", string.Empty) != string.Empty)
                                {
                                    Mime = Node.GetAttributeValue("type", string.Empty);
                                }

                                Uri Link = new Uri(Info.Url, Src);

                                if (string.IsNullOrWhiteSpace(RName))
                                {
                                    continue;
                                }

                                byte[] Buffer = await Link.TryDownloadAsync();

                                if (Buffer == null)
                                {
                                    continue;
                                }

                                await Epub.AddResourceAsync(RName, Mime, Buffer);
                                Node.SetAttributeValue("src", RName);
                            }

                            if (Node.Name == "link" && Node.GetAttributeValue("rel", string.Empty) == "stylesheet" && Node.GetAttributeValue("href", string.Empty) != string.Empty)
                            {
                                string Src   = Node.GetAttributeValue("href", string.Empty);
                                string RName = Path.GetFileName(Src.Substring(null, "?", IgnoreMissmatch: true));
                                string Mime  = ResourceHandler.GetMimeType(Path.GetExtension(RName));

                                if (Node.GetAttributeValue("type", string.Empty) != string.Empty)
                                {
                                    Mime = Node.GetAttributeValue("type", string.Empty);
                                }

                                Uri Link = new Uri(Info.Url, Src);

                                if (string.IsNullOrWhiteSpace(RName))
                                {
                                    continue;
                                }

                                byte[] Buffer = await Link.TryDownloadAsync();

                                if (Buffer == null)
                                {
                                    continue;
                                }

                                await Epub.AddResourceAsync(RName, Mime, Buffer);
                                Node.SetAttributeValue("href", RName);
                            }
                        }

                        Epub.Publisher = lblTitle.Text;

                        await Epub.AddChapterAsync("chapter.xhtml", Chapter.Title, Chapter.HTML.ToHTML());


                        await Epub.WriteEndOfPackageAsync();
                    }
                });
                break;

            default:
                throw new Exception("Invalid Content Type");
            }
        }