commit 255ac994b8dcde49c22e5b7b9251630db53ff72c · pyrox.dev/nixpkgs

+40 -50
nixos/modules/services/networking/firefox-syncserver.nix
···

       32
       32
        
           };

     

       33
       33
        
         };

     

       34
       34
        
         configFile = format.generate "syncstorage.toml" (lib.recursiveUpdate settings cfg.settings);

     

       35
       35
       +
         setupScript = pkgs.writeShellScript "firefox-syncserver-setup" ''

     

       36
       36
       +
               set -euo pipefail

     

       37
       37
       +
               shopt -s inherit_errexit

     

       38
       38
       +
       

     

       39
       39
       +
               schema_configured() {

     

       40
       40
       +
                 mysql ${cfg.database.name} -Ne 'SHOW TABLES' | grep -q services

     

       41
       41
       +
               }

     

       42
       42
       +
       

     

       43
       43
       +
               update_config() {

     

       44
       44
       +
                 mysql ${cfg.database.name} <<"EOF"

     

       45
       45
       +
                   BEGIN;

     

       46
       46
       +
       

     

       47
       47
       +
                   INSERT INTO `services` (`id`, `service`, `pattern`)

     

       48
       48
       +
                     VALUES (1, 'sync-1.5', '{node}/1.5/{uid}')

     

       49
       49
       +
                     ON DUPLICATE KEY UPDATE service='sync-1.5', pattern='{node}/1.5/{uid}';

     

       50
       50
       +
                   INSERT INTO `nodes` (`id`, `service`, `node`, `available`, `current_load`,

     

       51
       51
       +
                                        `capacity`, `downed`, `backoff`)

     

       52
       52
       +
                     VALUES (1, 1, '${cfg.singleNode.url}', ${toString cfg.singleNode.capacity},

     

       53
       53
       +
                     0, ${toString cfg.singleNode.capacity}, 0, 0)

     

       54
       54
       +
                     ON DUPLICATE KEY UPDATE node = '${cfg.singleNode.url}', capacity=${toString cfg.singleNode.capacity};

     

       55
       55
       +
       

     

       56
       56
       +
                   COMMIT;

     

       57
       57
       +
               EOF

     

       58
       58
       +
               }

     

       59
       59
       +
       

     

       60
       60
       +
       

     

       61
       61
       +
               for (( try = 0; try < 60; try++ )); do

     

       62
       62
       +
                 if ! schema_configured; then

     

       63
       63
       +
                   sleep 2

     

       64
       64
       +
                 else

     

       65
       65
       +
                   update_config

     

       66
       66
       +
                   exit 0

     

       67
       67
       +
                 fi

     

       68
       68
       +
               done

     

       69
       69
       +
       

     

       70
       70
       +
               echo "Single-node setup failed"

     

       71
       71
       +
               exit 1

     

       72
       72
       +
             '';

     

       35
       73
        
       in

     

       36
       74
        
       

     

       37
       75
        
       {

     
···

       210
       248
        
             wantedBy = [ "multi-user.target" ];

     

       211
       249
        
             requires = lib.mkIf dbIsLocal [ "mysql.service" ];

     

       212
       250
        
             after = lib.mkIf dbIsLocal [ "mysql.service" ];

     

       251
       251
       +
             restartTriggers = lib.optional cfg.singleNode.enable setupScript;

     

       213
       252
        
             environment.RUST_LOG = cfg.logLevel;

     

       214
       253
        
             serviceConfig = {

     

       215
       254
        
               User = defaultUser;

     
···

       255
       294
        
             requires = [ "firefox-syncserver.service" ] ++ lib.optional dbIsLocal "mysql.service";

     

       256
       295
        
             after = [ "firefox-syncserver.service" ] ++ lib.optional dbIsLocal "mysql.service";

     

       257
       296
        
             path = [ config.services.mysql.package ];

     

       258
       258
       -
             script = ''

     

       259
       259
       -
               set -euo pipefail

     

       260
       260
       -
               shopt -s inherit_errexit

     

       261
       261
       -
       

     

       262
       262
       -
               schema_configured() {

     

       263
       263
       -
                 mysql ${cfg.database.name} -Ne 'SHOW TABLES' | grep -q services

     

       264
       264
       -
               }

     

       265
       265
       -
       

     

       266
       266
       -
               services_configured() {

     

       267
       267
       -
                 [ 1 != $(mysql ${cfg.database.name} -Ne 'SELECT COUNT(*) < 1 FROM `services`') ]

     

       268
       268
       -
               }

     

       269
       269
       -
       

     

       270
       270
       -
               create_services() {

     

       271
       271
       -
                 mysql ${cfg.database.name} <<"EOF"

     

       272
       272
       -
                   BEGIN;

     

       273
       273
       -
       

     

       274
       274
       -
                   INSERT INTO `services` (`id`, `service`, `pattern`)

     

       275
       275
       -
                     VALUES (1, 'sync-1.5', '{node}/1.5/{uid}');

     

       276
       276
       -
                   INSERT INTO `nodes` (`id`, `service`, `node`, `available`, `current_load`,

     

       277
       277
       -
                                        `capacity`, `downed`, `backoff`)

     

       278
       278
       -
                     VALUES (1, 1, '${cfg.singleNode.url}', ${toString cfg.singleNode.capacity},

     

       279
       279
       -
                             0, ${toString cfg.singleNode.capacity}, 0, 0);

     

       280
       280
       -
       

     

       281
       281
       -
                   COMMIT;

     

       282
       282
       -
               EOF

     

       283
       283
       -
               }

     

       284
       284
       -
       

     

       285
       285
       -
               update_nodes() {

     

       286
       286
       -
                 mysql ${cfg.database.name} <<"EOF"

     

       287
       287
       -
                   UPDATE `nodes`

     

       288
       288
       -
                     SET `capacity` = ${toString cfg.singleNode.capacity}

     

       289
       289
       -
                     WHERE `id` = 1;

     

       290
       290
       -
               EOF

     

       291
       291
       -
               }

     

       292
       292
       -
       

     

       293
       293
       -
               for (( try = 0; try < 60; try++ )); do

     

       294
       294
       -
                 if ! schema_configured; then

     

       295
       295
       -
                   sleep 2

     

       296
       296
       -
                 elif services_configured; then

     

       297
       297
       -
                   update_nodes

     

       298
       298
       -
                   exit 0

     

       299
       299
       -
                 else

     

       300
       300
       -
                   create_services

     

       301
       301
       -
                   exit 0

     

       302
       302
       -
                 fi

     

       303
       303
       -
               done

     

       304
       304
       -
       

     

       305
       305
       -
               echo "Single-node setup failed"

     

       306
       306
       -
               exit 1

     

       307
       307
       -
             '';

     

       297
       297
       +
             serviceConfig.ExecStart = [ "${setupScript}" ];

     

       308
       298
        
           };

     

       309
       299
        
       

     

       310
       300
        
           services.nginx.virtualHosts = lib.mkIf cfg.singleNode.enableNginx {