]> git.immae.eu Git - perso/Immae/Projets/Puppet.git/blob - modules/role/manifests/cryptoportfolio/bot.pp
2f01c98ab6735b59bdaa0267cadf6781c017b747
[perso/Immae/Projets/Puppet.git] / 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