From cd502278357a7ee7f0f21b7cc7054f4179d492b2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BE=AE=E5=87=89?= <927625802@qq.com>
Date: Tue, 30 Nov 2021 21:30:50 +0800
Subject: [PATCH] :sparkler: support lanzou

---
 drivers/lanzou.go        | 235 +++++++++++++++++++++++++++++++++++++++
 drivers/lanzou_driver.go | 163 +++++++++++++++++++++++++++
 model/file.go            |   4 +-
 3 files changed, 401 insertions(+), 1 deletion(-)
 create mode 100644 drivers/lanzou.go
 create mode 100644 drivers/lanzou_driver.go

diff --git a/drivers/lanzou.go b/drivers/lanzou.go
new file mode 100644
index 00000000..2839b4bc
--- /dev/null
+++ b/drivers/lanzou.go
@@ -0,0 +1,235 @@
+package drivers
+
+import (
+	"fmt"
+	"github.com/Xhofe/alist/conf"
+	"github.com/Xhofe/alist/model"
+	"github.com/Xhofe/alist/utils"
+	"github.com/go-resty/resty/v2"
+	log "github.com/sirupsen/logrus"
+	"path/filepath"
+	"regexp"
+	"strconv"
+)
+
+var lanzouClient = resty.New()
+
+type LanZouFile struct {
+	Name    string `json:"name"`
+	NameAll string `json:"name_all"`
+	Id      string `json:"id"`
+	FolId   string `json:"fol_id"`
+	Size    string `json:"size"`
+	Time    string `json:"time"`
+	Folder  bool
+}
+
+func (driver *Lanzou) FormatFile(file *LanZouFile) *model.File {
+	f := &model.File{
+		Id:      file.Id,
+		Name:    file.Name,
+		Driver:  driver.Config().Name,
+		SizeStr: file.Size,
+		TimeStr: file.Time,
+	}
+	if file.Folder {
+		f.Type = conf.FOLDER
+		f.Id = file.FolId
+	} else {
+		f.Name = file.NameAll
+		f.Type = utils.GetFileType(filepath.Ext(file.NameAll))
+	}
+	return f
+}
+
+type LanZouFilesResp struct {
+	Zt   int          `json:"zt"`
+	Info interface{}  `json:"info"`
+	Text []LanZouFile `json:"text"`
+}
+
+func (driver *Lanzou) GetFiles(folderId string, account *model.Account) ([]LanZouFile, error) {
+	if account.OnedriveType == "cookie" {
+		files := make([]LanZouFile, 0)
+		var resp LanZouFilesResp
+		// folders
+		res, err := lanzouClient.R().SetResult(&resp).SetHeader("Cookie", account.AccessToken).
+			SetFormData(map[string]string{
+				"task":      "47",
+				"folder_id": folderId,
+			}).Post("https://pc.woozooo.com/doupload.php")
+		if err != nil {
+			return nil, err
+		}
+		log.Debug(res.String())
+		if resp.Zt != 1 && resp.Zt != 2 {
+			return nil, fmt.Errorf("%v", resp.Info)
+		}
+		for _, file := range resp.Text {
+			file.Folder = true
+			files = append(files, file)
+		}
+		// files
+		pg := 1
+		for {
+			_, err = lanzouClient.R().SetResult(&resp).SetHeader("Cookie", account.AccessToken).
+				SetFormData(map[string]string{
+					"task":      "5",
+					"folder_id": folderId,
+					"pg":        strconv.Itoa(pg),
+				}).Post("https://pc.woozooo.com/doupload.php")
+			if err != nil {
+				return nil, err
+			}
+			if resp.Zt != 1 {
+				return nil, fmt.Errorf("%v", resp.Info)
+			}
+			if len(resp.Text) == 0 {
+				break
+			}
+			files = append(files, resp.Text...)
+			pg++
+		}
+		return files, nil
+	} else {
+		return driver.GetFilesByUrl(account)
+	}
+}
+
+func (driver *Lanzou) GetFilesByUrl(account *model.Account) ([]LanZouFile, error) {
+	files := make([]LanZouFile, 0)
+	shareUrl := account.SiteUrl
+	res, err := lanzouClient.R().Get(shareUrl)
+	if err != nil {
+		return nil, err
+	}
+	lxArr := regexp.MustCompile(`'lx':(.+?),`).FindStringSubmatch(res.String())
+	if len(lxArr) == 0 {
+		return nil, fmt.Errorf("get empty page")
+	}
+	lx := lxArr[1]
+	fid := regexp.MustCompile(`'fid':(.+?),`).FindStringSubmatch(res.String())[1]
+	uid := regexp.MustCompile(`'uid':'(.+?)',`).FindStringSubmatch(res.String())[1]
+	rep := regexp.MustCompile(`'rep':'(.+?)',`).FindStringSubmatch(res.String())[1]
+	up := regexp.MustCompile(`'up':(.+?),`).FindStringSubmatch(res.String())[1]
+	ls := regexp.MustCompile(`'ls':(.+?),`).FindStringSubmatch(res.String())[1]
+	tName := regexp.MustCompile(`'t':(.+?),`).FindStringSubmatch(res.String())[1]
+	kName := regexp.MustCompile(`'k':(.+?),`).FindStringSubmatch(res.String())[1]
+	t := regexp.MustCompile(`var ` + tName + ` = '(.+?)';`).FindStringSubmatch(res.String())[1]
+	k := regexp.MustCompile(`var ` + kName + ` = '(.+?)';`).FindStringSubmatch(res.String())[1]
+	pg := 1
+	for {
+		var resp LanZouFilesResp
+		res, err = lanzouClient.R().SetResult(&resp).SetFormData(map[string]string{
+			"lx":  lx,
+			"fid": fid,
+			"uid": uid,
+			"pg":  strconv.Itoa(pg),
+			"rep": rep,
+			"t":   t,
+			"k":   k,
+			"up":  up,
+			"ls":  ls,
+			"pwd": account.Password,
+		}).Post("https://wwa.lanzouo.com/filemoreajax.php")
+		if err != nil {
+			log.Debug(err)
+			break
+		}
+		log.Debug(res.String())
+		if resp.Zt != 1 {
+			return nil, fmt.Errorf("%v", resp.Info)
+		}
+		if len(resp.Text) == 0 {
+			break
+		}
+		pg++
+		files = append(files, resp.Text...)
+	}
+	return files,nil
+}
+
+//type LanzouDownInfo struct {
+//	FId    string `json:"f_id"`
+//	IsNewd string `json:"is_newd"`
+//}
+
+// 获取下载页面的ID
+func (driver *Lanzou) GetDownPageId(fileId string, account *model.Account) (string, error) {
+	var resp LanZouFilesResp
+	res, err := lanzouClient.R().SetResult(&resp).SetHeader("Cookie", account.AccessToken).
+		SetFormData(map[string]string{
+			"task":    "22",
+			"file_id": fileId,
+		}).Post("https://pc.woozooo.com/doupload.php")
+	if err != nil {
+		return "", err
+	}
+	log.Debug(res.String())
+	if resp.Zt != 1 {
+		return "", fmt.Errorf("%v", resp.Info)
+	}
+	info, ok := resp.Info.(map[string]interface{})
+	if !ok {
+		return "", fmt.Errorf("%v", resp.Info)
+	}
+	fid, ok := info["f_id"].(string)
+	if !ok {
+		return "", fmt.Errorf("%v", info["f_id"])
+	}
+	return fid, nil
+}
+
+type LanzouLinkResp struct {
+	Dom string `json:"dom"`
+	Url string `json:"url"`
+	Zt  int    `json:"zt"`
+}
+
+func (driver *Lanzou) GetLink(downId string) (string, error) {
+	res, err := lanzouClient.R().Get("https://wwa.lanzouo.com/" + downId)
+	if err != nil {
+		return "", err
+	}
+	iframe := regexp.MustCompile(`<iframe class="ifr2" name=".{2,20}" src="(.+?)"`).FindStringSubmatch(res.String())
+	if len(iframe) == 0 {
+		return "", fmt.Errorf("get down empty page")
+	}
+	iframeUrl := "https://wwa.lanzouo.com" + iframe[1]
+	res, err = lanzouClient.R().Get(iframeUrl)
+	if err != nil {
+		return "", err
+	}
+	ajaxdata := regexp.MustCompile(`var ajaxdata = '(.+?)'`).FindStringSubmatch(res.String())
+	if len(ajaxdata) == 0 {
+		return "", fmt.Errorf("get iframe empty page")
+	}
+	signs := ajaxdata[1]
+	sign := regexp.MustCompile(`var ispostdowns = '(.+?)';`).FindStringSubmatch(res.String())[1]
+	websignkey := regexp.MustCompile(`'websignkey':'(.+?)'`).FindStringSubmatch(res.String())[1]
+	var resp LanzouLinkResp
+	form := map[string]string{
+		"action":     "downprocess",
+		"signs":      signs,
+		"sign":       sign,
+		"ves":        "1",
+		"websign":    "",
+		"websignkey": websignkey,
+	}
+	log.Debugf("form: %+v", form)
+	_, err = lanzouClient.R().SetResult(&resp).
+		SetHeader("origin", "https://wwa.lanzouo.com").
+		SetHeader("referer", iframeUrl).
+		SetFormData(form).Post("https://wwa.lanzouo.com/ajaxm.php")
+	if resp.Zt == 1 {
+		return resp.Dom + "/file/" + resp.Url, nil
+	}
+	return "", fmt.Errorf("can't get link")
+}
+
+func init() {
+	RegisterDriver(&Lanzou{})
+	lanzouClient.
+		SetRetryCount(3).
+		SetHeader("user-agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/87.0.4280.88 Safari/537.36")
+}
diff --git a/drivers/lanzou_driver.go b/drivers/lanzou_driver.go
new file mode 100644
index 00000000..1a7d9853
--- /dev/null
+++ b/drivers/lanzou_driver.go
@@ -0,0 +1,163 @@
+package drivers
+
+import (
+	"fmt"
+	"github.com/Xhofe/alist/conf"
+	"github.com/Xhofe/alist/model"
+	"github.com/Xhofe/alist/utils"
+	"github.com/gin-gonic/gin"
+	log "github.com/sirupsen/logrus"
+	"path/filepath"
+)
+
+type Lanzou struct{}
+
+func (driver Lanzou) Config() DriverConfig {
+	return DriverConfig{
+		Name:      "Lanzou",
+		OnlyProxy: false,
+	}
+}
+
+func (driver Lanzou) Items() []Item {
+	return []Item{
+		{
+			Name:     "onedrive_type",
+			Label:    "lanzou type",
+			Type:     SELECT,
+			Required: true,
+			Values:   "cookie,url",
+		},
+		{
+			Name:        "access_token",
+			Label:       "cookie",
+			Type:        STRING,
+			Description: "about 15 days valid",
+		},
+		{
+			Name:  "root_folder",
+			Label: "root folder file_id",
+			Type:  STRING,
+		},
+		{
+			Name:  "site_url",
+			Label: "share url",
+			Type:  STRING,
+		},
+		{
+			Name:  "password",
+			Label: "share password",
+			Type:  STRING,
+		},
+	}
+}
+
+func (driver Lanzou) Save(account *model.Account, old *model.Account) error {
+	if account.OnedriveType == "cookie" {
+		if account.RootFolder == "" {
+			account.RootFolder = "-1"
+		}
+	}
+	account.Status = "work"
+	_ = model.SaveAccount(account)
+	return nil
+}
+
+func (driver Lanzou) File(path string, account *model.Account) (*model.File, error) {
+	path = utils.ParsePath(path)
+	if path == "/" {
+		return &model.File{
+			Id:        account.RootFolder,
+			Name:      account.Name,
+			Size:      0,
+			Type:      conf.FOLDER,
+			Driver:    driver.Config().Name,
+			UpdatedAt: account.UpdatedAt,
+		}, nil
+	}
+	dir, name := filepath.Split(path)
+	files, err := driver.Files(dir, account)
+	if err != nil {
+		return nil, err
+	}
+	for _, file := range files {
+		if file.Name == name {
+			return &file, nil
+		}
+	}
+	return nil, PathNotFound
+}
+
+func (driver Lanzou) Files(path string, account *model.Account) ([]model.File, error) {
+	path = utils.ParsePath(path)
+	var rawFiles []LanZouFile
+	cache, err := conf.Cache.Get(conf.Ctx, fmt.Sprintf("%s%s", account.Name, path))
+	if err == nil {
+		rawFiles, _ = cache.([]LanZouFile)
+	} else {
+		file, err := driver.File(path, account)
+		if err != nil {
+			return nil, err
+		}
+		rawFiles, err = driver.GetFiles(file.Id, account)
+		if err != nil {
+			return nil, err
+		}
+		if len(rawFiles) > 0 {
+			_ = conf.Cache.Set(conf.Ctx, fmt.Sprintf("%s%s", account.Name, path), rawFiles, nil)
+		}
+	}
+	files := make([]model.File, 0)
+	for _, file := range rawFiles {
+		files = append(files, *driver.FormatFile(&file))
+	}
+	return files, nil
+}
+
+func (driver Lanzou) Link(path string, account *model.Account) (string, error) {
+	file, err := driver.File(path, account)
+	if err != nil {
+		return "", err
+	}
+	log.Debugf("down file: %+v", file)
+	downId := file.Id
+	if account.OnedriveType == "cookie" {
+		downId, err = driver.GetDownPageId(file.Id, account)
+		if err != nil {
+			return "", err
+		}
+	}
+	link, err := driver.GetLink(downId)
+	if err != nil {
+		return "", err
+	}
+	return link, nil
+}
+
+func (driver Lanzou) Path(path string, account *model.Account) (*model.File, []model.File, error) {
+	path = utils.ParsePath(path)
+	log.Debugf("lanzou path: %s", path)
+	file, err := driver.File(path, account)
+	if err != nil {
+		return nil, nil, err
+	}
+	if file.Type != conf.FOLDER {
+		file.Url, _ = driver.Link(path, account)
+		return file, nil, nil
+	}
+	files, err := driver.Files(path, account)
+	if err != nil {
+		return nil, nil, err
+	}
+	return nil, files, nil
+}
+
+func (driver Lanzou) Proxy(c *gin.Context, account *model.Account) {
+	c.Request.Header.Del("Origin")
+}
+
+func (driver Lanzou) Preview(path string, account *model.Account) (interface{}, error) {
+	return nil, NotSupport
+}
+
+var _ Driver = (*Lanzou)(nil)
diff --git a/model/file.go b/model/file.go
index 563c89c5..55a5ab88 100644
--- a/model/file.go
+++ b/model/file.go
@@ -16,6 +16,8 @@ type File struct {
 	UpdatedAt *time.Time `json:"updated_at"`
 	Thumbnail string     `json:"thumbnail"`
 	Url       string     `json:"url"`
+	SizeStr   string     `json:"size_str"`
+	TimeStr   string     `json:"time_str"`
 }
 
 func SortFiles(files []File, account *Account) {
@@ -63,4 +65,4 @@ func (f File) ModTime() time.Time {
 
 func (f File) IsDir() bool {
 	return f.Type == conf.FOLDER
-}
\ No newline at end of file
+}