PicGo用golang写自定义图床上传服务

本文阅读量: 2021-01-16

一背景

如果你有写博客习惯，并且是自己搭建的博客，会面临一个问题，你的图片放哪。其实有很多选择，比如：七牛图床、Github图床等都可以免费使用。但是多少有些问题，比如七牛图床如果你想使用https，很抱歉是需要收费的，如果流量不高，费用其实也很低；再比如Github图床，国内访问不稳定。对于我自己的博客，经过几番尝试之后还是想放在自己的服务器。

有写博客习惯的人，基本都用过一些markdown工具（除非你的博客不是用markdown来编写的），比如我，如果涉及到有图片的markdown编辑，我会选择Typora，因为可以配合PicGo完成图片自动上传转换成markdown图片链接，而PicGo还支持自定义图床，即可以支持把自己的图片上传的自己的服务器并返回事先定义好的图片地址，非常方便。

二 PicGo自定义图床的搭建

先看自定义图床需要哪些配置

从配置可以看到，上传图片的Web图床设置，主要是两个必填。

API地址：即上传图片的地址
POST参数名：上传图片post的参数名

非必填的有三个，我们通常会用到以下两个

JSON路径：是返回body中图片链接url的json路径
自定义请求头：主要自定义安全控制、上传途径等请求头，方便随时调整上传的途径。

这次简单的一个上传服务，我们可以直接用golang语言写一个简单web上传即可，先简单定义一下我们的Web图床的一些参数：

API地址：http://[hostname]:8081
POST参数名：file
JSON路径：data.url
自定义请求头：{“Authorization”: “Basic 【basic验证的账号密码base64】”, “uri”: “【uri路径如：mafgwo/picgo/】”, “imgHost”: “【图片根http路径如：https://img.qicoder.com/】"}

实现代码如下：

package main

import (
    "io"
    "log"
    "net/http"
    "os"
)

var (
    // 文件 key
    uploadFileKey = "file"
    // 上传的图片保存根路径
    filePath = "/usr/share/nginx/html/imgs/"
)


func main() {
    http.HandleFunc("/upload", uploadHandler)
    if err := http.ListenAndServe(":8081", nil); err != nil {
        log.Fatalf("error to start http server:%s", err.Error())
    }
}

func uploadHandler(w http.ResponseWriter, r *http.Request) {
    // 接受文件
    file, header, err := r.FormFile(uploadFileKey)
    if err != nil {
        // ignore the error handler
    }
    log.Printf("selected file name is %s", header.Filename)
    authValue := r.Header.Get("Authorization")
    if (authValue != "Basic 【basic验证的账号密码base64】") {
        log.Printf("auth fail")
        return
    }
    path := filePath + r.Header.Get("Uri")
    createFile(path)
    imgPath := path + header.Filename
    log.Printf("path is %s", path)

    // 将文件拷贝到指定路径下，或者其他文件操作
    dst, err := os.Create(imgPath)
    if err != nil {
        log.Fatalf("create file error :%s", err.Error())
        // ignore
    }
    _, err = io.Copy(dst, file)
    if err != nil {
        log.Fatalf("copy file error :%s", err.Error())
        // ignore
    }
    log.Printf("upload success")
    io.WriteString(w, "{\"code\": 0, \"data\": {\"url\": \"" + (r.Header.Get("ImgHost") + r.Header.Get("Uri") + header.Filename) + "\"}}")
}


//调用os.MkdirAll递归创建文件夹
func createFile(filePath string)  error  {
    if !isExist(filePath) {
        err := os.MkdirAll(filePath,os.ModePerm)
        return err
    }
    return nil
}

// 判断所给路径文件/文件夹是否存在(返回true是存在)
func isExist(path string) bool {
    _, err := os.Stat(path)    //os.Stat获取文件信息
    if err != nil {
        if os.IsExist(err) {
            return true
        }
        return false
    }
    return true
}

图片访问是通过nginx访问，nginx的配置如下：

server {
  listen                     80;
  listen                     443 ssl http2;
  server_name                img.qicoder.com imgs.qicoder.com;

  client_max_body_size 1000m;
  ssl_certificate            /etc/nginx/conf.d/ssl/qicoder.com.crt;
  ssl_certificate_key        /etc/nginx/conf.d/ssl/qicoder.com.key;
  ssl_buffer_size            16k;
  ssl_protocols              TLSv1 TLSv1.1 TLSv1.2;
  # ssl_ecdh_curve             X25519:P-256:P-384:P-224:P-521;
  ssl_ciphers                TLS13-AES-256-GCM-SHA384:TLS13-CHACHA20-POLY1305-SHA256:TLS13-AES-128-GCM-SHA256:TLS13-AES-128-CCM-8-SHA256:TLS13-AES-128-CCM-SHA256:ECDHE-ECDSA-AES256-GCM-SHA384:ECDHE-RSA-AES256-GCM-SHA384:ECDHE-ECDSA-CHACHA20-POLY1305:ECDHE-RSA-CHACHA20-POLY1305:ECDHE-ECDSA-AES128-GCM-SHA256:ECDHE-RSA-AES128-GCM-SHA256:ECDHE-ECDSA-AES256-SHA384:ECDHE-RSA-AES256-SHA384:ECDHE-ECDSA-AES128-SHA256:ECDHE-RSA-AES128-SHA256;
  ssl_prefer_server_ciphers  on;
  ssl_session_timeout        3h;
  ssl_stapling               on;
  ssl_session_tickets        on;

  access_log                 /var/log/nginx/access-img.log main;

  add_header                 Strict-Transport-Security "max-age=31536000; preload; includeSubDomains" always;
  add_header                 X-Frame-Options SAMEORIGIN;

  if ($scheme != 'https') {
    rewrite ^(.*)$  https://$server_name$1 permanent;
  } 
 
  location /upload {
    proxy_pass http://localhost:8081;
  }

  location / {
    alias   /usr/share/nginx/html/imgs/;
  }
}

nginx中使用的https，可以通过acme.sh免费生成ssl。

到此为止，我们的的服务就算是搭建好了。

一 背景

二 PicGo自定义图床的搭建

一背景