From 729a55b7a34165cf3d2d4fba0fb5497e55becd61 Mon Sep 17 00:00:00 2001
From: CodeRedDev <codered.dev.germany@gmail.com>
Date: Sat, 2 Dec 2023 22:23:40 +0100
Subject: [PATCH] Add support for FTP on admin-lists.js

---
 squad-server/utils/admin-lists.js | 67 +++++++++++++++++++++++++++++++
 1 file changed, 67 insertions(+)

diff --git a/squad-server/utils/admin-lists.js b/squad-server/utils/admin-lists.js
index ff2a876b9..5fdfb57b2 100644
--- a/squad-server/utils/admin-lists.js
+++ b/squad-server/utils/admin-lists.js
@@ -1,6 +1,8 @@
 import fs from 'fs';
 import path from 'path';
 import { fileURLToPath } from 'url';
+import crypto from 'crypto';
+import { Client } from 'basic-ftp';
 
 import axios from 'axios';
 import Logger from 'core/logger';
@@ -31,6 +33,15 @@ export default async function fetchAdminLists(adminLists) {
           data = fs.readFileSync(listPath, 'utf8');
           break;
         }
+        case 'ftp': {
+          const provider = new FTPAdminListProvider({
+            ...list.ftp,
+            source: list.source
+          });
+          data = await provider.fetch();
+          provider.close();
+          break;
+        }
         default:
           throw new Error(`Unsupported AdminList type:${list.type}`);
       }
@@ -84,3 +95,59 @@ export default async function fetchAdminLists(adminLists) {
   Logger.verbose('SquadServer', 1, `${Object.keys(admins).length} admins loaded...`);
   return admins;
 }
+
+class FTPAdminListProvider {
+  constructor(options) {
+    this.options = {
+      host: options.host,
+      port: options.port || 21,
+      user: options.user,
+      password: options.password,
+      secure: options.secure || false,
+      timeout: options.timeout || 2000,
+      downloadPath: options.source,
+      encoding: options.encoding || 'utf8',
+      verbose: options.verbose || false
+    }
+
+    this.client = new Client(this.options.timeout);
+    this.client.ftp.verbose = this.options.verbose;
+    this.client.encoding = this.options.encoding;
+
+    this.tempFilePath = path.join(
+      process.cwd(),
+      crypto
+        .createHash('md5')
+        .update(`${this.options.host}:${this.options.port}:${this.options.path}:Admins.cfg`)
+        .digest('hex') + '.tmp'
+    );
+  }
+
+  async fetch() {
+    if (!this.client.closed) throw new Error("Tried to fetch admin list on closed FTP client!");
+    try {
+      await this.client.access({
+        host: this.options.host,
+        port: this.options.port,
+        user: this.options.user,
+        password: this.options.password,
+        secure: this.options.secure
+      });
+
+      await this.client.downloadTo(this.tempFilePath, this.options.downloadPath);
+
+      if (!fs.existsSync(this.tempFilePath)) throw new Error(`No admin file ${this.tempFilePath} found!`);
+
+      return fs.readFileSync(this.tempFilePath, 'utf8');
+    } catch (err) {
+      console.log(err);
+      throw new Error("Failed to fetch admin list from FTP!");
+    }
+  }
+
+  close() {
+    if (this.client.closed) return;
+
+    this.client.close();
+  }
+}
\ No newline at end of file