From fb96d02f5871d41a81ff67a973af158c9078e445 Mon Sep 17 00:00:00 2001
From: Zero <LordZero25@gmail.com>
Date: Sun, 22 Jan 2017 17:20:52 +0800
Subject: [PATCH] Version 2.2

---
 README.md                            |   7 +-
 syosetuDownloader/Constants.cs       |   9 +-
 syosetuDownloader/MainWindow.xaml    |   9 +-
 syosetuDownloader/MainWindow.xaml.cs | 110 +++++++++++-------
 syosetuDownloader/Syousetsu.cs       | 164 +++++++++++++++++++++++++--
 syosetuDownloader/format.ini         |   4 +-
 6 files changed, 240 insertions(+), 63 deletions(-)

diff --git a/README.md b/README.md
index 2a21544..c3ca577 100644
--- a/README.md
+++ b/README.md
@@ -7,8 +7,9 @@ Download chapters from Syosetsu ni Narou
 - Option to specify a chapter range i.e. if you want to only download say chapters 3-10
 - Download mutiple series
 - Clicking the progress bar will cancel the download job
-- Clicking the novel title will open it download folder
+- Clicking the novel title will open its download folder
 - Custom filename (see format.ini for details)
+- Chapter list generation (nxxxxxx.htm)
 
 **Notes:**
 - If no chapter range is specified it will download everything
@@ -17,8 +18,6 @@ Download chapters from Syosetsu ni Narou
 - No volume support
 - "chapter 1" is "http://*.syosetu.com/xxxxxxx/1"
 - Replaces illegal characters on filename with "□"
-
-**ToDo:**
-- Table of contents creation
+- No chapter list generation for wayback machine
 
 **[Releases Here](https://github.com/LordZero25/Syosetu-Downloader/releases/)**
diff --git a/syosetuDownloader/Constants.cs b/syosetuDownloader/Constants.cs
index 3623c41..79ba4ef 100644
--- a/syosetuDownloader/Constants.cs
+++ b/syosetuDownloader/Constants.cs
@@ -31,6 +31,7 @@ public enum FileType { Text, HTML };
         string _link = String.Empty;
         string _seriesCode = String.Empty;
         private string _fileNameFormat = String.Empty;
+        private List<string> chapterTitle = new List<string>();
 
         public CookieContainer SyousetsuCookie
         {
@@ -46,7 +47,7 @@ public CookieContainer SyousetsuCookie
             }
         }
 
-        public string Title
+        public string SeriesTitle
         {
             get { return _title; }
             set { _title = value; }
@@ -94,5 +95,11 @@ public string FilenameFormat
             get { return _fileNameFormat; }
             set { _fileNameFormat = value; }
         }
+
+        public List<string> ChapterTitle
+        {
+            get { return chapterTitle; }
+            set { chapterTitle = value; }
+        }
     }
 }
diff --git a/syosetuDownloader/MainWindow.xaml b/syosetuDownloader/MainWindow.xaml
index 344752c..728dd48 100644
--- a/syosetuDownloader/MainWindow.xaml
+++ b/syosetuDownloader/MainWindow.xaml
@@ -1,7 +1,7 @@
 <Window x:Class="syosetuDownloader.MainWindow"
         xmlns="http://schemas.microsoft.com/winfx/2006/xaml/presentation"
         xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml"
-        Title="Narou Downloader" ResizeMode="CanMinimize" mc:Ignorable="d" xmlns:d="http://schemas.microsoft.com/expression/blend/2008" xmlns:mc="http://schemas.openxmlformats.org/markup-compatibility/2006" d:DesignHeight="501" d:DesignWidth="317" SizeToContent="WidthAndHeight" WindowStartupLocation="CenterScreen">
+        Title="Narou Downloader" ResizeMode="CanMinimize" mc:Ignorable="d" xmlns:d="http://schemas.microsoft.com/expression/blend/2008" xmlns:mc="http://schemas.openxmlformats.org/markup-compatibility/2006" d:DesignHeight="532" d:DesignWidth="317" SizeToContent="WidthAndHeight" WindowStartupLocation="CenterScreen">
     <Grid>
         <Label Content="Link:" Height="28" HorizontalAlignment="Left" Name="lblLink" VerticalAlignment="Top" Margin="26,5,0,0" />
         <TextBox Height="23" HorizontalAlignment="Left" Name="txtLink" VerticalAlignment="Top" Width="197" Margin="66,5,0,0" />
@@ -9,16 +9,17 @@
         <TextBox Height="23" HorizontalAlignment="Left" Margin="118,34,0,0" Name="txtFrom" VerticalAlignment="Top" Width="50" />
         <Label Content="To" Height="28" HorizontalAlignment="Left" Margin="180,34,0,0" Name="label2" VerticalAlignment="Top" />
         <TextBox Height="23" HorizontalAlignment="Left" Margin="213,34,0,0" Name="txtTo" VerticalAlignment="Top" Width="50" />
-        <GroupBox Header="File Type" Height="61" HorizontalAlignment="Center" Name="groupBox1" VerticalAlignment="Top" Width="237" Margin="29,66,29,0">
+        <GroupBox Header="File Type" Height="56" HorizontalAlignment="Center" Name="groupBox1" VerticalAlignment="Top" Width="237" Margin="29,66,29,0">
             <Grid>
                 <RadioButton Content="Text" Height="16" HorizontalAlignment="Center" Name="rbText" VerticalAlignment="Top" Margin="67,11,117,0" IsChecked="True" Checked="rbText_Checked" />
                 <RadioButton Content="HTML" Height="16" HorizontalAlignment="Center" Name="rbHtml" VerticalAlignment="Top" Margin="114,11,60,0" Checked="rbHtml_Checked" />
             </Grid>
         </GroupBox>
-        <Button Content="Download" Height="23" HorizontalAlignment="Center" Name="btnDownload" VerticalAlignment="Top" Width="107" Margin="94,138,94,0" Click="btnDownload_Click" />
-        <ScrollViewer Height="255" HorizontalAlignment="Left" Margin="0,167,0,0" Name="scrollViewer1" VerticalAlignment="Top" Width="295" VerticalScrollBarVisibility="Auto" HorizontalScrollBarVisibility="Auto">
+        <Button Content="Download" Height="23" HorizontalAlignment="Center" Name="btnDownload" VerticalAlignment="Top" Width="107" Margin="94,151,94,0" Click="btnDownload_Click" />
+        <ScrollViewer Height="255" HorizontalAlignment="Left" Margin="0,193,0,0" Name="scrollViewer1" VerticalAlignment="Top" Width="295" VerticalScrollBarVisibility="Auto" HorizontalScrollBarVisibility="Auto">
             <StackPanel Name="stackPanel1" />
         </ScrollViewer>
         <Button Content="Delete Completed" Height="23" HorizontalAlignment="Center" Name="btnDelete" VerticalAlignment="Bottom" Width="103" Margin="96,0,96,10" Click="btnDelete_Click" />
+        <CheckBox Content="Generate Chapter List" Height="16" HorizontalAlignment="Center" Name="chkList" VerticalAlignment="Top" Margin="80,128,80,0" />
     </Grid>
 </Window>
diff --git a/syosetuDownloader/MainWindow.xaml.cs b/syosetuDownloader/MainWindow.xaml.cs
index adfe8ea..c632fac 100644
--- a/syosetuDownloader/MainWindow.xaml.cs
+++ b/syosetuDownloader/MainWindow.xaml.cs
@@ -48,64 +48,88 @@ public MainWindow()
 
         private void btnDownload_Click(object sender, RoutedEventArgs e)
         {
-            _row += 1;
-
             this._link = txtLink.Text;
             this._start = txtFrom.Text;
             this._end = txtTo.Text;
 
-            if (_link != String.Empty)
+            bool fromToValid = (System.Text.RegularExpressions.Regex.IsMatch(_start, @"^\d+$") || _start.Equals(String.Empty)) &&
+                (System.Text.RegularExpressions.Regex.IsMatch(_end, @"^\d+$") || _end.Equals(String.Empty));
+            if (_link != String.Empty && fromToValid)
             {
                 if (!_link.StartsWith("http")) { _link = @"http://" + _link; }
                 if (!_link.EndsWith("/")) { _link = _link + "/"; }
 
-                Syousetsu.Constants sc = new Syousetsu.Constants();
-                HtmlDocument toc = Syousetsu.Methods.GetTableOfContents(_link, sc.SyousetsuCookie);
-                if (Syousetsu.Methods.IsValidLink(_link) &&
-                    Syousetsu.Methods.IsValid(toc))
+                if (Syousetsu.Methods.IsValidLink(_link))
                 {
-                    Label lb = new Label();
-                    lb.Content = Syousetsu.Methods.GetTitle(toc);
-                    lb.ToolTip = "Click to open folder";
-
-                    ProgressBar pb = new ProgressBar();
-                    pb.Maximum = (_end == String.Empty) ? Syousetsu.Methods.GetTotalChapters(toc) : Convert.ToDouble(_end);
-                    pb.ToolTip = "Click to stop download";
-                    pb.Height = 10;
-
-                    Separator s = new Separator();
-                    s.Height = 5;
-
-                    sc.Title = lb.Content.ToString();
-                    sc.Link = _link;
-                    sc.Start = _start;
-                    sc.End = _end;
-                    sc.CurrentFileType = _fileType;
-                    sc.SeriesCode = Syousetsu.Methods.GetSeriesCode(_link);
-                    sc.FilenameFormat = _format;
-
-                    System.Threading.CancellationTokenSource ct = Syousetsu.Methods.AddDownloadJob(sc, pb);
-                    pb.MouseDown += (snt, evt) =>
-                    {
-                        ct.Cancel();
-                        pb.ToolTip = "";
-                    };
-                    lb.MouseDown += (snt, evt) =>
+                    Syousetsu.Constants sc = new Syousetsu.Constants();
+                    sc.ChapterTitle.Add("");
+                    HtmlDocument toc = Syousetsu.Methods.GetTableOfContents(_link, sc.SyousetsuCookie);
+
+                    if (Syousetsu.Methods.IsValid(toc))
                     {
-                        string path = System.IO.Path.Combine(System.IO.Path.GetDirectoryName(System.Reflection.Assembly.GetExecutingAssembly().Location), sc.Title);
-                        if (System.IO.Directory.Exists(path))
+                        _row += 1;
+
+                        Label lb = new Label();
+                        lb.Content = Syousetsu.Methods.GetTitle(toc);
+                        lb.ToolTip = "Click to open folder";
+
+                        ProgressBar pb = new ProgressBar();
+                        pb.Maximum = (_end == String.Empty) ? Syousetsu.Methods.GetTotalChapters(toc) : Convert.ToDouble(_end);
+                        pb.ToolTip = "Click to stop download";
+                        pb.Height = 10;
+
+                        Separator s = new Separator();
+                        s.Height = 5;
+
+                        sc.SeriesTitle = lb.Content.ToString();
+                        sc.Link = _link;
+                        sc.Start = _start;
+                        sc.End = _end;
+                        sc.CurrentFileType = _fileType;
+                        sc.SeriesCode = Syousetsu.Methods.GetSeriesCode(_link);
+                        sc.FilenameFormat = _format;
+                        Syousetsu.Methods.GetAllChapterTitles(sc, toc);
+
+                        if (chkList.IsChecked == true)
                         {
-                            System.Diagnostics.Process.Start("explorer.exe", path);
+                            Syousetsu.Create.GenerateTableOfContents(sc, toc);
                         }
-                    };
-
-                    stackPanel1.Children.Add(lb);
-                    stackPanel1.Children.Add(pb);
-                    stackPanel1.Children.Add(s);
 
-                    _controls.Add(new Syousetsu.Controls { ID = _row, Label = lb, ProgressBar = pb, Separator = s });
+                        System.Threading.CancellationTokenSource ct = Syousetsu.Methods.AddDownloadJob(sc, pb);
+                        pb.MouseDown += (snt, evt) =>
+                        {
+                            ct.Cancel();
+                            pb.ToolTip = null;
+                        };
+                        lb.MouseDown += (snt, evt) =>
+                        {
+                            string path = System.IO.Path.Combine(System.IO.Path.GetDirectoryName(System.Reflection.Assembly.GetExecutingAssembly().Location), sc.SeriesTitle);
+                            if (System.IO.Directory.Exists(path))
+                            {
+                                System.Diagnostics.Process.Start("explorer.exe", path);
+                            }
+                        };
+
+                        stackPanel1.Children.Add(lb);
+                        stackPanel1.Children.Add(pb);
+                        stackPanel1.Children.Add(s);
+
+                        _controls.Add(new Syousetsu.Controls { ID = _row, Label = lb, ProgressBar = pb, Separator = s });
+                    }
+                    else
+                    {
+                        MessageBox.Show("Link not valid!", "Error", MessageBoxButton.OK, MessageBoxImage.Error);
+                    }
+                }
+                else
+                {
+                    MessageBox.Show("Link not valid!", "Error", MessageBoxButton.OK, MessageBoxImage.Error);
                 }
             }
+            else
+            {
+                MessageBox.Show("Error parsing link and/or chapter range!", "", MessageBoxButton.OK, MessageBoxImage.Error);
+            }
         }
 
         private void rbText_Checked(object sender, RoutedEventArgs e)
diff --git a/syosetuDownloader/Syousetsu.cs b/syosetuDownloader/Syousetsu.cs
index debf6f8..ac8f585 100644
--- a/syosetuDownloader/Syousetsu.cs
+++ b/syosetuDownloader/Syousetsu.cs
@@ -60,9 +60,10 @@ public static CancellationTokenSource AddDownloadJob(Syousetsu.Constants details
                         break;
                     }
                 }
-                pb.Dispatcher.Invoke((Action)(() => { 
+                pb.Dispatcher.Invoke((Action)(() =>
+                {
                     pb.Value = max;
-                    pb.ToolTip = "";
+                    pb.ToolTip = null;
                 }));
             }, ct.Token);
 
@@ -194,6 +195,22 @@ public static string GetSeriesCode(string link)
 
             return m.Groups["seriesCode"].Value;
         }
+
+        public static void GetAllChapterTitles(Syousetsu.Constants details, HtmlDocument doc)
+        {
+            //edit all href
+            HtmlNodeCollection chapterNode = doc.DocumentNode.SelectNodes("//div[@class='index_box']/dl[@class='novel_sublist2']");
+            foreach (HtmlNode node in chapterNode)
+            {
+                //get current chapter number
+                string pattern = "(href=\"/)(?<series>.+)/(?<num>.+)/\">(?<title>.+)(?=</a>)";
+                Regex r = new Regex(pattern);
+                Match m = r.Match(node.ChildNodes["dd"].OuterHtml);
+                int current = Convert.ToInt32(m.Groups["num"].Value);
+                details.ChapterTitle.Add(m.Groups["title"].Value);
+
+            }
+        }
     }
 
     public class Create
@@ -221,7 +238,7 @@ public static string[] GenerateContents(Syousetsu.Constants details, HtmlDocumen
                 chapter[1] += "<html xmlns=\"http://www.w3.org/1999/xhtml\">\n";
                 chapter[1] += "<head>\n";
                 chapter[1] += "\t<title></title>\n";
-                chapter[1] += "\t<link href=\"../Styles/Style0001.css\" rel=\"stylesheet\" type=\"text/css\" />\n";
+                chapter[1] += "\t<link href=\"ChapterStyle.css\" rel=\"stylesheet\" type=\"text/css\" />\n";
                 chapter[1] += "</head>\n";
                 chapter[1] += "<body>\n";
                 chapter[1] += "\n<h2>" + chapter[0] + "</h2>\n\n";
@@ -238,10 +255,7 @@ public static string[] GenerateContents(Syousetsu.Constants details, HtmlDocumen
 
         public static void SaveFile(Syousetsu.Constants details, string[] chapter, int current)
         {
-            //create novel folder if it doesn't exist
-            string path = Path.Combine(Path.GetDirectoryName(System.Reflection.Assembly.GetExecutingAssembly().Location), details.Title);
-            bool folderExists = Directory.Exists(path);
-            if (!folderExists) { Directory.CreateDirectory(path); }
+            string path = CheckDirectory(details, current);
 
             //replace illigal character(s) with "□"
             string regexSearch = new string(Path.GetInvalidFileNameChars()) + new string(Path.GetInvalidPathChars());
@@ -249,17 +263,147 @@ public static void SaveFile(Syousetsu.Constants details, string[] chapter, int c
             chapter[0] = r.Replace(chapter[0], "□");
 
             //save the chapter
-            string fileName = details.FilenameFormat;
+            string fileName = details.FilenameFormat.Replace("/", "\\")
+                    .Split('\\').Last();
             if (details.CurrentFileType == Constants.FileType.Text)
             {
-                fileName = String.Format(fileName + ".txt", current, chapter[0]);
+                fileName = String.Format(fileName + ".txt",
+                    new object[] { current, chapter[0], details.SeriesCode });
             }
             else if (details.CurrentFileType == Constants.FileType.HTML)
             {
-                fileName = String.Format(fileName + ".htm", current, chapter[0]);
+                fileName = String.Format(fileName + ".htm",
+                    new object[] { current, chapter[0], details.SeriesCode });
+                chapter[0] = String.Empty;
+
+                File.WriteAllText(Path.Combine(path, "ChapterStyle.css"), "/*chapter css here*/");
             }
             fileName = Path.Combine(path, fileName);
             File.WriteAllLines(fileName, chapter, Encoding.Unicode);
         }
+
+        public static void GenerateTableOfContents(Syousetsu.Constants details, HtmlDocument doc)
+        {
+            //create novel folder if it doesn't exist
+            CheckDirectory(details);
+
+            HtmlNode tocNode = doc.DocumentNode.SelectSingleNode("//div[@class='index_box']");
+            HtmlNodeCollection cssNodeList = doc.DocumentNode.SelectNodes("//link[@rel='stylesheet']");
+
+            var cssNode = (from n in cssNodeList
+                           where n.Attributes["href"].Value.Contains("syosetu")
+                           select n).ToList();
+
+            //get css link and download
+            string pattern = "(href=\")(?<link>.+)(?=\" media)";
+            Regex r = new Regex(pattern);
+            Match m = r.Match(cssNode[0].OuterHtml);
+            string cssink = m.Groups["link"].Value;
+            DownloadCss(details, cssink);
+
+            StringBuilder sb = new StringBuilder();
+            sb.AppendLine("<html>");
+            sb.AppendLine("<head>");
+            sb.AppendLine("\t<meta charset=\"UTF-8\">");
+            sb.AppendLine("\t<link rel=\"stylesheet\" type=\"text/css\" href=\"" + details.SeriesCode + ".css\" media=\"screen,print\" />");
+            sb.AppendLine("</head>");
+
+            sb.AppendLine("<body>");
+
+            //edit all href
+            int i = 1;
+            HtmlNodeCollection chapterNode = doc.DocumentNode.SelectNodes("//div[@class='index_box']/dl[@class='novel_sublist2']");
+            foreach (HtmlNode node in chapterNode)
+            {
+                //get current chapter number
+                pattern = "(href=\"/)(?<series>.+)/(?<num>.+)/\">(?<title>.+)(?=</a>)";
+                r = new Regex(pattern);
+                m = r.Match(node.ChildNodes["dd"].OuterHtml);
+                int current = Convert.ToInt32(m.Groups["num"].Value);
+                details.ChapterTitle.Add(m.Groups["title"].Value);
+
+                //edit href
+                string fileName = details.FilenameFormat;
+                fileName = String.Format(fileName + ".htm", current, details.ChapterTitle[current], details.SeriesCode);
+                node.ChildNodes["dd"].ChildNodes["a"].Attributes["href"].Value = "./" + fileName;
+
+                if (i <= Convert.ToInt32(details.End))
+                {
+                    CheckDirectory(details, current);
+                }
+                i++;
+            }
+            sb.AppendLine(tocNode.OuterHtml);
+
+            sb.AppendLine("</body>");
+            sb.AppendLine("</html>");
+
+            File.WriteAllText(Path.Combine(details.Path, details.SeriesTitle, details.SeriesCode + ".htm"), sb.ToString());
+        }
+
+        private static void DownloadCss(Syousetsu.Constants details, string link)
+        {
+            HttpWebRequest request = (HttpWebRequest)WebRequest.Create(link);
+            request.Method = "GET";
+
+            HttpWebResponse response = (HttpWebResponse)request.GetResponse();
+            var stream = response.GetResponseStream();
+
+            using (StreamReader reader = new StreamReader(stream))
+            {
+                string css = reader.ReadToEnd();
+                File.WriteAllText(Path.Combine(details.Path, details.SeriesTitle, details.SeriesCode + ".css"), css);
+            }
+        }
+
+        private static string CheckDirectory(Syousetsu.Constants details)
+        {
+            string path;
+            if (!details.FilenameFormat.Contains('/') && !details.FilenameFormat.Contains('\\'))
+            {
+                path = Path.Combine(Path.GetDirectoryName(System.Reflection.Assembly.GetExecutingAssembly().Location), details.SeriesTitle);
+            }
+            else
+            {
+                string[] temp = details.FilenameFormat
+                    .Replace("/", "\\")
+                    .Split('\\');
+                temp = temp.Take(temp.Length - 1).ToArray();
+                string tempFormat = (temp.Length > 1) ? String.Join("\\", temp) : temp[0];
+
+                path = Path.Combine(new string[] { 
+                    Path.GetDirectoryName(System.Reflection.Assembly.GetExecutingAssembly().Location), 
+                    details.SeriesTitle,
+                    String.Format(tempFormat, new object[]{ 0, details.ChapterTitle[0], details.SeriesCode})
+                });
+            }
+            if (!Directory.Exists(path)) { Directory.CreateDirectory(path); }
+            return path;
+        }
+
+        private static string CheckDirectory(Syousetsu.Constants details, int current)
+        {
+            string path;
+            if (!details.FilenameFormat.Contains('/') && !details.FilenameFormat.Contains('\\'))
+            {
+                path = Path.Combine(Path.GetDirectoryName(System.Reflection.Assembly.GetExecutingAssembly().Location), details.SeriesTitle);
+            }
+            else
+            {
+                string[] temp = details.FilenameFormat
+                    .Replace("/", "\\")
+                    .Split('\\');
+                temp = temp.Take(temp.Length - 1).ToArray();
+                string tempFormat = (temp.Length > 1) ? String.Join("\\", temp) : temp[0];
+
+                path = Path.Combine(new string[] { 
+                    Path.GetDirectoryName(System.Reflection.Assembly.GetExecutingAssembly().Location), 
+                    details.SeriesTitle,
+                    String.Format(tempFormat, new object[]{ current, details.ChapterTitle[current], details.SeriesCode})
+                });
+            }
+            if (!Directory.Exists(path)) { Directory.CreateDirectory(path); }
+            return path;
+        }
     }
 }
diff --git a/syosetuDownloader/format.ini b/syosetuDownloader/format.ini
index 83aa903..250a131 100644
--- a/syosetuDownloader/format.ini
+++ b/syosetuDownloader/format.ini
@@ -1,6 +1,8 @@
 ;{0} = incrementing numbers
 ;{1} = chapter title
+;{2} = series code (n******)
 ;Chapter {0} - {1} => Chapter 1 - Prolouge
 ;{0:D4} => 0001
 ;Reffer to https://msdn.microsoft.com/en-us/library/dwhawy9k(v=vs.100).aspx
-Chapter {0} - {1}
\ No newline at end of file
+;Chapter {0} - {1}
+{2}/{0:0000}-{1}
\ No newline at end of file