server.js

// _____     _____ _ _      ____  _   _      _   ____
// |_   _|_ _|  ___(_) | ___|___ \| | | |_ __| | / ___|  ___ _ ____   _____ _ __
//   | |/ _` | |_  | | |/ _ \ __) | | | | '__| | \___ \ / _ \ '__\ \ / / _ \ '__|
//   | | (_| |  _| | | |  __// __/| |_| | |  | |  ___) |  __/ |   \ V /  __/ |
//   |_|\__, |_|   |_|_|\___|_____|\___/|_|  |_| |____/ \___|_|    \_/ \___|_|
//      |___/
//
// Version 1.9 | By BLxcwg666 <https://github.com/BLxcwg666/TgFile2Url> | @xcnya / @xcnyacn
// Lastest Update at 2023/10/22 12:47
//「 人与人的相遇，不是恩赐就是劫。」

const express = require("express");
const axios = require("axios");
const dotenv = require("dotenv").config();
const sqlite3 = require("sqlite3").verbose();
const https = require("https");
const forceHttps = require('express-force-https');
const fs = require("fs");

const app = express();
const host = process.env.SERVER_HOST;
const port = process.env.SERVER_PORT;
const ssl = process.env.SERVER_SSL === 'true';
const db = new sqlite3.Database('sqlite.db');

app.use((req, res, next) => {
    res.header('Access-Control-Allow-Origin', '*');
    res.header('Access-Control-Allow-Headers', 'Origin, X-Requested-With, Content-Type, Accept');
    res.header('Server', 'NekoServer');
    res.header('X-Powered-By', 'NekoCore');
    next();
  });

app.get('/:UserID/:ID/:FileName', (req, res) => {
    const { UserID, ID, FileName } = req.params;

    db.get("SELECT OriginLink FROM files WHERE ID = ? AND UserID = ? AND FileName = ?", [ID, UserID, FileName], (err, row) => {
        if (err) {
            console.error(err);
            res.status(500).json({ code:"500", msg:"服务器坏掉惹呜呜呜~ 请检查控制台输出喵~"});
            return;
        }

        if (row) {
            const OriginLink = row.OriginLink;
            axios.get(OriginLink, { responseType: 'stream' })
                .then(response => {
                    res.setHeader('Content-Disposition', `attachment; filename="${FileName}"`);
                    response.data.pipe(res);
                })
                .catch(error => {
                    console.error(error);
                    res.status(500).json({ code:"500", msg:"代理源链接时出现错误呜呜呜~ 请检查控制台输出喵~"});
                });
        } else {
            res.status(404).json({ code:"404", msg:"文件不存在喵~"});
        }
    });
});

app.use((req, res) => {
    res.status(404).json({ code:"404", msg:"你在找什么喵？"});
});

if (ssl) {
    app.use(forceHttps);
    const server = https.createServer(
      {
        cert: fs.readFileSync(process.env.CERT_PATH),
        key: fs.readFileSync(process.env.CERT_KEY_PATH)
      },
      app
    );
  
    server.listen(port, host, () => {
      console.log(`Server started on ${host} port ${port} with SSL`);
    });
  } else {
    app.listen(port, host, () => {
      console.log(`Server started on ${host} port ${port}`);
    });
  }