···

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

     

       
2
       
-
       


     

       
3
       
-
       
with lib;

     

       
4
       
-
       


     

       
5
       
 
       
let

     

       
0
       
 
       

     

       
6
       
 
       
  cfg = config.services.atuin;

     

       
7
       
 
       
in

     

       
8
       
 
       
{

     

       
9
       
 
       
  options = {

     

       
10
       
 
       
    services.atuin = {

     

       
11
       
-
       
      enable = mkEnableOption (mdDoc "Enable server for shell history sync with atuin");

     

       
12
       
 
       


     

       
13
       
 
       
      openRegistration = mkOption {

     

       
14
       
 
       
        type = types.bool;

     
···

       
50
       
 
       
        createLocally = mkOption {

     

       
51
       
 
       
          type = types.bool;

     

       
52
       
 
       
          default = true;

     

       
53
       
-
       
          description = lib.mdDoc "Create the database and database user locally.";

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
54
       
 
       
        };

     

       
55
       
 
       
      };

     

       
56
       
 
       
    };

     

       
57
       
 
       
  };

     

       
58
       
 
       


     

       
59
       
 
       
  config = mkIf cfg.enable {

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
60
       
 
       


     

       
61
       
-
       
    # enable postgres to host atuin db

     

       
62
       
-
       
    services.postgresql = {

     

       
63
       
 
       
      enable = true;

     

       
64
       
 
       
      ensureUsers = [{

     

       
65
       
 
       
        name = "atuin";

     
···

       
73
       
 
       
    systemd.services.atuin = {

     

       
74
       
 
       
      description = "atuin server";

     

       
75
       
 
       
      requires = lib.optionals cfg.database.createLocally [ "postgresql.service" ];

     

       
76
       
-
       
      after = [ "network.target" ] ++ lib.optionals cfg.database.createLocally [ "postgresql.service" ] ;

     

       
77
       
 
       
      wantedBy = [ "multi-user.target" ];

     

       
78
       
 
       


     

       
79
       
 
       
      serviceConfig = {

     
···

       
87
       
 
       
        ATUIN_HOST = cfg.host;

     

       
88
       
 
       
        ATUIN_PORT = toString cfg.port;

     

       
89
       
 
       
        ATUIN_MAX_HISTORY_LENGTH = toString cfg.maxHistoryLength;

     

       
90
       
-
       
        ATUIN_OPEN_REGISTRATION = boolToString cfg.openRegistration;

     

       
91
       
-
       
        ATUIN_DB_URI =  mkIf cfg.database.createLocally "postgresql:///atuin";

     

       
92
       
 
       
        ATUIN_PATH = cfg.path;

     

       
93
       
 
       
        ATUIN_CONFIG_DIR = "/run/atuin"; # required to start, but not used as configuration is via environment variables

     

       
94
       
 
       
      };

     

       
95
       
 
       
    };

     

       
96
       
 
       


     

       
97
       
 
       
    networking.firewall.allowedTCPPorts = mkIf cfg.openFirewall [ cfg.port ];

     

       
98
       
-
       


     

       
99
       
 
       
  };

     

       
100
       
 
       
}

     

···

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

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
2
       
 
       
let

     

       
3
       
+
       
  inherit (lib) mkOption types mdDoc mkIf;

     

       
4
       
 
       
  cfg = config.services.atuin;

     

       
5
       
 
       
in

     

       
6
       
 
       
{

     

       
7
       
 
       
  options = {

     

       
8
       
 
       
    services.atuin = {

     

       
9
       
+
       
      enable = lib.mkEnableOption (mdDoc "Enable server for shell history sync with atuin");

     

       
10
       
 
       


     

       
11
       
 
       
      openRegistration = mkOption {

     

       
12
       
 
       
        type = types.bool;

     
···

       
48
       
 
       
        createLocally = mkOption {

     

       
49
       
 
       
          type = types.bool;

     

       
50
       
 
       
          default = true;

     

       
51
       
+
       
          description = mdDoc "Create the database and database user locally.";

     

       
52
       
+
       
        };

     

       
53
       
+
       


     

       
54
       
+
       
        uri = mkOption {

     

       
55
       
+
       
          type = types.str;

     

       
56
       
+
       
          default = "postgresql:///atuin?host=/run/postgresql";

     

       
57
       
+
       
          example = "postgresql://atuin@localhost:5432/atuin";

     

       
58
       
+
       
          description = mdDoc "URI to the database";

     

       
59
       
 
       
        };

     

       
60
       
 
       
      };

     

       
61
       
 
       
    };

     

       
62
       
 
       
  };

     

       
63
       
 
       


     

       
64
       
 
       
  config = mkIf cfg.enable {

     

       
65
       
+
       
    assertions = [

     

       
66
       
+
       
      {

     

       
67
       
+
       
        assertion = cfg.database.createLocally -> config.services.postgresql.enable;

     

       
68
       
+
       
        message = "Postgresql must be enabled to create a local database";

     

       
69
       
+
       
      }

     

       
70
       
+
       
    ];

     

       
71
       
 
       


     

       
72
       
+
       
    services.postgresql = mkIf cfg.database.createLocally {

     

       
0
       
 
       

     

       
73
       
 
       
      enable = true;

     

       
74
       
 
       
      ensureUsers = [{

     

       
75
       
 
       
        name = "atuin";

     
···

       
83
       
 
       
    systemd.services.atuin = {

     

       
84
       
 
       
      description = "atuin server";

     

       
85
       
 
       
      requires = lib.optionals cfg.database.createLocally [ "postgresql.service" ];

     

       
86
       
+
       
      after = [ "network.target" ] ++ lib.optionals cfg.database.createLocally [ "postgresql.service" ];

     

       
87
       
 
       
      wantedBy = [ "multi-user.target" ];

     

       
88
       
 
       


     

       
89
       
 
       
      serviceConfig = {

     
···

       
97
       
 
       
        ATUIN_HOST = cfg.host;

     

       
98
       
 
       
        ATUIN_PORT = toString cfg.port;

     

       
99
       
 
       
        ATUIN_MAX_HISTORY_LENGTH = toString cfg.maxHistoryLength;

     

       
100
       
+
       
        ATUIN_OPEN_REGISTRATION = lib.boolToString cfg.openRegistration;

     

       
101
       
+
       
        ATUIN_DB_URI = cfg.database.uri;

     

       
102
       
 
       
        ATUIN_PATH = cfg.path;

     

       
103
       
 
       
        ATUIN_CONFIG_DIR = "/run/atuin"; # required to start, but not used as configuration is via environment variables

     

       
104
       
 
       
      };

     

       
105
       
 
       
    };

     

       
106
       
 
       


     

       
107
       
 
       
    networking.firewall.allowedTCPPorts = mkIf cfg.openFirewall [ cfg.port ];

     

       
0
       
 
       

     

       
108
       
 
       
  };

     

       
109
       
 
       
}

     

···

       
14
       
 
       
    server =

     

       
15
       
 
       
      { ... }:

     

       
16
       
 
       
      {

     

       
0
       
 
       

     

       
0
       
 
       

     

       
17
       
 
       
        services.atuin = {

     

       
18
       
 
       
          enable = true;

     

       
19
       
 
       
          port = testPort;

     

···

       
14
       
 
       
    server =

     

       
15
       
 
       
      { ... }:

     

       
16
       
 
       
      {

     

       
17
       
+
       
        services.postgresql.enable = true;

     

       
18
       
+
       


     

       
19
       
 
       
        services.atuin = {

     

       
20
       
 
       
          enable = true;

     

       
21
       
 
       
          port = testPort;