md/src/api/file.js

import fetch from "./fetch";
import CryptoJS from "crypto-js";
import OSS from "ali-oss";
import COS from "cos-js-sdk-v5";
import Buffer from "buffer-from";
import { v4 as uuidv4 } from "uuid";
import * as qiniu from "qiniu-js";
import { utf16to8, base64encode, safe64 } from "../assets/scripts/tokenTools";

function fileUpload(content, file) {
    const imgHost = localStorage.getItem("imgHost");
    !imgHost && localStorage.setItem("imgHost", "default");
    switch (imgHost) {
        case "aliOSS":
            return aliOSSFileUpload(content, file.name);
        case "txCOS":
            return txCOSFileUpload(file);
        case "qiniu":
            return qiniuUpload(file);
        case "gitee":
            return giteeUpload(content, file.name);
        case "github":
            return ghFileUpload(content, file.name);
        default:
            // return file.size / 1024 < 1024
            //     ? giteeUpload(content, file.name)
            //     : ghFileUpload(content, file.name);
            return ghFileUpload(content, file.name);
    }
}

function getDir() {
    const date = new Date();
    const year = date.getFullYear();
    const month = (date.getMonth() + 1).toString().padStart(2, "0");
    const day = date.getDate().toString().padStart(2, "0");
    return `${year}/${month}/${day}`;
}

function getDateFilename(filename) {
    const currentTimestamp = new Date().getTime();
    const fileSuffix = filename.split(".")[1];
    return `${currentTimestamp}-${uuidv4()}.${fileSuffix}`;
}

//-----------------------------------------------------------------------
// GitHub File Upload
//-----------------------------------------------------------------------

function getGitHubCommonConfig(username, repo, branch, token) {
    const dir = getDir();
    return {
        method: "put",
        headers: {
            Authorization: "token " + token,
        },
        username: username,
        repo: repo,
        branch: branch,
        url: `https://api.github.com/repos/${username}/${repo}/contents/${dir}/`,
    };
}

function getGitHubDefaultConfig() {
    const defaultConfig = {
        username: "filess",
        repo: `img${Math.floor(Math.random() * 10)}`,
        branch: "main",
        accessToken: [
            "7715d7ca67b5d3837cfdoocsmde8c38421815aa423510af",
            "c411415bf95dbe39625doocsmd5047ba9b7a2a6c9642abe",
            "2821cd8819fa345c053doocsmdca86ac653f8bc20db1f1b",
            "445f0dae46ef1f2a4d6doocsmdc797301e94797b4750a4c",
            "cc1d0c1426d0fd0902bdoocsmdd2d7184b14da61b86ec46",
            "b67e9d15cb6f910492fdoocsmdac6b44d379c953bb19eff",
            "618c4dc2244ccbbc088doocsmd125d17fd31b7d06a50cf3",
            "a4b581732e1c1507458doocsmdc5b223b27dae5e2e16a55",
        ],
    };

    const randIndex = Math.floor(
        Math.random() * defaultConfig.accessToken.length
    );
    const token = defaultConfig.accessToken[randIndex].replace("doocsmd", "");
    return getGitHubCommonConfig(
        defaultConfig.username,
        defaultConfig.repo,
        defaultConfig.branch,
        token
    );
}

function getGitHubConfig() {
    const githubConfig = JSON.parse(localStorage.getItem("githubConfig"));
    const repoUrl = githubConfig.repo
        .replace("https://github.com/", "")
        .replace("http://github.com/", "")
        .replace("github.com/", "")
        .split("/");
    const username = repoUrl[0];
    const repo = repoUrl[1];
    return getGitHubCommonConfig(
        username,
        repo,
        githubConfig.branch,
        githubConfig.accessToken
    );
}

async function ghFileUpload(content, filename) {
    const isDefault = localStorage.getItem("imgHost") !== "github";
    const config = isDefault ? getGitHubDefaultConfig() : getGitHubConfig();
    const dateFilename = getDateFilename(filename);

    const branch = config.branch || "master";
    const res = await fetch({
        url: config.url + dateFilename,
        method: config.method,
        headers: config.headers,
        data: {
            branch: branch,
            message: `Upload by ${window.location.href}`,
            content: content,
        },
    });

    const githubResourceUrl = `raw.githubusercontent.com/${config.username}/${config.repo}/${branch}/`;
    const cdnResourceUrl = `cdn.jsdelivr.net/gh/${config.username}/${config.repo}@${branch}/`;
    return isDefault
        ? res.content.download_url.replace(githubResourceUrl, cdnResourceUrl)
        : res.content.download_url;
}

//-----------------------------------------------------------------------
// Gitee File Upload
//-----------------------------------------------------------------------

function getGiteeDefaultConfig() {
    const defaultConfig = {
        username: "filesss",
        repo: `img${Math.floor(Math.random() * 10)}`,
        branch: "main",
        accessToken: [
            "ed5fc9866bd6c2fdoocsmddd433f806fd2f399c",
            "5448ffebbbf1151doocsmdc4e337cf814fc8a62",
            "25b05efd2557ca2doocsmd75b5c0835e3395911",
            "11628c7a5aef015doocsmd2eeff9fb9566f0458",
            "cb2f5145ed938dedoocsmdbd063b4ed244eecf8",
            "d8c0b57500672c1doocsmd55f48b866b5ebcd98",
            "78c56eadb88e453doocsmd43ddd95753351771a",
            "03e1a688003948fdoocsmda16fcf41e6f03f1f0",
        ],
    };

    const randIndex = Math.floor(
        Math.random() * defaultConfig.accessToken.length
    );
    const token = defaultConfig.accessToken[randIndex].replace("doocsmd", "");
    return {
        repo: `gitee.com/${defaultConfig.username}/${defaultConfig.repo}`,
        branch: defaultConfig.branch,
        accessToken: token,
    };
}

async function giteeUpload(content, filename) {
    const isDefault = localStorage.getItem("imgHost") == "default";
    const giteeConfig = isDefault
        ? getGiteeDefaultConfig()
        : JSON.parse(localStorage.getItem("giteeConfig"));
    const repoUrl = giteeConfig.repo
        .replace("https://gitee.com/", "")
        .replace("http://gitee.com/", "")
        .replace("gitee.com/", "")
        .split("/");
    const username = repoUrl[0];
    const repo = repoUrl[1];
    const dir = getDir();
    const dateFilename = getDateFilename(filename);
    const res = await fetch({
        url: `https://gitee.com/api/v5/repos/${username}/${repo}/contents/${dir}/${dateFilename}`,
        method: "POST",
        data: {
            access_token: giteeConfig.accessToken,
            branch: giteeConfig.branch || "master",
            content: content,
            message: `Upload by ${window.location.href}`,
        },
    });
    return encodeURI(res.content.download_url);
}

//-----------------------------------------------------------------------
// Qiniu File Upload
//-----------------------------------------------------------------------

function getQiniuToken(accessKey, secretKey, putPolicy) {
    const policy = JSON.stringify(putPolicy);
    const encoded = base64encode(utf16to8(policy));
    const hash = CryptoJS.HmacSHA1(encoded, secretKey);
    const encodedSigned = hash.toString(CryptoJS.enc.Base64);
    return `${accessKey}:${safe64(encodedSigned)}:${encoded}`;
}

async function qiniuUpload(file) {
    const qiniuConfig = JSON.parse(localStorage.getItem("qiniuConfig"));
    const putPolicy = {
        scope: qiniuConfig.bucket,
        deadline: Math.trunc(new Date().getTime() / 1000) + 3600,
    };
    const token = getQiniuToken(
        qiniuConfig.accessKey,
        qiniuConfig.secretKey,
        putPolicy
    );
    const dir = qiniuConfig.path ? qiniuConfig.path + "/" : "";
    const dateFilename = dir + getDateFilename(file.name);
    const config = {
        region: qiniuConfig.region,
    };
    const observable = qiniu.upload(file, dateFilename, token, {}, config);
    return new Promise((resolve, reject) => {
        observable.subscribe({
            next: (result) => {
                console.log(result);
            },
            error: (err) => {
                reject(err.message);
            },
            complete: (result) => {
                resolve(`${qiniuConfig.domain}/${result.key}`);
            },
        });
    });
}

//-----------------------------------------------------------------------
// AliOSS File Upload
//-----------------------------------------------------------------------

async function aliOSSFileUpload(content, filename) {
    const dateFilename = getDateFilename(filename);
    const aliOSSConfig = JSON.parse(localStorage.getItem("aliOSSConfig"));
    const buffer = Buffer(content, "base64");
    try {
        const dir = `${aliOSSConfig.path}/${dateFilename}`;
        const client = new OSS({
            region: aliOSSConfig.region,
            bucket: aliOSSConfig.bucket,
            accessKeyId: aliOSSConfig.accessKeyId,
            accessKeySecret: aliOSSConfig.accessKeySecret,
        });
        const res = await client.put(dir, buffer);
        return aliOSSConfig.cdnHost == ""
            ? res.url
            : `${aliOSSConfig.cdnHost}/${
                  aliOSSConfig.path == "" ? dateFilename : dir
              }`;
    } catch (e) {
        return Promise.reject(e);
    }
}

//-----------------------------------------------------------------------
// TxCOS File Upload
//-----------------------------------------------------------------------

async function txCOSFileUpload(file) {
    const dateFilename = getDateFilename(file.name);
    const txCOSConfig = JSON.parse(localStorage.getItem("txCOSConfig"));
    const cos = new COS({
        SecretId: txCOSConfig.secretId,
        SecretKey: txCOSConfig.secretKey,
    });
    return new Promise((resolve, reject) => {
        cos.putObject(
            {
                Bucket: txCOSConfig.bucket,
                Region: txCOSConfig.region,
                Key: `${txCOSConfig.path}/${dateFilename}`,
                Body: file,
            },
            function (err, data) {
                if (err) {
                    reject(err);
                } else if (txCOSConfig.cdnHost) {
                    resolve(
                        txCOSConfig.path != ""
                            ? `${txCOSConfig.cdnHost}/${txCOSConfig.path}/${dateFilename}`
                            : `${txCOSConfig.cdnHost}/${dateFilename}`
                    );
                } else {
                    resolve(`https://${data.Location}`);
                }
            }
        );
    });
}

export default {
    fileUpload,
};