aboutsummaryrefslogtreecommitdiffstats
path: root/resources/libraries/bash/function/ansible.sh
blob: 6cf4d16f43024c46406bb9f28822e621da0f529c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
#!/usr/bin/env bash

# Copyright (c) 2024 Cisco and/or its affiliates.
# 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.

set -exuo pipefail


function ansible_adhoc () {

    # Run ansible ad-hoc command module on hosts in working topology file.
    #
    # Variable read:
    # - ${WORKING_TOPOLOGY} - Reserved working topology.
    # - ${CSIT_DIR} - CSIT main directory, where ansible playbooks are located.
    # - ${FLAVOR} - Node flavor string, see common.sh

    set -exuo pipefail

    case "$FLAVOR" in
        "aws" | "c6in" | "c6gn" | "c7gn")
            INVENTORY_PATH="cloud_inventory"
            ;;
        "x-"*)
            INVENTORY_PATH="external_inventory"
            ;;
        *)
            INVENTORY_PATH="lf_inventory"
            ;;
    esac

    if ! installed sshpass; then
        die "Please install sshpass!"
    fi

    hosts=($(fgrep host "${WORKING_TOPOLOGY}" | cut -d ":" -f 2)) || {
        die "Failed to read hosts from working topology!"
    }
    pushd "${CSIT_DIR}"/fdio.infra.ansible || die "Pushd failed!"
    export ANSIBLE_HOST_KEY_CHECKING=False
    export ANSIBLE_STDOUT_CALLBACK=yaml
    export ANSIBLE_PIPELINING=true
    ansible-playbook \
        --vault-password-file=vault_pass \
        --extra-vars '@vault.yml' \
        --inventory inventories/$INVENTORY_PATH/hosts site.yaml \
        --limit "$(echo ${hosts[@]//\"})" \
        --module-name shell \
        --args \"$(echo $@)\" || die "Failed to run ansible on host!"
    popd || die "Popd failed!"
}

function ansible_playbook () {

    # Run ansible playbook on hosts in working topology file. Ansible scope is
    # determined by tags passed as parameters to this function.
    #
    # Variable read:
    # - ${WORKING_TOPOLOGY} - Reserved working topology.
    # - ${CSIT_DIR} - CSIT main directory, where ansible playbooks are located.
    # - ${FLAVOR} - Node flavor string, see common.sh

    set -exuo pipefail

    case "$FLAVOR" in
        "aws" | "c6in" | "c6gn" | "c7gn")
            INVENTORY_PATH="cloud_inventory"
            ;;
        "x-"*)
            INVENTORY_PATH="external_inventory"
            ;;
        *)
            INVENTORY_PATH="lf_inventory"
            ;;
    esac

    if ! installed sshpass; then
        die "Please install sshpass!"
    fi

    hosts=($(fgrep host "${WORKING_TOPOLOGY}" | cut -d ":" -f 2)) || {
        die "Failed to read hosts from working topology!"
    }
    pushd "${CSIT_DIR}"/fdio.infra.ansible || die "Pushd failed!"
    export ANSIBLE_HOST_KEY_CHECKING=False
    export ANSIBLE_STDOUT_CALLBACK=yaml
    export ANSIBLE_PIPELINING=true
    ansible-playbook \
        --vault-password-file=vault_pass \
        --extra-vars '@vault.yml' \
        --inventory inventories/$INVENTORY_PATH/hosts site.yaml \
        --limit "$(echo ${hosts[@]//\"})" \
        --tags "$(echo $@)" || die "Failed to run ansible on host!"
    popd || die "Popd failed!"
}


function installed () {

    # Check if the given utility is installed. Fail if not installed.
    #
    # Arguments:
    # - ${1} - Utility to check.
    # Returns (implicitly):
    # - 0 - If command is installed.
    # - 1 - If command is not installed.

    set -exuo pipefail

    command -v "${1}"
}