package main

import (
	"context"
	"encoding/json"
	"fmt"
	"io/ioutil"
	"log"
	"net/http"
	"net/url"
	"os"
	"path"
	"runtime"
	"strconv"
	"strings"
	"sync"
	"time"

	"github.com/schollz/progressbar"

	"github.com/chromedp/cdproto/cdp"
	"github.com/chromedp/cdproto/page"
	"github.com/chromedp/chromedp"
	"github.com/chromedp/chromedp/kb"
)

// 全局变量
const (
	homeUrl  string = "https://manhuabika.com/pLogin/"
	pageUrl  string = "https://manhuabika.com/pcomicview/?cid="
	pageUrl2 string = "https://manhuabika.com/pchapter/?cid="
)

var (
	account  string
	password string
	idList   []string
	savePath string
	PROXY    string //以上从json读取
	ctx      context.Context
	wg       sync.WaitGroup
	NmList   [100][100]string //对于漫画i，[i][0]为书的path，[i][1]-[i][n]为章节的path
	picChan  chan *pic
)

type pic struct { //图片中包含书编号,章编号,图编号
	bkNum  int
	chNum  int
	picNum int
	src    string
}

func main() {
	if len(idList) == 0 {
		fmt.Println("no books")
		return
	}
	Login() //登录

	wg.Add(1)
	go PicsDownload() //开启图片监听线程

	for index, id := range idList {
		LoadBook(index, id)
	}
	close(picChan)

	wg.Wait()
}

func standarlizeName(oriStr string) string {
	replacer := strings.NewReplacer("/", "", "|", "", "\\", "", ":", " ", "*", "", "?", "", "<", "[", ">", "]", ":", " ")
	return replacer.Replace(oriStr)
}

func PicsDownload() {
	defer wg.Done()
	picChan = make(chan *pic, 100000)
	for picture := range picChan {
		wg.Add(1)
		go PicDownload(picture)
	}
}

func PicDownload(picture *pic) {
	defer wg.Done()
	//HTTP代理
	proxyAddress, _ := url.Parse(PROXY)
	client := &http.Client{Transport: &http.Transport{Proxy: http.ProxyURL(proxyAddress)}}

	//打开链接
	req, _ := http.NewRequest("GET", picture.src, nil)
	resp, err := client.Do(req)
	Check(err)

	//读取内容
	body, err := ioutil.ReadAll(resp.Body)
	Check(err)
	defer resp.Body.Close()

	//检查路径,保存
	filePath := NmList[picture.bkNum][picture.chNum] + "/" + fmt.Sprintf("%02d", picture.picNum) + ".jpg"

	// 以写|创建的方式打开目标文件(因为文件夹在创建时就已经建好，所以k可以不用检查)
	f, err := os.OpenFile(filePath, os.O_WRONLY|os.O_CREATE, 0644)
	Check(err)
	defer f.Close()

	f.Write(body)
}

func Load(url string) {
	err := chromedp.Run(ctx,
		chromedp.Navigate(url),
		chromedp.WaitVisible(`//div[@class='appBottomMenu']`))
	Check(err)
	time.Sleep(2 * time.Second) //无论如何，等待2秒再进行下一步操作
}

func LoadBook(index int, id string) {
	bookNm := ""
	var contentNode []*cdp.Node

	Load(pageUrl + id)
	err := chromedp.Run(ctx,
		chromedp.TextContent(`//div[@class='comic-title text-start']`, &bookNm, chromedp.BySearch),
		chromedp.Nodes(`//div[@class="col-3 m-0 mb-1"]`, &contentNode))
	Check(err)

	fmt.Printf("\nStart to process book [%d]%s \n", index+1, bookNm)

	NmList[index][0] = savePath + "/" + standarlizeName(bookNm)

	//章节数
	chaps := len(contentNode)
	fmt.Printf("\nTotal: %d Chapters \n", chaps)

	//章节base1
	for i := 1; i <= chaps; i++ {
		chapNm := ""
		chromedp.Run(ctx, chromedp.TextContent(contentNode[chaps-i].PartialXPath(), &chapNm))
		chapNm = standarlizeName(chapNm)
		fmt.Printf("[%d] %s \n", i, chapNm)
		NmList[index][i] = NmList[index][0] + "/[" + fmt.Sprintf("%02d", i) + "] " + chapNm //章节数补全2位,加上章节名

		if !pathExists(NmList[index][i]) { //建好文件夹
			err = os.MkdirAll(NmList[index][i], 0766)
			Check(err)
		}
	}
	fmt.Printf("\n")
	bar := progressbar.New(chaps) //显示一个进度条，用来展示每本书的进度
	for i := 1; i <= chaps; i++ {
		LoadChap(index, i, bar)
	}
}

func LoadChap(bookNum int, chapNum int, bar *progressbar.ProgressBar) {
	//分为两步，下拉
	Load(pageUrl2 + idList[bookNum] + "&chapter=" + strconv.Itoa(chapNum))
	ScrollDown()

	var picNodes []*cdp.Node
	err := chromedp.Run(ctx,
		chromedp.Nodes(`//div[@class='chapter-images wide-block pt-2 pb-2 my-bg-white']/img`, &picNodes))
	Check(err)

	// println("total Picnum = ", len(picNodes))

	for i := 0; i < len(picNodes); i++ {
		picture := &pic{bookNum, chapNum, i, picNodes[i].AttributeValue("data-src")}
		picChan <- picture
	}

	bar.Add(1) //标记这章节已经完成
}

func ScrollDown() {
	var prgsBar string
	for i := 0; ; {
		err := chromedp.Run(ctx,
			chromedp.Evaluate("window.scrollBy(0,document.body.scrollHeight)", nil),
			chromedp.Sleep(2*time.Second),
			chromedp.TextContent(`.//div[@class="w-100 text-center text-black-50 my-read-tip"]`, &prgsBar))
		Check(err)
		if prgsBar == "100 % （點擊可跳轉）" {
			if i > 2 {
				break
			} else {
				i++
			}
		} else {
			if i != 0 {
				i = 0
			}
		}
	}
}

func Login() {
	err := chromedp.Run(ctx,
		chromedp.Navigate(homeUrl),
		chromedp.WaitVisible("#email1", chromedp.ByID),
		chromedp.SendKeys("#email1", account+kb.Tab+password, chromedp.ByID),
		chromedp.Click(`.//button[@type="submit"]`, chromedp.BySearch),
		chromedp.WaitVisible(`//div[@class='appBottomMenu']`))
	Check(err)
	fmt.Println("login complete")

	fmt.Println("\n...you've 15 second to change the picture quality...")
	time.Sleep(100)
	Load(pageUrl2 + idList[0] + "&chapter=1")
	time.Sleep(10 * time.Second)
	fmt.Println("processing start")

}

func pathExists(path string) bool {
	_, err := os.Stat(path)
	if err == nil {
		return true
	}
	if os.IsNotExist(err) {
		return false
	}
	return false
}

func init() { //chromedp初始化
	headlessFlag := chromedp.Flag("headless", false) //有头模式
	opts := append(
		chromedp.DefaultExecAllocatorOptions[:],
		chromedp.NoDefaultBrowserCheck,                               //不检查默认浏览器
		headlessFlag,                                                 //无头
		chromedp.IgnoreCertErrors,                                    //忽略错误
		chromedp.Flag("blink-settings", "imagesEnabled=false"),       //不加载gif图像 因为有可能会卡住
		chromedp.DisableGPU,                                          //关闭GPU渲染
		chromedp.NoSandbox,                                           //不适用谷歌的sanbox模式运行
		chromedp.NoFirstRun,                                          //设置网站不是首次运行
		chromedp.Flag("disable-web-security", true),                  //禁用网络安全标志
		chromedp.Flag("disable-extensions", true),                    //关闭插件支持
		chromedp.Flag("disable-default-apps", true),                  //关闭默认浏览器检查
		chromedp.WindowSize(1280, 1024),                              //初始大小
		chromedp.Flag("run-all-compositor-stages-before-draw", true), //在呈现所有数据之前防止创建Pdf
		chromedp.UserAgent(`Mozilla/5.0 (Windows NT 6.3; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/73.0.3683.103 Safari/537.36`), //设置UserAgent
	)

	allocCtx, _ := chromedp.NewExecAllocator(context.Background(), opts...)
	ctx, _ = chromedp.NewContext(
		allocCtx,
		chromedp.WithLogf(log.Printf),
	)
	chromedp.Run(ctx, make([]chromedp.Action, 0, 1)...) //打开，但是什么都不干
	fmt.Println("-- chromedp start success --")
}

func init() { //customsettings初始化
	type custom struct {
		Account  string   `json:"Account"`
		Password string   `json:"Password"`
		IDList   []string `json:"IDList"` //漫画列表
		SavePath string   `json:"SavePath"`
		PROXY    string   `json:"PROXY"`
	}

	_, projectPath, _, _ := runtime.Caller(0)
	filePath := path.Dir(projectPath) + "/custom.json"

	fmt.Println(">fetch customInfo from", filePath)

	jsonFile, err := os.Open(filePath)
	Check(err)
	defer jsonFile.Close()

	jsonData, err := ioutil.ReadAll(jsonFile)
	Check(err)

	var customInfo custom
	json.Unmarshal(jsonData, &customInfo)
	account, password, idList, savePath, PROXY = customInfo.Account, customInfo.Password, customInfo.IDList, customInfo.SavePath, customInfo.PROXY

	if PROXY == "" {
		PROXY = "http://127.0.0.1:1080/"
	}

	fmt.Println("-- custom.json loadng success --")
	fmt.Println("Welcome!", account)
	fmt.Println("your savePath:", savePath)
	fmt.Println("your cidList:", idList)
	fmt.Println("your proxy:", idList)
}

func Check(err error) {
	if err != nil {
		log.Fatal(err)
	}
}