Make the reverse_proxy configurable

Author: webNeat

src/blocks/create_domain.ts renamed to src/blocks/create_caddy_domain.ts

@@ -4,25 +4,18 @@ import { block } from './block.js'
 
 type Config = {
   domain: string
-  ports_var: string
   caddyfile_path: string
+  caddyfile_content: string
 }
 
-const reverse_proxy = (x: Config) =>
-  `${x.domain} {
-  reverse_proxy {{ ${x.ports_var} | map('regex_replace', '^', '127.0.0.1:') | join(' ') }} {
-    lb_policy client_ip_hash
-  }
-}`
-
-export function create_domain(config: Config): Block {
-  return block(`Configure domain: ${config.domain}`, {}, [
+export function create_caddy_domain(config: Config): Block {
+  return block(`Configure Caddy domain: ${config.domain}`, {}, [
     builtin.lineinfile(
       `Ensure ${config.domain} is in /etc/hosts`,
       { path: '/etc/hosts', line: `127.0.0.1 ${config.domain}`, state: 'present' },
       { become: true },
     ),
-    builtin.copy(`Create Caddyfile for ${config.domain}`, { dest: config.caddyfile_path, content: reverse_proxy(config) }, { register: 'caddyfile' }),
+    builtin.copy(`Create Caddyfile for ${config.domain}`, { dest: config.caddyfile_path, content: config.caddyfile_content }, { register: 'caddyfile' }),
     builtin.command(`Reload caddy`, { cmd: `sudo systemctl reload caddy` }, { become: true, when: 'caddyfile.changed' }),
   ]).get()
 }

src/blocks/create_directory.ts

@@ -1,9 +1,14 @@
 import { builtin } from '../ansible/tasks/index.js'
 
-export function create_directory(path: string) {
-  return builtin.file(
-    `Create directory ${path}`,
-    { path, state: 'directory', owner: '{{ansible_user}}', group: '{{ansible_user}}', mode: '0755' },
-    { become: true },
-  )
+type Config = {
+  owner?: string
+  group?: string
+  mode?: string
+}
+
+export function create_directory(path: string, config: Config = {}) {
+  config.owner ||= '{{ansible_user}}'
+  config.group ||= '{{ansible_user}}'
+  config.mode ||= '0755'
+  return builtin.file(`Create directory ${path}`, { path, state: 'directory', ...config }, { become: true })
 }

src/blocks/index.ts

@@ -1,7 +1,7 @@
 export * as assert from './assert.js'
 export * from './build_repo.js'
 export * from './create_directory.js'
-export * from './create_domain.js'
+export * from './create_caddy_domain.js'
 export * from './create_service.js'
 export * from './delete_directory.js'
 export * from './delete_docker_image.js'

src/blocks/install_caddy.ts

@@ -1,8 +1,12 @@
+import { block } from './block.js'
 import { Block } from '../ansible/types.js'
 import { builtin } from '../ansible/tasks/index.js'
-import { block } from './block.js'
 
-export function install_caddy(caddyfiles_pattern: string): Block {
+type Config = {
+  caddyfile_content: string
+}
+
+export function install_caddy(config: Config): Block {
   return block(`Install Caddy`, {}, [
     builtin.apt(
       `Install Caddy's dependencies`,
@@ -26,14 +30,7 @@ export function install_caddy(caddyfiles_pattern: string): Block {
     ),
     builtin.apt(`Update apt cache`, { update_cache: true }, { become: true }),
     builtin.apt(`Install Caddy`, { name: 'caddy', state: 'present' }, { become: true }),
-    builtin.copy(
-      `Configure Caddy`,
-      {
-        dest: '/etc/caddy/Caddyfile',
-        content: `import ${caddyfiles_pattern}\n`,
-      },
-      { become: true },
-    ),
+    builtin.copy(`Configure Caddy`, { dest: '/etc/caddy/Caddyfile', content: config.caddyfile_content }, { become: true }),
     builtin.command(`Reload Caddy config`, { cmd: `sudo systemctl start caddy` }, { become: true }),
   ]).get()
 }

src/files.ts

@@ -0,0 +1,25 @@
+import fs from 'fs'
+import path from 'path'
+import { fileURLToPath } from 'url'
+
+type FileData = {
+  server_caddyfile: {
+    log_path: string
+    service_caddyfiles_pattern: string
+  }
+  service_caddyfile: {
+    domain: string
+    local_urls: string
+  }
+}
+
+const __dirname = path.dirname(fileURLToPath(import.meta.url))
+const files_dir = path.join(__dirname, 'files')
+
+export function get_file<Name extends keyof FileData>(name: Name, data: FileData[Name]) {
+  let content = fs.readFileSync(path.join(files_dir, name), 'utf-8')
+  for (const [key, value] of Object.entries(data)) {
+    content = content.replace(new RegExp(`\\{\\{${key}\\}\\}`, 'g'), value)
+  }
+  return content
+}

src/files/server_caddyfile

@@ -0,0 +1,11 @@
+{
+	log {
+		format json
+		output file {{log_path}} {
+			roll_size 10mb
+			roll_keep 10
+		}
+	}
+}
+
+import {{service_caddyfiles_pattern}}

src/files/service_caddyfile

@@ -0,0 +1,5 @@
+{{domain}} {
+  reverse_proxy {{local_urls}} {
+    lb_policy client_ip_hash
+  }
+}

src/reverse_proxy/caddy.ts

@@ -0,0 +1,36 @@
+import path from 'path'
+import { get_file } from '../files.js'
+import * as blocks from '../blocks/index.js'
+import { CaddyConfig, ReverseProxy, ReverseProxyConfig, Server } from '../types.js'
+
+const default_config: Required<CaddyConfig> = {
+  get_server_caddyfile: (server) =>
+    get_file('server_caddyfile', {
+      log_path: `${server.logs_dir}/caddy.log`,
+      service_caddyfiles_pattern: `${server.services_dir}/*/Caddyfile`,
+    }),
+  get_service_caddyfile: (server, config) => get_file('service_caddyfile', config),
+}
+
+export function caddy(config: CaddyConfig = {}): ReverseProxy {
+  const normalized_config = { ...default_config, ...config }
+  return {
+    get_log_path: (server) => `${server.logs_dir}/caddy.log`,
+    get_server_tasks: (server) => get_server_tasks(normalized_config, server),
+    get_service_tasks: (server, reverse_proxy_config) => get_service_tasks(normalized_config, server, reverse_proxy_config),
+  }
+}
+
+function get_server_tasks(config: Required<CaddyConfig>, server: Server) {
+  return [blocks.install_caddy({ caddyfile_content: config.get_server_caddyfile(server) })]
+}
+
+function get_service_tasks(config: Required<CaddyConfig>, server: Server, reverse_proxy_config: ReverseProxyConfig) {
+  return [
+    blocks.create_caddy_domain({
+      domain: reverse_proxy_config.domain,
+      caddyfile_path: path.join(server.get_service_dir(reverse_proxy_config.service_name), 'Caddyfile'),
+      caddyfile_content: config.get_service_caddyfile(server, reverse_proxy_config),
+    }),
+  ]
+}

src/reverse_proxy/index.ts

@@ -0,0 +1 @@
+export * from './caddy.js'

src/server.ts

@@ -3,6 +3,7 @@ import path from 'path'
 import { Host } from './ansible/types.js'
 import * as blocks from './blocks/index.js'
 import { Server, ServerConfig } from './types.js'
+import { caddy } from './reverse_proxy/index.js'
 
 export function server(config: ServerConfig): Server {
   const user = os.userInfo().username
@@ -14,16 +15,20 @@ export function server(config: ServerConfig): Server {
   const hosty_dir = config.hosty_dir || '/srv/hosty'
   const backups_dir = path.join(hosty_dir, 'backups')
   const services_dir = path.join(hosty_dir, 'services')
+  const logs_dir = path.join(hosty_dir, 'logs')
+
   return {
     connection,
     hosty_dir,
     backups_dir,
     services_dir,
+    logs_dir,
     name: config.name,
     ssh_key: config.ssh_key || { path: '~/.ssh/id_rsa', passphrase: '' },
     git_config: config.git_config || {},
     docker_network: config.docker_network || 'hosty',
     docker_prefix: config.docker_prefix || '',
+    reverse_proxy: config.reverse_proxy || caddy(),
     get_service_dir: (name) => path.join(services_dir, name),
     get_backups_dir: (name) => path.join(backups_dir, name),
   }
@@ -58,6 +63,7 @@ export function get_setup_tasks(server: Server) {
     blocks.generate_ssh_key(server.ssh_key),
     blocks.install_nixpacks(),
     blocks.create_directory(server.hosty_dir),
-    blocks.install_caddy(`${server.services_dir}/*/Caddyfile`),
+    blocks.create_directory(server.logs_dir, { mode: '0777' }),
+    ...server.reverse_proxy.get_server_tasks(server),
   ]
 }