{ config, lib, ... }:

let
  cfg = config.services.matrix-synapse;
  dbGroup = "medium";
  streamWriterType = "shared"; # unused, from template
  workers = lib.range 0 (cfg.presenceStreamWriters - 1);
  workerName = "shared_stream_writer";
  workerRoutes = {
    client = [ ];
    federation = [ ];
    media = [ ];
  };
in
let
  enabledResources =
    lib.optionals (lib.length workerRoutes.client > 0) [ "client" ]
    ++ lib.optionals (lib.length workerRoutes.federation > 0) [ "federation" ]
    ++ lib.optionals (lib.length workerRoutes.media > 0) [ "media" ];
in
{
  config = lib.mkIf (cfg.presenceStreamWriters > 0) {
    monitoring.synapse.workerNames = lib.map (index: "${workerName}-${toString index}") workers;
    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
        );

        stream_writers.account_data = lib.map (index: "${workerName}-${toString index}") workers;
        stream_writers.events = lib.map (index: "${workerName}-${toString index}") workers;
        stream_writers.typing = lib.map (index: "${workerName}-${toString index}") workers;
        stream_writers.to_device = lib.map (index: "${workerName}-${toString index}") workers;
        stream_writers.receipts = lib.map (index: "${workerName}-${toString index}") workers;
        stream_writers.push_rules = lib.map (index: "${workerName}-${toString index}") 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;
                    }
                  ];
                }
              ]
              ++ lib.map (type: {
                type = "http";
                path = "/run/matrix-synapse/${workerName}-${type}-${toString index}.sock";
                mode = "666";
                resources = [
                  {
                    names = [ type ];
                    compress = false;
                  }
                ];
              }) enabledResources;
            database = (
              import ../../db.nix {
                inherit dbGroup;
                workerName = "${workerName}-${toString index}";
              }
            );
          };
        }) workers
      );
    };

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

      virtualHosts."${cfg.nginxVirtualHostName}".locations = lib.listToAttrs (
        lib.flatten (
          lib.forEach enabledResources (
            type:
            lib.map (route: {
              name = route;
              value = {
                proxyPass = "http://${workerName}-${type}";
                extraConfig = ''
                  proxy_http_version 1.1;
                  proxy_set_header Connection "";
                '';
              };
            }) workerRoutes.${type}
          )
        )
      );
    };
  };
}