---
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

- include: sahara_install_apt.yml
  when:
    - ansible_pkg_mgr == 'apt'
  tags:
    - sahara-install
    - sahara-apt-packages

- name: Create developer mode constraint file
  copy:
    dest: "/opt/developer-pip-constraints.txt"
    content: |
      {% for item in sahara_developer_constraints %}
      {{ item }}
      {% endfor %}
  when:
    - sahara_developer_mode | bool
  tags:
    - sahara-install
    - sahara-pip-packages

- name: Clone requirements git repository
  git:
    repo: "{{ sahara_requirements_git_repo }}"
    dest: "/opt/requirements"
    clone: yes
    update: yes
    version: "{{ sahara_requirements_git_install_branch }}"
  when:
    - sahara_developer_mode | bool
  tags:
    - sahara-install
    - sahara-pip-packages

- name: Add constraints to pip_install_options fact for developer mode
  set_fact:
    pip_install_options_fact: "{{ pip_install_options|default('') }} --constraint /opt/developer-pip-constraints.txt --constraint /opt/requirements/upper-constraints.txt"
  when:
    - sahara_developer_mode | bool
  tags:
    - sahara-install
    - sahara-pip-packages

- name: Set pip_install_options_fact when not in developer mode
  set_fact:
    pip_install_options_fact: "{{ pip_install_options|default('') }}"
  when:
    - not sahara_developer_mode | bool
  tags:
    - sahara-install
    - sahara-pip-packages

- name: Install required pip packages
  pip:
    name: "{{ sahara_required_pip_packages | join(' ') }}"
    state: present
    extra_args: "{{ pip_install_options_fact }}"
  register: install_packages
  until: install_packages|success
  retries: 5
  delay: 2
  tags:
    - sahara-install
    - sahara-pip-packages

- name: Get local venv checksum
  stat:
    path: "/var/cache/{{ sahara_venv_download_url | basename }}"
    get_md5: False
  when:
    - not sahara_developer_mode | bool
  register: local_venv_stat
  tags:
    - sahara-install
    - sahara-pip-packages

- name: Get remote venv checksum
  uri:
    url: "{{ sahara_venv_download_url | replace('tgz', 'checksum') }}"
    return_content: True
  when:
    - not sahara_developer_mode | bool
  register: remote_venv_checksum
  tags:
    - sahara-install
    - sahara-pip-packages

# TODO: When project moves to ansible 2 we can pass this a sha256sum which will:
#       a) allow us to remove force: yes
#       b) allow the module to calculate the checksum of dest file which would
#          result in file being downloaded only if provided and dest sha256sum
#          checksums differ
- name: Attempt venv download
  get_url:
    url: "{{ sahara_venv_download_url }}"
    dest: "/var/cache/{{ sahara_venv_download_url | basename }}"
    force: yes
  ignore_errors: true
  register: get_venv
  when:
    - not sahara_developer_mode | bool
    - (local_venv_stat.stat.exists == False or
        {{ local_venv_stat.stat.checksum is defined and local_venv_stat.stat.checksum != remote_venv_checksum.content | trim }})
  tags:
    - sahara-install
    - sahara-pip-packages

- name: Set sahara get_venv fact
  set_fact:
    sahara_get_venv: "{{ get_venv }}"
  tags:
    - sahara-install
    - sahara-pip-packages

- name: Remove existing venv
  file:
    path: "{{ sahara_bin | dirname }}"
    state: absent
  when:
    - sahara_get_venv | changed
  tags:
    - sahara-install
    - sahara-pip-packages

- name: Create sahara venv dir
  file:
    path: "{{ sahara_bin | dirname }}"
    state: directory
  register: sahara_venv_dir
  tags:
    - sahara-install
    - sahara-pip-packages

- name: Unarchive pre-built venv
  unarchive:
    src: "/var/cache/{{ sahara_venv_download_url | basename }}"
    dest: "{{ sahara_bin | dirname }}"
    copy: "no"
  when:
    - not sahara_developer_mode | bool
    - sahara_get_venv | changed or sahara_venv_dir | changed
  notify:
    - Restart sahara services
  tags:
    - sahara-install
    - sahara-pip-packages

- name: Install pip packages (venv)
  pip:
    name: "{{ sahara_pip_packages | join(' ') }}"
    state: present
    virtualenv: "{{ sahara_bin | dirname }}"
    virtualenv_site_packages: "no"
    extra_args: "{{ pip_install_options_fact }}"
  register: install_packages
  until: install_packages|success
  retries: 5
  delay: 2
  when:
    - sahara_get_venv | failed or sahara_developer_mode | bool
  notify:
    - Restart sahara services
  tags:
    - sahara-install
    - sahara-pip-packages

- name: Update virtualenv path
  command: >
    virtualenv-tools --update-path=auto {{ sahara_bin | dirname }}
  when:
    - not sahara_developer_mode | bool
    - sahara_get_venv | success
  tags:
    - sahara-install
    - sahara-pip-packages