photoncloud-monorepo/nix/modules/flaredb.nix

{ config, lib, pkgs, ... }:

let
  cfg = config.services.flaredb;
in
{
  options.services.flaredb = {
    enable = lib.mkEnableOption "flaredb service";

    port = lib.mkOption {
      type = lib.types.port;
      default = 2479;
      description = "Port for flaredb API";
    };

    raftPort = lib.mkOption {
      type = lib.types.port;
      default = 2480;
      description = "Port for flaredb Raft protocol";
    };

    httpPort = lib.mkOption {
      type = lib.types.port;
      default = 8082;
      description = "Port for flaredb HTTP/admin API (used for cluster join)";
    };

    dataDir = lib.mkOption {
      type = lib.types.path;
      default = "/var/lib/flaredb";
      description = "Data directory for flaredb";
    };

    settings = lib.mkOption {
      type = lib.types.attrs;
      default = {};
      description = "Additional configuration settings";
    };

    package = lib.mkOption {
      type = lib.types.package;
      default = pkgs.flaredb-server or (throw "flaredb-server package not found");
      description = "Package to use for flaredb";
    };
  };

  config = lib.mkIf cfg.enable {
    # Create system user
    users.users.flaredb = {
      isSystemUser = true;
      group = "flaredb";
      description = "FlareDB service user";
      home = cfg.dataDir;
    };

    users.groups.flaredb = {};

    # Create systemd service
    systemd.services.flaredb = {
      description = "FlareDB Distributed Database Service";
      wantedBy = [ "multi-user.target" ];
      after = [ "network.target" "chainfire.service" ];
      requires = [ "chainfire.service" ];

      environment = {
        FLAREDB_HTTP_ADDR = "0.0.0.0:${toString cfg.httpPort}";
      };

      serviceConfig = {
        Type = "simple";
        User = "flaredb";
        Group = "flaredb";
        Restart = "on-failure";
        RestartSec = "10s";

        # State directory management
        StateDirectory = "flaredb";
        StateDirectoryMode = "0750";

        # Security hardening
        NoNewPrivileges = true;
        PrivateTmp = true;
        ProtectSystem = "strict";
        ProtectHome = true;
        ReadWritePaths = [ cfg.dataDir ];

        # Start command
        ExecStart = "${cfg.package}/bin/flaredb-server --addr 0.0.0.0:${toString cfg.port} --data-dir ${cfg.dataDir}";
      };
    };
  };
}