# update_immich.yml

- name: Update Immich on VM via Proxmox
  hosts: linux_servers
  gather_facts: false
  become: true
  become_user: root
  become_method: sudo

  vars:
    # VM connection (provided by Semaphore env vars)
    vm_ip:   "{{ lookup('env', 'VM_IP') }}"
    vm_user: "{{ lookup('env', 'VM_USER') }}"
    vm_pass: "{{ lookup('env', 'VM_PASS') }}"
    use_sudo: false

    # --- Debug mode (controlled via Semaphore variable) ---
    DEBUG: "{{ lookup('env', 'DEBUG') | default(0) | int }}"
    RETRIES: "{{ lookup('env', 'RETRIES') | default(25) | int }}"

    # Immich specifics
    immich_dir: "/opt/immich"
    immich_project: "immich"
    immich_compose_url: "https://github.com/immich-app/immich/releases/latest/download/docker-compose.yml"
    immich_compose_file: "/opt/immich/docker-compose.yml"
    immich_override_file: "/opt/immich/docker-compose.override.yml"
    immich_port: 2283

    # Optional external URL for controller-side readiness check (e.g., https://photos.example.com)
    immich_url: "{{ lookup('env', 'IMMICH_URL') | default('', true) }}"

    # Retry policy
    immich_retries: "{{ RETRIES }}"
    immich_delay: 2

    # Docker command prefix (consistent behavior)
    docker_prefix: "unalias docker 2>/dev/null || true; DOCKER_CLI_HINTS=0; command docker"

    # Compose command (always include override to keep local mounts separate from upstream compose)
    immich_compose_cmd: >-
      {{ docker_prefix }} compose
      -p {{ immich_project }}
      -f {{ immich_compose_file }}
      -f {{ immich_override_file }}

    # Commands to run on the target VM
    immich_commands:
      - "cd {{ immich_dir }}"

      - |
        cd {{ immich_dir }}
        mkdir -p backups
        if [ -f docker-compose.yml ]; then
          cp -a docker-compose.yml "backups/docker-compose.yml.$(date +%F_%H%M%S).bak"
        fi
        if [ -f .env ]; then
          cp -a .env "backups/.env.$(date +%F_%H%M%S).bak"
        fi
        if [ -f docker-compose.override.yml ]; then
          cp -a docker-compose.override.yml "backups/docker-compose.override.yml.$(date +%F_%H%M%S).bak"
        fi

      - |
        cd {{ immich_dir }}
        # Download latest compose from Immich releases (requires curl or wget)
        if command -v curl >/dev/null 2>&1; then
          curl -fsSL -o docker-compose.yml "{{ immich_compose_url }}"
        elif command -v wget >/dev/null 2>&1; then
          wget -qO docker-compose.yml "{{ immich_compose_url }}"
        else
          echo "Neither curl nor wget is available on the VM."
          exit 1
        fi

      - |
        cd {{ immich_dir }}
        # Create override file if missing (keep local mounts here so upstream compose can be replaced safely)
        python3 - <<'PY'
        # Create docker-compose.override.yml only if missing
        from pathlib import Path
        import textwrap

        p = Path("docker-compose.override.yml")
        if p.exists():
            raise SystemExit(0)

        content = textwrap.dedent("""\
        # Local overrides for Immich (kept separate from upstream docker-compose.yml).
        # If you want read-only mounts, append ':ro' to the right side.
        services:
          immich-server:
            volumes:
              - /mnt/nextcloud-howard-photos:/mnt/nextcloud-howard-photos
              - /mnt/nextcloud-kamilkaprdelka-photos:/mnt/nextcloud-kamilkaprdelka-photos
        """)
        p.write_text(content, encoding="utf-8")
        PY

      - |
        cd {{ immich_dir }}
        # Ensure .env exists. If missing, try to reconstruct it from running containers to avoid breaking DB creds.
        python3 - <<'PY'
        import json
        import subprocess
        from pathlib import Path

        env_path = Path(".env")
        if env_path.exists():
            raise SystemExit(0)

        def run(cmd):
            p = subprocess.run(cmd, capture_output=True, text=True)
            return p.returncode, p.stdout, p.stderr

        rc, out, err = run(["bash", "-lc", "command docker inspect immich_postgres immich_server"])
        if rc != 0 or not out.strip():
            print("ERROR: .env is missing and cannot inspect running containers (immich_postgres/immich_server).", flush=True)
            print("Create /opt/immich/.env manually or ensure the containers exist.", flush=True)
            raise SystemExit(1)

        data = json.loads(out)

        by_name = {}
        for c in data:
            name = (c.get("Name") or "").lstrip("/")
            by_name[name] = c

        pg = by_name.get("immich_postgres")
        srv = by_name.get("immich_server")
        if not pg or not srv:
            print("ERROR: Could not find immich_postgres and immich_server in docker inspect output.", flush=True)
            raise SystemExit(1)

        def env_map(container):
            m = {}
            for kv in (container.get("Config", {}).get("Env") or []):
                if "=" in kv:
                    k, v = kv.split("=", 1)
                    m[k] = v
            return m

        def find_mount_source(container, dest):
            for m in (container.get("Mounts") or []):
                if m.get("Destination") == dest:
                    return m.get("Source")
            return ""

        pg_env = env_map(pg)
        db_user = pg_env.get("POSTGRES_USER", "")
        db_pass = pg_env.get("POSTGRES_PASSWORD", "")
        db_name = pg_env.get("POSTGRES_DB", "")

        db_data = find_mount_source(pg, "/var/lib/postgresql/data")
        upload_loc = find_mount_source(srv, "/usr/src/app/upload")

        # Try to preserve the currently used image tag as IMMICH_VERSION (optional but safer)
        immich_version = ""
        image = (srv.get("Config", {}).get("Image") or "")
        if ":" in image and "@" not in image:
            immich_version = image.rsplit(":", 1)[-1]
        elif ":" in image and "@" in image:
            # image like repo:tag@sha256:...
            immich_version = image.split("@", 1)[0].rsplit(":", 1)[-1]

        missing = []
        for k, v in [
            ("DB_USERNAME", db_user),
            ("DB_PASSWORD", db_pass),
            ("DB_DATABASE_NAME", db_name),
            ("DB_DATA_LOCATION", db_data),
            ("UPLOAD_LOCATION", upload_loc),
        ]:
            if not v:
                missing.append(k)

        if missing:
            print("ERROR: Could not reconstruct these .env values from containers: " + ", ".join(missing), flush=True)
            raise SystemExit(1)

        lines = [
            f"UPLOAD_LOCATION={upload_loc}",
            f"DB_DATA_LOCATION={db_data}",
            f"DB_USERNAME={db_user}",
            f"DB_PASSWORD={db_pass}",
            f"DB_DATABASE_NAME={db_name}",
        ]
        if immich_version:
            lines.append(f"IMMICH_VERSION={immich_version}")

        env_path.write_text("\n".join(lines) + "\n", encoding="utf-8")
        print("Created .env from running containers.", flush=True)
        PY

      - |
        cd {{ immich_dir }}
        # Comment out healthcheck.start_interval if present (safe no-op if missing)
        sed -i -E 's/^([[:space:]]*)start_interval:/\1# start_interval:/' docker-compose.yml || true

      - "cd {{ immich_dir }} && {{ immich_compose_cmd }} config >/dev/null"
      - "cd {{ immich_dir }} && {{ immich_compose_cmd }} pull >/dev/null"
      - "cd {{ immich_dir }} && {{ immich_compose_cmd }} up -d --remove-orphans >/dev/null"

  tasks:
    - name: Ensure sshpass is installed (for password-based SSH)  # English comments
      ansible.builtin.apt:
        name: sshpass
        state: present
        update_cache: yes

    - name: Run Immich update commands on VM (via SSH)  # use SSHPASS env, hide item label
      ansible.builtin.command:
        argv:
          - sshpass
          - -e
          - ssh
          - -o
          - StrictHostKeyChecking=no
          - -o
          - ConnectTimeout=15
          - "{{ vm_user }}@{{ vm_ip }}"
          - bash
          - -lc
          - "{{ ('sudo ' if use_sudo else '') + item }}"
      environment:
        SSHPASS: "{{ vm_pass }}"
      loop: "{{ immich_commands }}"
      loop_control:
        index_var: idx
        label: "cmd-{{ idx }}"
      register: immich_cmds
      changed_when: false
      no_log: "{{ DEBUG == 0 }}"
      run_once: true

    - name: Show outputs for each Immich command
      ansible.builtin.debug:
        msg: |
          CMD: {{ item.item }}
          RC:  {{ item.rc }}
          STDOUT:
          {{ (item.stdout | default('')).strip() }}
          STDERR:
          {{ (item.stderr | default('')).strip() }}
      loop: "{{ immich_cmds.results }}"
      when: DEBUG == 1
      run_once: true

    - name: Fail play if any Immich command failed
      ansible.builtin.assert:
        that: "item.rc == 0"
        fail_msg: "Immich update failed on VM: {{ item.item }} (rc={{ item.rc }})"
        success_msg: "All Immich update commands succeeded."
      loop: "{{ immich_cmds.results }}"
      loop_control:
        index_var: idx
        label: "cmd-{{ idx }}"
      run_once: true

    # -------------------------
    # Readiness checks (controller first, then VM fallback)
    # -------------------------

    - name: Immich | Wait for API ping (controller first, with retries)
      ansible.builtin.uri:
        url: "{{ (immich_url | regex_replace('/$','')) + '/api/server/ping' }}"
        method: GET
        return_content: true
        validate_certs: "{{ (immich_url | default('')) is match('^https://') }}"
        status_code: 200
      register: immich_controller
      delegate_to: localhost
      run_once: true
      when: immich_url is defined and (immich_url | length) > 0
      retries: "{{ immich_retries }}"
      delay: "{{ immich_delay }}"
      until: immich_controller.status == 200 and ('pong' in (immich_controller.content | default('')))
      failed_when: false
      changed_when: false

    - name: Immich | VM-side ping (JSON via Python, with retries)
      ansible.builtin.command:
        argv:
          - sshpass
          - -e
          - ssh
          - -o
          - StrictHostKeyChecking=no
          - -o
          - ConnectTimeout=15
          - "{{ vm_user }}@{{ vm_ip }}"
          - bash
          - -lc
          - |
            python3 - <<'PY'
            # Ping Immich API from localhost and print response to stdout
            import urllib.request, sys
            try:
                with urllib.request.urlopen("http://127.0.0.1:{{ immich_port }}/api/server/ping", timeout=15) as r:
                    sys.stdout.write(r.read().decode(errors='ignore'))
            except Exception:
                pass
            PY
      environment:
        SSHPASS: "{{ vm_pass }}"
      register: immich_vm
      changed_when: false
      failed_when: false
      when: immich_controller.status | default(0) != 200
      retries: "{{ immich_retries }}"
      delay: "{{ immich_delay }}"
      until: (immich_vm.stdout | default('') | trim | length) > 0 and ('pong' in (immich_vm.stdout | default('')))
      no_log: "{{ DEBUG == 0 }}"
      run_once: true

    - name: Immich | Print concise summary
      ansible.builtin.debug:
        msg: >-
          Immich API ping {{ 'OK' if (('pong' in (immich_controller.content|default(''))) or ('pong' in (immich_vm.stdout|default('')))) else 'NOT OK' }}.
          Source={{ 'controller' if (immich_controller.status|default(0))==200 else 'vm' if (immich_vm.stdout|default('')|trim|length>0) else 'n/a' }}.
      when: DEBUG == 1
      run_once: true