mirror of
https://github.com/mindoc-org/mindoc.git
synced 2025-10-25 02:44:52 +08:00
实现导入Markdown
This commit is contained in:
45
utils/requests/requests.go
Normal file
45
utils/requests/requests.go
Normal file
@@ -0,0 +1,45 @@
|
||||
package requests
|
||||
|
||||
import (
|
||||
"io"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"os"
|
||||
"errors"
|
||||
"fmt"
|
||||
)
|
||||
|
||||
//下载远程文件并保存到指定位置
|
||||
func DownloadAndSaveFile(remoteUrl, dstFile string) (error) {
|
||||
client := &http.Client{}
|
||||
uri, err := url.Parse(remoteUrl)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
// Create the file
|
||||
out, err := os.Create(dstFile)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer out.Close()
|
||||
|
||||
request, err := http.NewRequest("GET", uri.String(), nil)
|
||||
request.Header.Add("Connection", "close")
|
||||
request.Header.Add("Host", uri.Host)
|
||||
request.Header.Add("Referer", uri.String())
|
||||
request.Header.Add("User-Agent", "Mozilla/5.0 (Windows NT 6.1; WOW64; rv:12.0) Gecko/20100101 Firefox/12.0")
|
||||
|
||||
resp, err := client.Do(request)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
|
||||
if resp.StatusCode == http.StatusOK {
|
||||
_, err = io.Copy(out, resp.Body)
|
||||
}else{
|
||||
return errors.New(fmt.Sprintf("bad status: %s", resp.Status))
|
||||
}
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user