---
- name: Expand Ubuntu root disk in vCenter and inside guest
  hosts: all
  gather_facts: yes
  become: yes

  vars:
    disk_controller_number: 0
    disk_unit_number: 0
    root_disk: /dev/sda
    root_partition: /dev/sda3
    root_partition_number: 3
    root_lv: /dev/mapper/ubuntu--vg-ubuntu--lv

  tasks:
    - name: Get VM info from vCenter
      community.vmware.vmware_guest_info:
        hostname: "{{ vcenter_hostname }}"
        username: "{{ vcenter_username }}"
        password: "{{ vcenter_password }}"
        validate_certs: false
        datacenter: "{{ vcenter_datacenter }}"
        name: "{{ vm_name }}"
      delegate_to: localhost
      register: vm_info

    - name: Get current disk size in GB
      ansible.builtin.set_fact:
        current_disk_gb: >-
          {{
            (
              vm_info.instance.hw_disk
              | selectattr('controller_number', 'equalto', disk_controller_number)
              | selectattr('unit_number', 'equalto', disk_unit_number)
              | map(attribute='capacity_in_kb')
              | first
            ) | int // 1024 // 1024
          }}

    - name: Calculate target disk size
      ansible.builtin.set_fact:
        target_disk_gb: "{{ current_disk_gb + (increase_gb | int) }}"

    - name: Show resize plan
      ansible.builtin.debug:
        msg: "Expanding {{ vm_name }} from {{ current_disk_gb }} GB to {{ target_disk_gb }} GB"

    - name: Expand disk in vCenter
      community.vmware.vmware_guest_disk:
        hostname: "{{ vcenter_hostname }}"
        username: "{{ vcenter_username }}"
        password: "{{ vcenter_password }}"
        validate_certs: false
        datacenter: "{{ vcenter_datacenter }}"
        name: "{{ vm_name }}"
        disk:
          - state: present
            unit_number: "{{ disk_unit_number }}"
            controller_number: "{{ disk_controller_number }}"
            size_gb: "{{ target_disk_gb }}"
      delegate_to: localhost

    - name: Rescan SCSI bus
      ansible.builtin.shell: |
        for host in /sys/class/scsi_host/host*; do
          echo "- - -" > "$host/scan"
        done
      args:
        executable: /bin/bash
      changed_when: true

    - name: Wait after rescan
      ansible.builtin.pause:
        seconds: 5

    - name: Ensure growpart is installed
      ansible.builtin.apt:
        name: cloud-guest-utils
        state: present
        update_cache: yes

    - name: Expand partition if needed
      ansible.builtin.command: "growpart {{ root_disk }} {{ root_partition_number }}"
      register: growpart_result
      changed_when: "'CHANGED:' in growpart_result.stdout"
      failed_when: >
        growpart_result.rc != 0 and
        'NOCHANGE:' not in growpart_result.stdout

    - name: Resize LVM physical volume
      ansible.builtin.command: "pvresize {{ root_partition }}"
      register: pvresize_result
      changed_when: "'changed' in (pvresize_result.stdout | lower) or 'resized' in (pvresize_result.stdout | lower)"
      failed_when: false

    - name: Extend root logical volume and filesystem
      ansible.builtin.command: "lvextend -r -l +100%FREE {{ root_lv }}"
      register: lvextend_result
      changed_when: "'Size of logical volume' in lvextend_result.stdout or 'Logical volume' in lvextend_result.stdout"
      failed_when: false

    - name: Show volume state
      ansible.builtin.shell: |
        pvs
        vgs
        lvs
        df -h /
      args:
        executable: /bin/bash
      register: final_state
      changed_when: false

    - name: Print final state
      ansible.builtin.debug:
        var: final_state.stdout_lines