---
- name: Install MariaDB server and client
  apt:
    name:
      - mariadb-server
      - mariadb-client
    state: present
    update_cache: yes

- name: Ensure PyMySQL is installed
  apt:
    name: python3-pymysql
    state: present

- name: Ensure MariaDB is running and enabled
  service:
    name: mariadb
    state: started
    enabled: true

- name: Try to connect to MariaDB with root password
  shell: |
    mysql -u root -p'{{ mariadb_root_password }}' -e "SELECT 1;"
  register: mysql_root_status
  failed_when: false
  changed_when: false

- name: Set MariaDB root password if not already set
  mysql_user:
    name: root
    host: "{{ item }}"
    password: "{{ mariadb_root_password }}"
    login_unix_socket: /run/mysqld/mysqld.sock
    check_implicit_admin: true
    state: present
  loop:
    - localhost
    - 127.0.0.1
    - ::1
  when: mysql_root_status.rc != 0

- debug:
    msg: "MariaDB root password is already set, skipping reset"
  when: mysql_root_status.rc == 0