modules/role/manifests/cryptoportfolio/bot.pp

   1 class role::cryptoportfolio::bot inherits role::cryptoportfolio {
   2   $password_seed = lookup("base_installation::puppet_pass_seed")
   3
   4   $cf_bot_app = "${home}/bot"
   5   $cf_bot_app_conf = "${home}/bot_config.ini"
   6   $cf_bot_app_conf_hourly = "${home}/bot_config_hourly.ini"
   7   $cf_bot_app_reports = "${home}/bot_reports"
   8
   9   ensure_packages(["python", "python-pip"])
  10
  11   file { $cf_bot_app:
  12     ensure  => "directory",
  13     mode    => "0700",
  14     owner   => $user,
  15     group   => $group,
  16     require => User["$user:"],
  17   }
  18
  19   archive { "${home}/trader_${trader_version}.tar.gz":
  20     path          => "${home}/trader_${trader_version}.tar.gz",
  21     source        => "https://git.immae.eu/releases/cryptoportfolio/trader/trader_${trader_version}.tar.gz",
  22     checksum_type => "sha256",
  23     checksum      => $trader_sha256,
  24     cleanup       => false,
  25     extract       => true,
  26     user          => $user,
  27     username      => lookup("base_installation::ldap_cn"),
  28     password      => generate_password(24, $password_seed, "ldap"),
  29     extract_path  => $cf_bot_app,
  30     require       => [User["$user:"], File[$cf_bot_app]],
  31   } ~>
  32   exec { "py-cryptoportfolio-dependencies":
  33     cwd         => $cf_bot_app,
  34     user        => $user,
  35     environment => ["HOME=${home}"],
  36     command     => "/usr/bin/make install",
  37     require     => User["$user:"],
  38     refreshonly => true,
  39     before      => [
  40       File[$cf_bot_app_conf],
  41       Cron::Job::Multiple["py-cryptoportfolio"],
  42     ]
  43   }
  44
  45   $pg_password = generate_password(24, $password_seed, "postgres_cryptoportfolio")
  46   file { $cf_bot_app_conf:
  47     owner   => $user,
  48     group   => $group,
  49     mode    => "0600",
  50     content => template("role/cryptoportfolio/bot_config.ini.erb"),
  51     require => [
  52       User["$user:"],
  53       Archive["${home}/trader_${trader_version}.tar.gz"],
  54     ],
  55   }
  56
  57   cron::job::multiple { "py-cryptoportfolio":
  58     ensure      => present,
  59     environment => ["HOME=${home}","PATH=/usr/bin/","CRON_TZ=UTC"],
  60     require     => [
  61       File[$cf_bot_app_conf],
  62       Archive["${home}/trader_${trader_version}.tar.gz"]
  63     ],
  64     jobs        => [
  65       {
  66         command     => "cd $cf_bot_app ; python main.py --quiet --config $cf_bot_app_conf --before",
  67         user        => $user,
  68         weekday     => 7, # Sunday
  69         hour        => 22,
  70         minute      => 30,
  71         description => "Run before the cryptoportfolio update",
  72       },
  73       {
  74         command     => "cd $cf_bot_app ; python main.py --quiet --config $cf_bot_app_conf --after",
  75         user        => $user,
  76         weekday     => 1, # Monday
  77         hour        => 0,
  78         minute      => 0,
  79         description => "Run after the cryptoportfolio update",
  80       }
  81     ],
  82   }
  83
  84   if versioncmp($trader_version, "v1.3") >= 0 {
  85     file { $cf_bot_app_conf_hourly:
  86       owner   => $user,
  87       group   => $group,
  88       mode    => "0600",
  89       content => template("role/cryptoportfolio/bot_config_hourly.ini.erb"),
  90       require => [
  91         User["$user:"],
  92         Archive["${home}/trader_${trader_version}.tar.gz"],
  93       ],
  94     }
  95
  96     cron::job::multiple { "py-cryptoportfolio-hourly":
  97       ensure      => present,
  98       environment => ["HOME=${home}","PATH=/usr/bin/"],
  99       require     => [
 100         File[$cf_bot_app_conf_hourly],
 101         Archive["${home}/trader_${trader_version}.tar.gz"]
 102       ],
 103       jobs        => [
 104         {
 105           command     => "cd $cf_bot_app ; python main.py --config $cf_bot_app_conf_hourly",
 106           user        => $user,
 107           minute      => 30,
 108           description => "Print the current state to redis",
 109         },
 110       ],
 111     }
 112   }
 113
 114   unless empty($webhook_url) {
 115     exec { "bot-slack-notify":
 116       refreshonly => true,
 117       environment => [
 118         "P_PROJECT=Trader",
 119         "P_WEBHOOK=${webhook_url}",
 120         "P_VERSION=${trader_version}",
 121         "P_HOST=${web_host}",
 122         "P_HTTPS=${web_ssl}",
 123       ],
 124       command     => "/usr/local/bin/slack-notify",
 125       require     => File["/usr/local/bin/slack-notify"],
 126       subscribe   => Exec["py-cryptoportfolio-dependencies"],
 127     }
 128   }
 129 }
 130