{ config, lib, pkgs, ... }: with lib; let cfg = config.services.dockerRegistry; blobCache = if cfg.enableRedisCache then "redis" else "inmemory"; registryConfig = { version = "0.1"; log.fields.service = "registry"; storage = { cache.blobdescriptor = blobCache; filesystem.rootdirectory = cfg.storagePath; delete.enabled = cfg.enableDelete; }; http = { addr = ":${builtins.toString cfg.port}"; headers.X-Content-Type-Options = ["nosniff"]; }; health.storagedriver = { enabled = true; interval = "10s"; threshold = 3; }; }; registryConfig.redis = mkIf cfg.enableRedisCache { addr = "${cfg.redisUrl}"; password = "${cfg.redisPassword}"; db = 0; dialtimeout = "10ms"; readtimeout = "10ms"; writetimeout = "10ms"; pool = { maxidle = 16; maxactive = 64; idletimeout = "300s"; }; }; configFile = pkgs.writeText "docker-registry-config.yml" (builtins.toJSON (registryConfig // cfg.extraConfig)); in { options.services.dockerRegistry = { enable = mkEnableOption "Docker Registry"; listenAddress = mkOption { description = "Docker registry host or ip to bind to."; default = "127.0.0.1"; type = types.str; }; port = mkOption { description = "Docker registry port to bind to."; default = 5000; type = types.int; }; storagePath = mkOption { type = types.path; default = "/var/lib/docker-registry"; description = "Docker registry storage path."; }; enableDelete = mkOption { type = types.bool; default = false; description = "Enable delete for manifests and blobs."; }; enableRedisCache = mkEnableOption "redis as blob cache"; redisUrl = mkOption { type = types.str; default = "localhost:6379"; description = "Set redis host and port."; }; redisPassword = mkOption { type = types.str; default = ""; description = "Set redis password."; }; extraConfig = mkOption { description = '' Docker extra registry configuration via environment variables. ''; default = {}; type = types.attrsOf types.str; }; enableGarbageCollect = mkEnableOption "garbage collect"; garbageCollectDates = mkOption { default = "daily"; type = types.str; description = '' Specification (in the format described by systemd.time 7) of the time at which the garbage collect will occur. ''; }; }; config = mkIf cfg.enable { systemd.services.docker-registry = { description = "Docker Container Registry"; wantedBy = [ "multi-user.target" ]; after = [ "network.target" ]; script = '' ${pkgs.docker-distribution}/bin/registry serve ${configFile} ''; serviceConfig = { User = "docker-registry"; WorkingDirectory = cfg.storagePath; }; }; systemd.services.docker-registry-garbage-collect = { description = "Run Garbage Collection for docker registry"; restartIfChanged = false; unitConfig.X-StopOnRemoval = false; serviceConfig.Type = "oneshot"; script = '' ${pkgs.docker-distribution}/bin/registry garbage-collect ${configFile} ${pkgs.systemd}/bin/systemctl restart docker-registry.service ''; startAt = optional cfg.enableGarbageCollect cfg.garbageCollectDates; }; users.extraUsers.docker-registry = { createHome = true; home = cfg.storagePath; }; }; }