proxyServer.ts

import * as http from 'http';
import * as request from 'request';
import * as express from 'express';
import * as Agent from 'socks5-http-client/lib/Agent';
import * as fs from 'fs';
import { parseAL, AL } from 'aigis-fuel';
import * as path from 'path';
import * as log from 'electron-log';

log.transports.file.level = 'info';
let mainFontPath = '';

function parse(buffer) {
  const result = parseAL(buffer);
  return result;
}

export class ProxyServer {
  FileList = {};
  ProxyHost = '127.0.0.1';
  ProxyPort = 1080;
  ProxyEnable = false;
  ProxyIsSocks5 = false;
  Port = 0;
  createServer(userDataPath: string) {
    const app = express();
    app.use(function(req, res, next) {
      res.setHeader('Access-Control-Allow-Origin', '*');
      res.setHeader('Cache-Control', 'immutable');
      next();
    });
    app.use((req, res) => {
      const headers = req.headers;
      headers.host = 'assets.millennium-war.net';
      // 设置代理
      const options: any = {
        url: 'http://assets.millennium-war.net' + req.path,
        headers: headers,
        encoding: null,
      };
      if (this.ProxyEnable === true && this.ProxyIsSocks5 === true) {
        options.agentClass = Agent;
        options.agentOptions = {
          socksHost: this.ProxyHost,
          socksPort: this.ProxyPort,
        };
      }
      if (this.ProxyEnable === true && this.ProxyIsSocks5 === false) {
        options.proxy = `http://${this.ProxyHost}:${this.ProxyPort}`;
      }
      let requestFileName = this.FileList[req.path];
      if (req.path.indexOf(mainFontPath) !== -1) {
        requestFileName = 'MainFont.aft';
      }
      let modifyFileName = '';
      if (requestFileName) {
        switch (path.extname(requestFileName)) {
          case '.atb':
            modifyFileName = requestFileName.replace('.atb', '.txt');
            break;
          case '.aar':
            modifyFileName = requestFileName.replace('.aar', '');
            break;
          default:
            modifyFileName = requestFileName;
        }
      }
      console.log(modifyFileName);
      // 文件热封装
      const protoablePath = process.env.PORTABLE_EXECUTABLE_DIR;
      const modPath = protoablePath
        ? protoablePath + '/mods'
        : path.join(userDataPath, 'mods');
      if (!fs.existsSync(modPath)) {
        fs.mkdirSync(modPath);
      }
      const modifyFilePath = path.join(modPath, modifyFileName);
      if (modifyFileName !== '' && fs.existsSync(modifyFilePath)) {
        log.info(requestFileName, 'modify by Server');
        // AFT和PNG文件直接回传
        if (
          modifyFileName === 'MainFont.aft' ||
          path.extname(modifyFileName) === 'png'
        ) {
          fs.createReadStream(modifyFilePath).pipe(res);
          return;
        }
        // 其他文件
        let result: AL;
        options.gzip = true;
        request(options, (err, response, body) => {
          result = parse(body);
          // 这边也需要添加一个任务队列，不然会爆炸
          const packaged = result.Package(modifyFilePath);
          // fs.writeFile('./test/' + requestFileName, packaged);
          res.send(packaged);
          // res.send(body);
        });
      } else {
        request(options, (err, res, body) => {
          if (body === undefined) {
            console.log('Error on ' + req.path);
          }
        }).pipe(res);
      }
    });
    const server = http.createServer(app);
    server.on('error', (e) => {
      console.log(e);
    });
    server.listen('19980', () => {
      this.Port = server.address()['port'];
      console.log('listen at ' + this.Port);
    });
  }
  setFileList(fileList) {
    this.FileList = fileList;
  }
  setFontPath(path: string) {
    mainFontPath = path;
  }
  setProxy(enable, isSocks5, host, port) {
    this.ProxyEnable = enable;
    this.ProxyIsSocks5 = isSocks5;
    if (enable) {
      this.ProxyHost = host;
      this.ProxyPort = port;
    }
  }
}