mirror of https://github.com/IoTcat/auxpi.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
101 lines
2.7 KiB
101 lines
2.7 KiB
// Copyright (c) 2019 aimerforreimu. All Rights Reserved. |
|
// Use of this source code is governed by a BSD-style |
|
// license that can be found in the LICENSE file. |
|
// |
|
// GNU GENERAL PUBLIC LICENSE |
|
// Version 3, 29 June 2007 |
|
// |
|
// Copyright (C) 2007 Free Software Foundation, Inc. <https://fsf.org/> |
|
// Everyone is permitted to copy and distribute verbatim copies |
|
// of this license document, but changing it is not allowed. |
|
// |
|
// repo: https://github.com/aimerforreimu/auxpi |
|
|
|
package server |
|
|
|
import ( |
|
"encoding/json" |
|
"io/ioutil" |
|
"net/http" |
|
"regexp" |
|
|
|
"github.com/astaxie/beego" |
|
"github.com/auxpi/auxpiAll" |
|
"github.com/auxpi/tools" |
|
) |
|
|
|
type Prnt struct { |
|
FileLimit []string |
|
MaxSize int |
|
} |
|
|
|
func (s *Prnt) Upload(image *ImageParam) (ImageReturn, error) { |
|
url := "https://prntscr.com/upload.php" |
|
|
|
|
|
file := &auxpi.FormFile{ |
|
Name: image.Name, |
|
Key: "image", |
|
Value: *image.Content, |
|
Type: image.Type, |
|
} |
|
var header = make(map[string]string) |
|
header["User-Agent"] = "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.13; rv:65.0) Gecko/20100101 Firefox/65.0" |
|
data := tools.FormPost(file, url, header) |
|
var d = map[string]string{} |
|
json.Unmarshal([]byte(data), &d) |
|
|
|
req, err := http.NewRequest("GET", d["data"], nil) |
|
if err != nil { |
|
return ImageReturn{}, err |
|
} |
|
|
|
req.Header.Add("User-Agent", "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.13; rv:65.0) Gecko/20100101 Firefox/65.0") |
|
|
|
res, err := http.DefaultClient.Do(req) |
|
if err != nil { |
|
return ImageReturn{}, err |
|
} |
|
defer res.Body.Close() |
|
|
|
body, err := ioutil.ReadAll(res.Body) |
|
if err != nil { |
|
return ImageReturn{}, err |
|
} |
|
|
|
var re = regexp.MustCompile(`(?m)<meta property="og:image" content="(.*?)"/>`) |
|
url = re.FindAllStringSubmatch(string(body), -1)[0][1] |
|
beego.Alert(url) |
|
return ImageReturn{ |
|
Url: url, |
|
ID: 7, |
|
}, nil |
|
} |
|
|
|
func (s *Prnt) UploadToPrnt(img []byte, imgInfo string, imgType string) string { |
|
url := "https://prntscr.com/upload.php" |
|
name := tools.GetFileNameByMimeType(imgInfo) |
|
|
|
file := &auxpi.FormFile{ |
|
Name: name, |
|
Key: "image", |
|
Value: img, |
|
Type: imgType, |
|
} |
|
var header = make(map[string]string) |
|
header["User-Agent"] = "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.13; rv:65.0) Gecko/20100101 Firefox/65.0" |
|
data := tools.FormPost(file, url, header) |
|
var d = map[string]string{} |
|
json.Unmarshal([]byte(data), &d) |
|
|
|
req, _ := http.NewRequest("GET", d["data"], nil) |
|
req.Header.Add("User-Agent", "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.13; rv:65.0) Gecko/20100101 Firefox/65.0") |
|
res, _ := http.DefaultClient.Do(req) |
|
defer res.Body.Close() |
|
body, _ := ioutil.ReadAll(res.Body) |
|
|
|
var re = regexp.MustCompile(`(?m)<meta property="og:image" content="(.*?)"/>`) |
|
url = re.FindAllStringSubmatch(string(body), -1)[0][1] |
|
beego.Alert(url) |
|
return url |
|
}
|
|
|