app.go

package main

import (
	"context"
	"encoding/json"
	"fmt"
	"github.com/wailsapp/wails/v2/pkg/runtime"
	"io"
	"net/http"
	"os"
	"path/filepath"
	"strings"
	"wd-reader/go/book"
	"wd-reader/go/book/epub/EpubBook"
	"wd-reader/go/constant"
	"wd-reader/go/log"
	"wd-reader/go/server"
)

// App struct
type App struct {
	ctx context.Context
}

// NewApp creates a new App application struct
func NewApp() *App {
	return &App{}
}

// startup is called when the app starts. The context is saved
// so we can call the runtime methods
func (a *App) startup(ctx context.Context) {
	a.ctx = ctx
}

// Greet returns a greeting for the given name
func (a *App) Greet(name string) string {
	return fmt.Sprintf("Hello %s, It's show time!", name)
}

// OpenFileDialog Deprecated 目前这个好像有BUG用不起来 就很烦
func (a *App) OpenFileDialog() string {
	// ".mobi", ".azw3", ".azw", ".docx", ".doc", ".pdf", ".html", ".htmlz"
	fmt.Println("开始调用")
	dialog, err := runtime.OpenFileDialog(a.ctx, runtime.OpenDialogOptions{
		Title:            "文件选择",
		DefaultDirectory: "./",
		DefaultFilename:  "",
		Filters: []runtime.FileFilter{
			{
				DisplayName: "*.txt;*.epub;*.mobi;*.azw3;*.azw;*.docx;*.dox;*.pdf;*.html;*.htmlz",
				Pattern:     "*.txt;*.epub;*.mobi;*.azw3;*.azw;*.docx;*.dox;*.pdf;*.html;*.htmlz",
			},
		},
		ShowHiddenFiles:            false,
		CanCreateDirectories:       false,
		ResolvesAliases:            false,
		TreatPackagesAsDirectories: false,
	})
	if err != nil {
		fmt.Println(fmt.Errorf("failed to open file dialog: %v", err))
	}
	fmt.Println("调用返回" + dialog)
	return dialog
}

// GetAppPath 获取目前程序运行目录
func (a *App) GetAppPath() string {
	return book.GetAppPath()
}

// GetBooksPath 获取books的目录
func (a *App) GetBooksPath() string {
	return filepath.Join(book.GetAppPath(), constant.BOOK_PATH)
}

// GetVersion 获取版本
func (a *App) GetVersion() string {
	return version
}
func (a *App) GetServerUrl() string {

	return `http://localhost` + HttpPort + "/"
}

// ParseEpubToTxt 解析epub文件
func (a *App) ParseEpubToTxt(filename string) string {
	book.CheckBooksPath()
	path := book.GetAppPath()
	bookPath := filepath.Join(path, constant.BOOK_PATH, filename)
	txtOutputPath := strings.Replace(bookPath, ".epub", ".txt", -1)
	epub, err := EpubBook.ParseEpub(bookPath, txtOutputPath)
	if err != nil {
		return book.WrapperException(err)
	}
	err = epub.WriteEpub()
	if err != nil {
		return book.WrapperException(err)
	}
	return "解析成功"
}

// DeleteEpubFile 删除文件
func (a *App) DeleteEpubFile(filename string) string {
	dir := book.GetAppPath()
	join := filepath.Join(dir, constant.BOOK_PATH, filename)
	err := os.Remove(join)
	if err != nil {
		return book.WrapperException(err)
	}
	return "ok"
}

// GetBookList 获取文件列表
func (a *App) GetBookList() string {
	defer func() {
		if err := recover(); err != nil {
			log.Logger.Fatal(err)
		}
	}()
	return book.GetBookListExtract()
}

// GetChapterListByFileName 传入文件名称获取文件章节或者卷列表
func (a *App) GetChapterListByFileName(_fileName string, splitType string, splitValue string) string {
	defer func() {
		if err := recover(); err != nil {
			log.Logger.Fatal(err)
		}
	}()
	return book.GetChapterListByFileNameExtract(_fileName, splitType, splitValue)
}

// GetChapterContentByChapterName 根据传入文件名和章节名称来获取这一章节的内容
func (a *App) GetChapterContentByChapterName(_fileName string, chapterName string, splitType string, splitValue string) string {
	defer func() {
		if err := recover(); err != nil {
			log.Logger.Fatal(err)
		}
	}()
	return book.GetChapterContentByChpaterNameExtract(_fileName, chapterName, splitType, splitValue)
}

// AddFile 添加文件
func (*App) AddFile(name []string) string {
	var str string
	str = book.TransferFileFromFileSys(name)
	return str
}

// DeleteFile 删除文件
func (*App) DeleteFile(name string) string {
	var str string
	str = book.DeleteFile(name)
	return str
}

// GetScOne 诗词获取
func (a *App) GetScOne() string {
	defer func() {
		if err := recover(); err != nil {
			log.Logger.Error(err)
		}
	}()
	resp, err := http.Get(constant.WD_SERVER + "/wd/getSc")
	if err != nil {
		log.Logger.Errorf("getscone error,%v", err)
		return ""
	}
	defer func(Body io.ReadCloser) {
		err := Body.Close()
		if err != nil {
			fmt.Println(err)
		}
	}(resp.Body)
	var data server.Data
	all, err := io.ReadAll(resp.Body)

	if err != nil {
		log.Logger.Errorf("getscone io error,%v", err)
		return ""
	}

	err = json.Unmarshal(all, &data)
	if err != nil {
		log.Logger.Errorf("getscone json error,%v", err)
		return ""
	}
	if data.Success == false {
		marshal, _ := json.Marshal(data)
		log.Logger.Error("line get error ", string(marshal))
		return ""
	}

	m := make(map[string]string)
	m["hitokoto"] = data.Data
	marshal, _ := json.Marshal(m)
	return string(marshal)
}