package plugin

import (
	"bytes"
	"encoding/json"
	"gopic/conf"
	"gopic/utils"
	"io/ioutil"
	"net/http"
)
//定义插件类
type GithubStorage struct {
	B *utils.BaseStorage
}
//定义插件名
const (
	githubPluginName = "github"
)
//声明插件配置
var githubConfig map[string]interface{}

//实现Upload方法
func (g *GithubStorage) Upload(im *utils.Image) (string, error) {
	responseURL := githubConfig["responseurl"].(string)
	requestURL := githubConfig["requesturl"].(string)
	token := githubConfig["token"].(string)
	return responseURL + im.OutSuffix, uploadPictureToGithub(requestURL, token, im.OutBase64, im.OutSuffix)
}

//实现插件实例构造函数
func NewGithubStorage() *GithubStorage {
	return &GithubStorage{utils.NewBaseStorage()}
}

//upload具体的实现，用来上传单个图片
func uploadPictureToGithub(requestURL, token, data, suffix string) error {
	url := requestURL + suffix
	body := make(map[string]interface{})
	body["branch"] = "main"
	body["message"] = "markdown upload picture"
	body["content"] = data
	jsonBody, _ := json.Marshal(body)
	req, _ := http.NewRequest("PUT", url, bytes.NewBuffer(jsonBody))
	req.Header.Set("Content-Type", "application/json")
	req.Header.Set("Authorization", "token "+token)
	client := &http.Client{}
	resp, err := client.Do(req)
	if err != nil {
		return err
	}
	defer resp.Body.Close()
	_, err = ioutil.ReadAll(resp.Body)
	return err
}
//初始化函数
func init() {
    //将帮助文档写入
	utils.StroageHelp[githubPluginName] = githubHelp()
    //读取插件配置
	githubConfig = conf.Viper.GetStringMap(githubPluginName)
    //判断插件是否在配置文件中存在
	if githubConfig == nil {
		return
	}
    //判断配置中的插件是否处于激活状态
	active := githubConfig["active"]
	if active == nil {
		return
	}
    //如果插件处于激活状态，将插件的实例传入utils.StroageMap
	if active == true {
		utils.StroageMap[githubPluginName] = NewGithubStorage()
	}
}
//帮助文档具体实现
func githubHelp() string {
	return "github plugin need this parameters:\nactive: false or true\nresponseURL: like https://gcore.jsdelivr.net/gh/yourUserName/pics@main/\nrequestURL: like https://api.github.com/repos/yourUserName/pics/contents/\ntoke: your github token"
}