1 files changed, 68 insertions, 0 deletions

diff --git a/packages/server-commands/src/server/servers.ts b/packages/server-commands/src/server/servers.ts
new file mode 100644
index 000000000..caf9866e1
--- /dev/null
+++ b/packages/server-commands/src/server/servers.ts
@@ -0,0 +1,68 @@
+import { ensureDir } from 'fs-extra/esm'
+import { isGithubCI } from '@peertube/peertube-node-utils'
+import { PeerTubeServer, RunServerOptions } from './server.js'
+
+async function createSingleServer (serverNumber: number, configOverride?: object, options: RunServerOptions = {}) {
+  const server = new PeerTubeServer({ serverNumber })
+
+  await server.flushAndRun(configOverride, options)
+
+  return server
+}
+
+function createMultipleServers (totalServers: number, configOverride?: object, options: RunServerOptions = {}) {
+  const serverPromises: Promise<PeerTubeServer>[] = []
+
+  for (let i = 1; i <= totalServers; i++) {
+    serverPromises.push(createSingleServer(i, configOverride, options))
+  }
+
+  return Promise.all(serverPromises)
+}
+
+function killallServers (servers: PeerTubeServer[]) {
+  return Promise.all(servers.map(s => s.kill()))
+}
+
+async function cleanupTests (servers: PeerTubeServer[]) {
+  await killallServers(servers)
+
+  if (isGithubCI()) {
+    await ensureDir('artifacts')
+  }
+
+  let p: Promise<any>[] = []
+  for (const server of servers) {
+    p = p.concat(server.servers.cleanupTests())
+  }
+
+  return Promise.all(p)
+}
+
+function getServerImportConfig (mode: 'youtube-dl' | 'yt-dlp') {
+  return {
+    import: {
+      videos: {
+        http: {
+          youtube_dl_release: {
+            url: mode === 'youtube-dl'
+              ? 'https://yt-dl.org/downloads/latest/youtube-dl'
+              : 'https://api.github.com/repos/yt-dlp/yt-dlp/releases',
+
+            name: mode
+          }
+        }
+      }
+    }
+  }
+}
+
+// ---------------------------------------------------------------------------
+
+export {
+  createSingleServer,
+  createMultipleServers,
+  cleanupTests,
+  killallServers,
+  getServerImportConfig
+}