docker-compose.yaml 6.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113
  1. {% set tpl = ix_lib.base.render.Render(values) %}
  2. {% set perm_container = tpl.deps.perms(values.consts.perms_container_name) %}
  3. {% set perms_config = {"uid": values.run_as.user, "gid": values.run_as.group, "mode": "check"} %}
  4. {% set pg_config = {
  5. "user": values.consts.db_user,
  6. "password": values.immich.db_password,
  7. "database": values.consts.db_name,
  8. "volume": values.storage.postgres_data,
  9. } %}
  10. {% set postgres = tpl.deps.postgres(values.consts.pgvecto_container_name, "pgvecto_image", pg_config, perm_container) %}
  11. {% do postgres.container.environment.add_env("DB_STORAGE_TYPE", values.immich.db_storage_type) %}
  12. {% do postgres.container.set_shm_size_mb(128) %}
  13. {% set redis_config = {
  14. "password": values.immich.redis_password,
  15. "volume": {"type": "temporary", "volume_config": {"volume_name": "redis-data"}},
  16. } %}
  17. {% set redis = tpl.deps.redis(values.consts.redis_container_name, "redis_image", redis_config, perm_container) %}
  18. {% set ml_container = namespace(x=None) %}
  19. {% set server_container = tpl.add_container(values.consts.server_container_name, "image") %}
  20. {% do server_container.set_user(values.run_as.user, values.run_as.group) %}
  21. {% do server_container.healthcheck.set_custom_test("immich-healthcheck") %}
  22. {% do server_container.depends.add_dependency(values.consts.pgvecto_container_name, "service_healthy") %}
  23. {% do server_container.depends.add_dependency(values.consts.redis_container_name, "service_healthy") %}
  24. {% if values.storage.use_old_storage_config %}
  25. {% set is_install = values.get("ix_context", {}).get("is_install", False) %}
  26. {% if is_install %}{% do tpl.funcs.fail("Old storage configuration is not supported in new installations.") %}{% endif %}
  27. {% do tpl.notes.add_deprecation(values.consts.old_storage_warning) %}
  28. {% do server_container.add_storage("%s/upload"|format(values.consts.base_path), values.storage.uploads) %}
  29. {% do perm_container.add_or_skip_action("uploads", values.storage.uploads, perms_config) %}
  30. {% do server_container.add_storage("%s/thumbs"|format(values.consts.base_path), values.storage.thumbs) %}
  31. {% do perm_container.add_or_skip_action("thumbs", values.storage.thumbs, perms_config) %}
  32. {% do server_container.add_storage("%s/library"|format(values.consts.base_path), values.storage.library) %}
  33. {% do perm_container.add_or_skip_action("library", values.storage.library, perms_config) %}
  34. {% do server_container.add_storage("%s/profile"|format(values.consts.base_path), values.storage.profile) %}
  35. {% do perm_container.add_or_skip_action("profile", values.storage.profile, perms_config) %}
  36. {% do server_container.add_storage("%s/backups"|format(values.consts.base_path), values.storage.backups) %}
  37. {% do perm_container.add_or_skip_action("backups", values.storage.backups, perms_config) %}
  38. {% do server_container.add_storage("%s/encoded-video"|format(values.consts.base_path), values.storage.video) %}
  39. {% do perm_container.add_or_skip_action("video", values.storage.video, perms_config) %}
  40. {% else %}
  41. {% do server_container.add_storage(values.consts.base_path, values.storage.data) %}
  42. {% do perm_container.add_or_skip_action("data", values.storage.data, perms_config) %}
  43. {% endif %}
  44. {% for store in values.storage.additional_storage %}
  45. {% do server_container.add_storage(store.mount_path, store) %}
  46. {% do perm_container.add_or_skip_action(store.mount_path, store, perms_config) %}
  47. {% endfor %}
  48. {% do server_container.add_port(values.network.web_port) %}
  49. {% do server_container.environment.add_env("NODE_ENV", "production") %}
  50. {% do server_container.environment.add_env("IMMICH_LOG_LEVEL", values.immich.log_level) %}
  51. {% do server_container.environment.add_env("IMMICH_PORT", values.network.web_port.port_number) %}
  52. {% do server_container.environment.add_env("DB_USERNAME", values.consts.db_user) %}
  53. {% do server_container.environment.add_env("DB_PASSWORD", values.immich.db_password) %}
  54. {% do server_container.environment.add_env("DB_HOSTNAME", values.consts.pgvecto_container_name) %}
  55. {% do server_container.environment.add_env("DB_DATABASE_NAME", values.consts.db_name) %}
  56. {% do server_container.environment.add_env("DB_PORT", "5432") %}
  57. {% do server_container.environment.add_env("REDIS_HOSTNAME", values.consts.redis_container_name) %}
  58. {% do server_container.environment.add_env("REDIS_PASSWORD", values.immich.redis_password) %}
  59. {% do server_container.environment.add_env("REDIS_PORT", "6379") %}
  60. {% do server_container.environment.add_env("REDIS_DBINDEX", "0") %}
  61. {% do server_container.environment.add_user_envs(values.immich.additional_envs) %}
  62. {% if values.immich.enable_ml %}
  63. {% set ml_container.x = tpl.add_container(values.consts.ml_container_name, values.immich.ml_image_selector) %}
  64. {% do ml_container.x.set_user(values.run_as.user, values.run_as.group) %}
  65. {% do ml_container.x.healthcheck.set_custom_test("python3 healthcheck.py") %}
  66. {% do ml_container.x.environment.add_env("NODE_ENV", "production") %}
  67. {% do ml_container.x.environment.add_env("IMMICH_LOG_LEVEL", values.immich.log_level) %}
  68. {% do ml_container.x.environment.add_env("IMMICH_PORT", values.consts.ml_port) %}
  69. {% do ml_container.x.environment.add_env("MACHINE_LEARNING_CACHE_FOLDER", values.consts.ml_cache_path) %}
  70. {% do ml_container.x.environment.add_env("TRANSFORMERS_CACHE", values.consts.ml_cache_path) %}
  71. {% do ml_container.x.environment.add_env("HF_HOME", values.consts.ml_cache_path) %}
  72. {% do ml_container.x.environment.add_env("MPLCONFIGDIR", "%s/matplotlib"|format(values.consts.ml_cache_path)) %}
  73. {% if values.immich.hugging_face_endpoint %}
  74. {% do ml_container.x.environment.add_env("HF_ENDPOINT", values.immich.hugging_face_endpoint) %}
  75. {% endif %}
  76. {% do ml_container.x.environment.add_user_envs(values.immich.additional_envs) %}
  77. {% set ml_cache_config = dict(values.storage.ml_cache, **{"volume_config": {"volume_name": "ml-cache"}}) %}
  78. {% do ml_container.x.add_storage(values.consts.ml_cache_path, ml_cache_config) %}
  79. {% do perm_container.add_or_skip_action("ml-cache", ml_cache_config, perms_config) %}
  80. {% do server_container.environment.add_env("IMMICH_MACHINE_LEARNING_ENABLED", "true") %}
  81. {% do server_container.environment.add_env("IMMICH_MACHINE_LEARNING_URL", "http://%s:%d"|format(values.consts.ml_container_name, values.consts.ml_port)) %}
  82. {% endif %}
  83. {% if perm_container.has_actions() %}
  84. {% do perm_container.activate() %}
  85. {% do postgres.add_dependency(values.consts.perms_container_name, "service_completed_successfully") %}
  86. {% do redis.container.depends.add_dependency(values.consts.perms_container_name, "service_completed_successfully") %}
  87. {% do server_container.depends.add_dependency(values.consts.perms_container_name, "service_completed_successfully") %}
  88. {% if values.immich.enable_ml %}
  89. {% do ml_container.x.depends.add_dependency(values.consts.perms_container_name, "service_completed_successfully") %}
  90. {% endif %}
  91. {% endif %}
  92. {% do tpl.portals.add(values.network.web_port) %}
  93. {{ tpl.render() | tojson }}