{ config, pkgs, ... }:

with pkgs.lib;
with pkgs;

let

  cfg = config.services.activemq;

  activemqBroker = stdenv.mkDerivation {
    name = "activemq-broker";
    phases = [ "installPhase" ];
    buildInputs = [ jdk ];
    installPhase = ''
      ensureDir $out/lib
      source ${activemq}/lib/classpath.env
      export CLASSPATH
      ln -s "${./ActiveMQBroker.java}" ActiveMQBroker.java
      javac -d $out/lib ActiveMQBroker.java
    '';
  };

in {

  options = {
    services.activemq = {
      enable = mkOption {
        type = types.bool;
        default = false;
        description = ''
          Enable the Apache ActiveMQ message broker service.
        '';
      };
      configurationDir = mkOption {
        default = "${activemq}/conf";
        description = ''
          The base directory for ActiveMQ's configuration.
          By default, this directory is searched for a file named activemq.xml,
          which should contain the configuration for the broker service.
        '';
      };
      configurationURI = mkOption {
        type = types.string;
        default = "xbean:activemq.xml";
        description = ''
          The URI that is passed along to the BrokerFactory to
          set up the configuration of the ActiveMQ broker service.
          You should not need to change this. For custom configuration,
          set the <literal>configurationDir</literal> instead, and create
          an activemq.xml configuration file in it.
        '';
      };
      baseDir = mkOption {
        type = types.string;
        default = "/var/activemq";
        description = ''
          The base directory where ActiveMQ stores its persistent data and logs.
          This will be overriden if you set "activemq.base" and "activemq.data"
          in the <literal>javaProperties</literal> option. You can also override
          this in activemq.xml.
        '';
      };
      javaProperties = mkOption {
        type = types.attrs;
        default = { };
        example = {
          "java.net.preferIPv4Stack" = "true";
        };
        apply = attrs: {
          "activemq.base" = "${cfg.baseDir}";
          "activemq.data" = "${cfg.baseDir}/data";
          "activemq.conf" = "${cfg.configurationDir}";
          "activemq.home" = "${activemq}";
        } // attrs;
        description = ''
          Specifies Java properties that are sent to the ActiveMQ
          broker service with the "-D" option. You can set properties
          here to change the behaviour and configuration of the broker.
          All essential properties that are not set here are automatically
          given reasonable defaults.
        '';
      };
      extraJavaOptions = mkOption {
        type = types.string;
        default = "";
        example = "-Xmx2G -Xms2G -XX:MaxPermSize=512M";
        description = ''
          Add extra options here that you want to be sent to the
          Java runtime when the broker service is started.
        '';
      };
      user = {
        create = mkOption {
          type = types.bool;
          default = true;
          description = ''
            If true, the a system user with the specified name will be added
            to the system configuration. If false, a user with the specified
            name is expected to exist.
          '';
        };
        name = mkOption {
          type = types.string;
          default = "activemq";
          description = ''
            The name of the user that should run the ActiveMQ process.
          '';
        };
      };
    };
  };

  config = mkIf cfg.enable {
    users.extraUsers = mkIf cfg.user.create (singleton {
      inherit (cfg.user) name;
      description = "ActiveMQ server user";
    });

    systemd.services.activemq_init = {
      wantedBy = [ "activemq.service" ];
      partOf = [ "activemq.service" ];
      before = [ "activemq.service" ];
      serviceConfig.Type = "oneshot";
      script = ''
        mkdir -p "${cfg.javaProperties."activemq.data"}"
        chown -R activemq "${cfg.javaProperties."activemq.data"}"
      '';
    };

    systemd.services.activemq = {
      wantedBy = [ "multi-user.target" ];
      after = [ "network.target" ];
      path = [ jre ];
      serviceConfig.User = "activemq";
      script = ''
        source ${activemq}/lib/classpath.env
        export CLASSPATH=${activemqBroker}/lib:${cfg.configurationDir}:$CLASSPATH
        exec java \
          ${concatStringsSep " \\\n" (mapAttrsToList (name: value: "-D${name}=${value}") cfg.javaProperties)} \
          ${cfg.extraJavaOptions} ActiveMQBroker "${cfg.configurationURI}"
      '';
    };

  };

}