{ config, lib, ... }:

let
  cfg = config.services.matrix-synapse;
  dbGroup = "small";
  hasClientResource = false;
  hasFederationResource = false;
  workers = lib.range 0 (cfg.clientReaders - 1);
  workerName = "client_reader";
  routes =
    [
      "~ ^/_matrix/client/(api/v1|r0|v3|unstable)/publicRooms$"
      "~ ^/_matrix/client/(api/v1|r0|v3|unstable)/rooms/.*/joined_members$"
      "~ ^/_matrix/client/(api/v1|r0|v3|unstable)/rooms/.*/context/.*$"
      "~ ^/_matrix/client/(api/v1|r0|v3|unstable)/rooms/.*/members$"
      "~ ^/_matrix/client/(api/v1|r0|v3|unstable)/rooms/.*/state$"
      "~ ^/_matrix/client/v1/rooms/.*/hierarchy$"
      "~ ^/_matrix/client/(v1|unstable)/rooms/.*/relations/"
      "~ ^/_matrix/client/v1/rooms/.*/threads$"
      "~ ^/_matrix/client/(api/v1|r0|v3|unstable)/rooms/.*/messages$"
      "~ ^/_matrix/client/(api/v1|r0|v3|unstable)/rooms/.*/event"
      "~ ^/_matrix/client/(api/v1|r0|v3|unstable)/joined_rooms"
      "~ ^/_matrix/client/(api/v1|r0|v3|unstable/.*)/rooms/.*/aliases"
      "~ ^/_matrix/client/v1/rooms/.*/timestamp_to_event$"
      "~ ^/_matrix/client/(api/v1|r0|v3|unstable)/search"
      "~ ^/_matrix/client/(r0|v3|unstable)/user/.*/filter(/|$)"
      "~ ^/_matrix/client/(api/v1|r0|v3|unstable)/directory/room/.*$"
      "~ ^/_matrix/client/(r0|v3|unstable)/notifications$"

      # unstable
      "~ ^/_matrix/client/unstable/im.nheko.summary/rooms/.*/summary$"
    ]
    ++ lib.optionals (cfg.authWorkers == 0) [
      "~ ^/_matrix/client/(api/v1|r0|v3|unstable)/login$"
      "~ ^/_matrix/client/(api/v1|r0|v3|unstable)/account/3pid$"
      "~ ^/_matrix/client/(api/v1|r0|v3|unstable)/account/whoami$"
      "~ ^/_matrix/client/versions$"
      "~ ^/_matrix/client/(api/v1|r0|v3|unstable)/voip/turnServer$"
      "~ ^/_matrix/client/(r0|v3|unstable)/register$"
      "~ ^/_matrix/client/(r0|v3|unstable)/register/available$"
      "~ ^/_matrix/client/(r0|v3|unstable)/auth/.*/fallback/web$"
      "~ ^/_matrix/client/(r0|v3|unstable)/password_policy$"
      "~ ^/_matrix/client/(r0|v3|unstable)/capabilities$"
    ];
in
{
  config = lib.mkIf (cfg.clientReaders > 0) {
    services.matrix-synapse = {
      settings = {
        instance_map = lib.listToAttrs (
          lib.map (index: {
            name = "${workerName}-${toString index}";
            value = {
              path = "/run/matrix-synapse/${workerName}-${toString index}.sock";
            };
          }) workers
        );
      };

      workers = lib.listToAttrs (
        lib.map (index: {
          name = "${workerName}-${toString index}";
          value = {
            worker_app = "synapse.app.generic_worker";
            worker_listeners = [
              {
                type = "http";
                path = "/run/matrix-synapse/${workerName}-${toString index}.sock";
                resources = [
                  {
                    names = [ "replication" ];
                    compress = false;
                  }
                ];
              }
              {
                type = "http";
                path = "/run/matrix-synapse/${workerName}-client-${toString index}.sock";
                mode = "666";
                resources = [
                  {
                    names = [ "client" ];
                    compress = false;
                  }
                ];
              }
            ];
            database = (
              import ../db.nix {
                inherit dbGroup;
                workerName = "${workerName}-${toString index}";
              }
            );
          };
        }) workers
      );
    };

    services.nginx.upstreams."${workerName}" = {
      extraConfig = ''
        keepalive 32;
      '';
      servers = lib.listToAttrs (
        lib.map (index: {
          name = "unix:/run/matrix-synapse/${workerName}-client-${toString index}.sock";
          value = {
            max_fails = 0;
          };
        }) workers
      );
    };

    services.nginx.virtualHosts."${cfg.nginxVirtualHostName}".locations = lib.listToAttrs (
      lib.map (route: {
        name = route;
        value = {
          proxyPass = "http://${workerName}";
        };
      }) routes
    );
  };
}