XLXZ's Blog

#include <fstream>
#include <vector>
#include <string>
#include <iostream>

const std::string base64_chars =
             "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
             "abcdefghijklmnopqrstuvwxyz"
             "0123456789+/";

#include <chrono>

std::string base64_encode(const std::vector<unsigned char>& buffer) {
    std::string encoded;
    int i = 0;
    int j = 0;
    unsigned char char_array_3[3];
    unsigned char char_array_4[4];

    int totalBytes = buffer.size();
    int progressInterval = totalBytes / 10; // Update progress every 10% of the total bytes

    for (const auto& byte : buffer) {
        char_array_3[i++] = byte;
        if (i == 3) {
            char_array_4[0] = (char_array_3[0] & 0xfc) >> 2;
            char_array_4[1] = ((char_array_3[0] & 0x03) << 4) + ((char_array_3[1] & 0xf0) >> 4);
            char_array_4[2] = ((char_array_3[1] & 0x0f) << 2) + ((char_array_3[2] & 0xc0) >> 6);
            char_array_4[3] = char_array_3[2] & 0x3f;

            for (i = 0; (i < 4); i++)
                encoded += base64_chars[char_array_4[i]];
            i = 0;

            // Update progress
            if (encoded.size() % progressInterval == 0) {
                int progress = (encoded.size() * 100) / totalBytes;
                std::cout << "Progress: " << progress << "%" << std::endl;
            }
        }
    }

    if (i) {
        for (j = i; j < 3; j++)
            char_array_3[j] = '\0';

        char_array_4[0] = (char_array_3[0] & 0xfc) >> 2;
        char_array_4[1] = ((char_array_3[0] & 0x03) << 4) + ((char_array_3[1] & 0xf0) >> 4);
        char_array_4[2] = ((char_array_3[1] & 0x0f) << 2) + ((char_array_3[2] & 0xc0) >> 6);

        for (j = 0; (j < i + 1); j++)
            encoded += base64_chars[char_array_4[j]];

        while ((i++ < 3))
            encoded += '=';
    }

    return encoded;
}

void fileToBase64(const std::string& path) {
    std::ifstream file(path, std::ios::binary);
    std::vector<unsigned char> buffer(std::istreambuf_iterator<char>(file), {});

    std::string base64 = base64_encode(buffer);
    std::string dataUrl = "data:application/x-rar-compressed;base64," + base64;
    //输出到 “downloadLink_”+文件名.md文件

    //1.获取不带路径的文件名
	std::string::size_type iPos = path.find_last_of('\\') + 1;
	std::string fileName = path.substr(iPos, path.length() - iPos);
	std::cout << fileName << std::endl;
 
	//2.获取不带后缀的文件名
	std::string fileNameWithoutExtension = fileName.substr(0, fileName.rfind("."));
	std::cout << fileNameWithoutExtension << std::endl;
 
	//3.获取后缀名
	std::string suffix_str = fileName.substr(fileName.find_last_of('.') + 1);
	std::cout << suffix_str << std::endl;

    std::string outputFileName = "downloadLink_" + fileNameWithoutExtension + ".md";
    std::ofstream output(outputFileName);
    //输出到md文件
    //为md文件添加属性
    //---
    // title: downloadLink_下载文件名称
    // date: 2024-05-02 17:36
    // mathjax: false
    // tags:
    //   - Download
    // ---
    output << "---" << std::endl;
    output << "title: " << fileName << std::endl;
    output << "date: 2024-03-01 17:36" << std::endl;
    output << "mathjax: false" << std::endl;
    output << "tags:" << std::endl;
    output << "  - Download" << std::endl;
    output << "---" << std::endl;
    //输出下载链接,文件名为原文件名
    //以按钮的形式显示下载链接
    output << "<a href=\"" << dataUrl << "\" download=\""<< fileName <<"\">点击下载</a>";
    output.close();

    //再在文件夹“DownloadLink”里添加一个只有属性但是不包含下载链接的md文件
    std::string outputFileName2 = "downloadLink_" + fileNameWithoutExtension + ".md";
    std::ofstream output2("DownloadLink\\" + outputFileName2);
    //输出到md文件
    //为md文件添加属性
    //---
    // title: downloadLink_下载文件名称
    // date: 2024-05-02 17:36
    // mathjax: false
    // tags:
    //   - Download
    // ---
    output2 << "---" << std::endl;
    output2 << "title: " << fileName << std::endl;
    output2 << "mathjax: false" << std::endl;
    output2 << "date: 2024-03-01 17:36" << std::endl;
    output2 << "tags:" << std::endl;
    output2 << "  - Download" << std::endl;
    output2 << "---" << std::endl;
    output2.close();

    std::cout << "Output file: " << outputFileName << std::endl;
}

//接收文件位置参数
int main(int argc, char* argv[]) {
    //检查参数
    if (argc < 2) {
        std::cout << "Usage: " << argv[0] << " file1 [file2] [file3] ..." << std::endl;
        return 1;
    }
    //支持多个文件
    for (int i = 1; i < argc; i++){
        //显示进度条
        std::cout << "Processing " << argv[i] << "..." << std::endl;
        fileToBase64(argv[i]);
    }

    //结束后暂停
    std::cout << "Press any key to exit..." << std::endl;
    std::cin.get();
    return 0;
}