tauri-plugins-workspace/packages/api-e2e/test/helpers/server.ts

// Copyright 2019-2023 Tauri Programme within The Commons Conservancy
// SPDX-License-Identifier: Apache-2.0
// SPDX-License-Identifier: MIT

import http from 'node:http'
import { WebSocketServer } from 'ws'

/**
 * Where the fixture server listens. The port is fixed because the updater
 * endpoint is baked into the app at build time (see `tauri.e2e.conf.json`).
 */
export const FIXTURE_SERVER_PORT = 3004
export const FIXTURE_SERVER_URL = `http://127.0.0.1:${FIXTURE_SERVER_PORT}`

/** Version the `/updater` endpoint advertises by default. */
export const UPDATER_FIXTURE_VERSION = '2.1.0'
export const UPDATER_FIXTURE_NOTES = 'Test update from the e2e fixture server'
/** `{{target}}` values with special behavior on the `/updater` endpoint. */
export const UPDATER_TARGET_NO_UPDATE = 'e2e-no-update'
export const UPDATER_TARGET_OLDER = 'e2e-older'
export const UPDATER_FIXTURE_OLDER_VERSION = '1.0.0'

/** Body served by `GET /download`. */
export const DOWNLOAD_FIXTURE_BODY =
  'hello from the plugins e2e fixture server\n'.repeat(64)

/** WebSocket endpoint of the fixture server (the `ws://` twin of {@link FIXTURE_SERVER_URL}). */
export const WEBSOCKET_FIXTURE_URL = `ws://127.0.0.1:${FIXTURE_SERVER_PORT}/ws`
/** Text message that makes the `/ws` endpoint close the connection with {@link WEBSOCKET_CLOSE_CODE}. */
export const WEBSOCKET_CLOSE_REQUEST = 'close-me'
export const WEBSOCKET_CLOSE_CODE = 4000
export const WEBSOCKET_CLOSE_REASON = 'closed by the fixture server'

export interface FixtureServer {
  close(): void
}

/**
 * A tiny HTTP server the network-facing specs (updater, upload) talk to:
 *
 * - `GET /updater/{{target}}/{{arch}}/{{current_version}}` — an updater
 *   manifest in the dynamic format. Advertises {@link UPDATER_FIXTURE_VERSION},
 *   or {@link UPDATER_FIXTURE_OLDER_VERSION} when the target is
 *   {@link UPDATER_TARGET_OLDER}, and replies `204 No Content` when it is
 *   {@link UPDATER_TARGET_NO_UPDATE}.
 * - `GET /download` — {@link DOWNLOAD_FIXTURE_BODY} with a `Content-Length`.
 * - `* /echo` — a JSON description of the request (`method`, `url`, `headers`
 *   and the utf-8 `body`).
 * - `ws /ws` — a WebSocket echo endpoint: text and binary messages are sent
 *   back as-is, and {@link WEBSOCKET_CLOSE_REQUEST} makes the server close the
 *   connection with {@link WEBSOCKET_CLOSE_CODE}. On `/ws/headers` the server
 *   first sends the upgrade request's headers as a JSON text message.
 */
export function startFixtureServer(): Promise<FixtureServer> {
  const server = http.createServer((req, res) => {
    const chunks: Buffer[] = []
    req.on('data', (chunk: Buffer) => chunks.push(chunk))
    req.on('end', () => {
      const body = Buffer.concat(chunks)
      const url = new URL(req.url ?? '/', FIXTURE_SERVER_URL)
      const [, route, ...rest] = url.pathname.split('/')

      if (route === 'updater' && req.method === 'GET') {
        const [target] = rest
        if (target === UPDATER_TARGET_NO_UPDATE) {
          res.writeHead(204).end()
          return
        }
        json(res, {
          version:
            target === UPDATER_TARGET_OLDER
              ? UPDATER_FIXTURE_OLDER_VERSION
              : UPDATER_FIXTURE_VERSION,
          notes: UPDATER_FIXTURE_NOTES,
          pub_date: '2026-03-01T14:04:20Z',
          url: `${FIXTURE_SERVER_URL}/download`,
          signature: ''
        })
        return
      }

      if (route === 'download' && req.method === 'GET') {
        res
          .writeHead(200, {
            'content-type': 'text/plain',
            'content-length': Buffer.byteLength(DOWNLOAD_FIXTURE_BODY)
          })
          .end(DOWNLOAD_FIXTURE_BODY)
        return
      }

      if (route === 'echo') {
        json(res, {
          method: req.method,
          url: req.url,
          headers: req.headers,
          body: body.toString('utf8')
        })
        return
      }

      res.writeHead(404).end()
    })
  })

  const wss = new WebSocketServer({ noServer: true })
  server.on('upgrade', (req, socket, head) => {
    const { pathname } = new URL(req.url ?? '/', FIXTURE_SERVER_URL)
    if (pathname !== '/ws' && pathname !== '/ws/headers') {
      socket.destroy()
      return
    }
    wss.handleUpgrade(req, socket, head, (ws) => {
      if (pathname === '/ws/headers') {
        ws.send(JSON.stringify(req.headers))
      }
      ws.on('message', (data, isBinary) => {
        if (
          !isBinary
          && Buffer.isBuffer(data)
          && data.toString('utf8') === WEBSOCKET_CLOSE_REQUEST
        ) {
          ws.close(WEBSOCKET_CLOSE_CODE, WEBSOCKET_CLOSE_REASON)
        } else {
          ws.send(data, { binary: isBinary })
        }
      })
    })
  })

  return new Promise((resolve, reject) => {
    server.once('error', reject)
    server.listen(FIXTURE_SERVER_PORT, '127.0.0.1', () => {
      server.off('error', reject)
      resolve({
        close: () => {
          for (const client of wss.clients) client.terminate()
          wss.close()
          server.closeAllConnections()
          server.close()
        }
      })
    })
  })
}

function json(res: http.ServerResponse, value: unknown) {
  const payload = JSON.stringify(value)
  res
    .writeHead(200, {
      'content-type': 'application/json',
      'content-length': Buffer.byteLength(payload)
    })
    .end(payload)
}