图床工具的实现-基础篇

在上一章，我们介绍了图床工具（上传组件）的使用，这一章，我们来解析图床工具的实现，手把手教你写一个图床工具（Node 小工具）。

在教学开始前，我们需要准备：

Node运行环境：图床上传工具使用 Node 开发

我们希望你掌握以下知识：

（以下知识也可以跳过，直接开始学习本教程；掌握是为了更好的理解）

我们的基本目标

对基本的文件格式进行限制；
可指定上传目录，完成本地目录与远程目录映射；
图片上传功能；

我们的进阶目标：

将上传组件解耦，做成可配置项；
将上传组件制作成图床工具，将配置与组件解耦，由外部实现调用；
提供 windows 与 mac 双平台的脚本，使非技术人员可以通过该脚本配置公司静态资源；（如 banner 图、广告图）

（以下代码实现保存在插件合集中的 demo 目录下）

基本目标

我们的第一个目标是可指定上传目录，完成本地目录与远程目录映射，我们需要完成这个目标的话，我们需要有一个本地目录，所以我们在根目录下新建一个本地目录，取名为 images，我们再预设置一些目录及图片文件，具体如下

图片目录

现在我们的目标是要将本地目录的图片文件上传至远程目录，我们假设本地目录地址为 local_url，远程目录为 remote_url, 那么本地目录文件就是 local_url/images/blogs/plugins/upload/1.png，我们需要将该目录上传到远程目录的 remote_url/images/blogs/plugins/upload/1.png。

所以我们第一步就是需要解析本地目录，并且将本地目录的图片文件都解析出来。

我们新建 src/upload/index.js，先定义支持的图片格式

const DEFAULT_ALLOW_FILE = ["png", "jpg"];

定义好了支持的图片格式后，我们开始对目录进行检索

// fs 模块，用于读取目录与文件
const fs = require('fs');
// path 模块，用于解析和拼装路径
const path = require('path');

const DEFAULT_ALLOW_FILE = ["png", "jpg"];

/**
 * 读取目录下的图片文件，收集在 images 中
 * @param {*} entry 入口
 * @param {*} images 图片集合
 */
function readDir(entry, images = []) {
  const dirInfo = fs.readdirSync(entry);
  for (let i = 0; i < dirInfo.length; i++) {
    const item = dirInfo[i];
    // 拼装出文件/文件夹的路径
    const location = path.join(entry, item);
    const isDir = fs.statSync(location).isDirectory();
    // 如果为文件夹则继续递归向下查询
    if (isDir) {
      readDir(location, images);
      // 判断是否为所允许的图片格式
    } else if (DEFAULT_ALLOW_FILE.some(allowScheme => location.endsWith(allowScheme))) {
      images.push(location);
    }
  }
  return images;
}

// 定义检索的入口文件夹（ images 文件夹）
const staticDirPath = path.join(__dirname, '../..', 'images');
const images = readDir(staticDirPath);
console.log(images);

我们使用 node src/upload/index.js 命令执行该脚本，最后检索出来的 images 是个数组，包含了本地 images 下所有的图片文件路径，也就是我们需要的 local_url/image_path 集合。

现在我们开始最后一步，开始上传，阿里云 oss 提供了一个库 —— ali-oss，我们使用 npm 进行安装，然后我们在头部导入

const OSS = require("ali-oss");

然后我们根据阿里云提供的 API，编写上传的代码

//...
// Convert Windows backslash paths to slash paths
const slash = require('slash');
const OSS = require("ali-oss");

//...
// 这些配置参数在上一章已经进行说明，不再复述
const client = new OSS({
  "region": "<your config>",
  "accessKeyId": "<your config>",
  "accessKeySecret": "<your config>",
  "bucket": "<your config>",
});

async function upload() {
  for (let i = 0; i < images.length; i++) {
    // slash 是为了兼容 windows 平台的路径划分符为 \ 的特性
    const local_url = slash(images[i]);
    // 阿里云 OSS 目标文件名
    // local_url.split() 是为了去除本地目录的前缀
    const remote_url = `images${local_url.split(staticDirPath)[1]}`;
    // 按顺序依次上传文件
    const result = await client.put(remote_url, local_url)
    console.log(`${result.url} 上传成功`);
  }
  console.log("所有文件上传成功");
}

upload();

我们使用 node src/upload/index.js 命令执行该脚本，下面可以看到执行结果（我的目录与教学的目录不同，所以执行的命令也不一致，可到该地址查看我的源码）