aboutsummaryrefslogtreecommitdiff
path: root/lib/hydra/config
blob: 0f64b4c047e913e450e8d8718e43cd474f094ee0 (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
#!/bin/bash

# Setup main configuration and load preferences
function hydra_config_load {
  local folder="`dirname $CONFIG`"

  if [ -f "$folder" ]; then
    echo "Converting legacy configuration scheme..."
    mv $folder $folder.tmp
    mkdir -p $folder
    mv $folder.tmp $CONFIG
  fi

  if [ ! -e "$CONFIG" ]; then
    echo "Creating $CONFIG..."
    mkdir `dirname $CONFIG`
    touch $CONFIG
    chmod 600 $CONFIG
    echo "# Hydra config file." > $CONFIG
    echo "" >> $CONFIG
  fi

  hydra_config_load_preferences
}

# Load config preferences
function hydra_config_load_preferences {
  # Load custom preferences
  if [ ! -z "$PREFERENCES" ] && [ -f "$PREFERENCES" ]; then
    source $PREFERENCES
  fi
  
  hydra_check_preferences
}

# Check preferences
function hydra_check_preferences {
  if [ ! -z "$PUPPET" ] && [ ! -d "$PUPPET" ]; then
    echo "Puppet folder not found: $PUPPET."
    exit 1
  fi

  if [ -z "$PUPPET_KEYS" ]; then
    PUPPET_KEYS="$PUPPET/modules/site-keys"
  fi

  export HYDRA_CONNECT="ssh -o ConnectTimeout=15"
}

# Load a parameter from config
function hydra_config {
  if [ -z "$CONFIG" ]; then
    echo "Your have to set CONFIG variable in the code"
    exit 1
  elif [ -e "$CONFIG" ]; then
    grep -e "^$1=" $CONFIG | tail -n 1 | cut -d = -f 2 | sed -e 's/"//g' -e "s/'//g" | sed -e 's/ *#.*$//'
  else
    echo "Config file not found: $CONFIG"
    exit 1
  fi
}