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: 122: 123: 124: 125: 126: 127: 128: 129: 130: 131: 132: 133: 134: 135: 136: 137: 138: 139: 140: 141: 142: 143: 144: 145: 146: 147: 148: 149: 150: 151: 152: 153: 154: 155: 156: 157: 158: 159: 160: 161: 162: 163: 164: 165: 166: 167: 168: 169: 170: 171: 172: 173: 174: 175: 176: 177: 178: 179: 180: 181: 182: 183: 184: 185: 186: 187: 188: 189: 190: 191: 192: 193: 194: 195: 196: 197: 198: 199: 200: 201: 202: 203: 204: 205: 206: 207: 208: 209: 210: 211: 212: 213: 214: 215: 216: 217: 218: 219: 220: 221: 222: 223: 224: 225: 226: 227: 228: 229: 230: 231: 232: 233: 234: 235: 236: 237: 238: 239: 240: 241: 242: 243: 244: 245: 246: 247: 248: 249: 250: 251: 252: 253: 254: 255: 256: 257: 258: 259: 260: 261: 262: 263: 264: 265: 266: 267: 268: 269: 270: 271: 272: 273: 274: 275: 276: 277: 278: 279: 280: 281: 282: 283: 284: 285: 286: 287: 288: 289: 290: 291: 292: 293: 294: 295: 296: 297: 298: 299: 300: 301: 302: 303: 304: 305: 306: 307: 308: 309: 310: 311: 312: 313: 314: 315: 316: 317: 318: 319: 320: 321: 322: 323: 324: 325: 326: 327: 328: 329: 330: 331: 332: 333: 334: 335: 336: 337: 338: 339: 340: 341: 342: 343: 344: 345: 346: 347: 348: 349: 350: 351: 352: 353: 354: 355: 356: 357: 358: 359: 360: 361: 362: 363: 364: 365: 366: 367: 368: 369: 370: 371: 372: 373: 374: 375: 376: 377: 378: 379: 380: 381: 382: 383: 384: 385: 386: 387: 388: 389: 390: 391: 392: 393: 394: 395: 396: 397: 398: 399: 400: 401: 402: 403: 404: 405: 406: 407: 408: 409: 410: 411: 412: 413: 414: 415: 416: 417: 418: 419: 420: 421: 422: 423: 424: 425: 426: 427: 428: 429: 430: 431: 432: 433: 434: 435: 436: 437: 438: 439: 440: 441: 442: 443: 444: 445: 446: 447: 448: 449: 450: 451: 452: 453: 454: 455: 456: 457: 458: 459:
<?php declare(strict_types=1);
class Argos_Module extends Module_Skeleton
{
const DEFAULT_BACKEND = 'default';
protected $exportedFunctions = ['*' => PRIVILEGE_ADMIN];
public function status(string $service = null): array
{
if (!IS_CLI) {
return $this->query('argos_status', $service);
}
$ret = \Util_Process_Safe::exec('monit -B status %s', $service);
if (!$ret['success']) {
error('Failed to query status: %s', $ret['stderr']);
return [];
}
$status = $this->filterStatus($ret['stdout']);
return $service ? $status[$service] ?? [] : $status;
}
public function restart(string $service): bool
{
if (!IS_CLI) {
return $this->query('argos_restart', $service);
}
$ret = Util_Process_Safe::exec('monit restart %s', $service);
return $ret['success'] ?: error('Failed to restart %s service: %s ', $service, $ret['stderr']);
}
public function unmonitor(string $service): bool
{
if (!IS_CLI) {
return $this->query('argos_unmonitor', $service);
}
$ret = \Util_Process_Safe::exec('monit unmonitor %s', $service);
return $ret['success'] ?: error('Failed to stop %s monitoring: %s ', $service, $ret['stderr']);
}
public function monitor(string $service): bool
{
if (!IS_CLI) {
return $this->query('argos_monitor', $service);
}
$ret = \Util_Process_Safe::exec('monit monitor %s', $service);
return $ret['success'] ?: error('Failed to start %s monitoring: %s ', $service, $ret['stderr']);
}
public function list_monitored(): array
{
$ret = \Util_Process::exec('monit summary -B');
if (!$ret['success']) {
error('Failed to query monit');
return [];
}
if (!preg_match_all('/^\s(?!Service Name)(\S+)/m', $ret['stdout'], $matches, PREG_SET_ORDER)) {
return [];
}
return array_column($matches, 1);
}
public function headcount(bool $down = true): int
{
if (!IS_CLI) {
return $this->query('argos_headcount', $down);
}
$ret = \Util_Process_Safe::exec('monit report ' . ($down ? 'down' : 'up'));
return $ret['success'] ? (int)$ret['stdout'] : (int)error('Failed to query Monit: %s', $ret['stderr']);
}
public function validate_all(): array
{
if (!IS_CLI) {
return $this->query('argos_validate_all');
}
$ret = \Util_Process_Safe::exec('monit validate', [0, 1]);
if (!$ret['success']) {
error('Failed to validate services: %s ', $ret['stdout']);
return [];
}
return $this->filterStatus($ret['stdout']);
}
private function filterStatus(string $response): array
{
$serviceBuckets = preg_split('/^(?=Process|Filesystem|Program|System\s+)/m', $response);
array_shift($serviceBuckets);
$services = [];
foreach ($serviceBuckets as $bucket) {
if (!preg_match_all(Regex::ARGOS_SERVICE_STATUS, $bucket, $matches, PREG_SET_ORDER)) {
continue;
}
$processName = null;
$fields = [
'timing' => null
];
foreach ($matches as $m) {
if ($m['proc']) {
$processName = $m['proc'];
$fields['type'] = strtolower($m['type']);
continue;
}
$var = str_replace(' ', '_', $m['name']);
$value = $m['value'];
switch ($var) {
case 'inodes_free':
$value = strtok($value, ' ');
case 'pid':
case 'parent_pid':
case 'uid':
case 'gid':
case 'threads':
case 'children':
case 'effective_uid':
case 'last_exit_value':
case 'inodes_total':
$value = (int)$value;
break;
case 'unix_socket_response_time':
case 'port_response_time':
$fields['timing'] = (float)strtok($value, ' ');
break;
case 'cpu':
case 'cpu_total':
if (false === strpos($value, ' ')) {
$value = (float)$value / 100;
}
break;
case 'memory':
case 'memory_total':
$fields[$var . '_raw'] = (int)Formatter::changeBytes(str_replace(' ', '', substr($value, strpos($value, '[') + 1, -1)));
break;
case 'disk_write':
case 'disk_read':
$fields[$var . '_raw'] = (int)Formatter::changeBytes(str_replace(' ', '',
substr($value, $pos = strpos($value, '[') + 1, strrpos($value, ' ') - $pos)));
$fields[$var . '_bw_raw'] = (int)Formatter::changeBytes(str_replace(' ', '',
substr($value, 0, strpos($value, '/'))));
break;
case 'block_size':
$value = (int)Formatter::changeBytes($value, 'B');
break;
case 'read':
case 'write':
$fields[$var . '_bw_raw'] = (int)Formatter::changeBytes(
str_replace(' ', '', substr($value, 0, strpos($value, '/')))
);
$fields[$var . '_iops_raw'] = (float)strtok(substr(
$value,
$pos = strpos($value, ',')+1
), ' ');
break;
case 'data_collected':
$value = DateTime::createFromFormat('D, d M Y H:i:s', $value,
new DateTimeZone(TIMEZONE))->getTimestamp();
break;
}
$fields[$var] = $value;
$fields['failed'] = $fields['status'] !== 'OK';
$fields['monitored'] = 0 !== strcasecmp($fields['status'], 'not monitored');
}
$services[$processName] = $fields;
}
return $services;
}
public function config(string $backend, ?array $newparams)
{
deprecated_func('use config_relay');
return $this->config_relay($backend, $newparams);
}
public function config_relay(string $backend, ?array $newparams)
{
if (!IS_CLI) {
return $this->query('argos_config_relay', $backend, $newparams);
}
if (!\in_array($backend, \Opcenter\Argos\Config::get()->getBackends(), true)) {
return error("Unknown backend `%s'", $backend);
}
if (null === $newparams) {
$provider = array_get(\Opcenter\Argos\Config::get()->backend($backend), 'backend', $backend);
if (!\Opcenter\Argos\Config::get()->deleteBackend($backend)) {
warn("Failed to delete backend `%s'", $backend);
}
if (!\Opcenter\Argos\Config::get()->createBackend($provider, $backend)) {
return error("Failed to create backend `%s'", $backend);
}
return true;
}
$cfg = \Opcenter\Argos\Config::get();
$backend = $cfg->backend($backend);
foreach ($newparams as $k => $v) {
$backend[$k] = $v;
}
return true;
}
public function get_config_relay(string $backend, $param = null): ?array
{
if (!IS_CLI) {
return $this->query('argos_get_config_relay', $backend);
}
if (!\in_array($backend, \Opcenter\Argos\Config::get()->getBackends(), true)) {
error("Unknown backend `%s'", $backend);
return null;
}
$cfg = \Opcenter\Argos\Config::get()->backend($backend)->toArray();
return $param ? array_get($cfg, $param, null) : $cfg;
}
public function set_default_relay($backend)
{
if (!IS_CLI) {
return $this->query('argos_set_default_relay', $backend);
}
$backends = $this->get_backends();
foreach ((array)$backend as $b) {
if (!\in_array($b, $backends, true)) {
return error("Invalid backend `%s'", $b);
}
}
return \Opcenter\Argos\Config::get()->setDefault($backend);
}
public function get_backends(): ?array
{
if (!IS_CLI) {
return $this->query('argos_get_backends');
}
if (!($cfg = \Opcenter\Argos\Config::get())) {
return null;
}
return $cfg->getBackends();
}
public function create_backend(string $name, string $driver): bool
{
if (!IS_CLI) {
return $this->query('argos_create_backend', $name, $driver);
}
if (\in_array($name, $this->get_backends(), true)) {
return error("Backend `%s' already exists", $name);
}
if (!\in_array($driver, $this->get_backend_relays(), true)) {
return error("Invalid backend relay `%s'. Use get_backend_relays() to view all", $driver);
}
$conf = \Opcenter\Argos\Config::get();
$conf->createBackend($driver, $name);
$conf->sync();
return true;
}
public function get_backend_relays(): array
{
if (!IS_CLI) {
return $this->query('argos_get_backend_relays');
}
return \Opcenter\Argos\Backend::getBackends();
}
public function test(string $backend = null)
{
return $this->send('Argos test alert', $backend, '💯 test');
}
public function send(string $msg, string $backend = null, string $title = null)
{
if (!IS_CLI) {
return $this->query('argos_send', $msg, $backend, $title);
}
if (!file_exists(\Opcenter\Argos\Config::CONFIGURATION_FILE)) {
return error(
"%s is missing - run argos.init Scope first. See Monitoring.md in docs/",
\Opcenter\Argos\Config::CONFIGURATION_FILE
);
}
if ($title) {
$title = '-t ' . escapeshellarg($title);
}
if ($backend) {
$backend = '-b ' . escapeshellarg($backend);
}
return array_get(
\Util_Process_Safe::exec('ntfy -c %(config)s ' . $title . ' ' . $backend . ' send %(msg)s',
[
'config' => \Opcenter\Argos\Config::CONFIGURATION_FILE,
'msg' => $msg,
]
),
'success',
false
);
}
public function active(): bool
{
return array_get(\Util_Process::exec('systemctl is-active monit', null, [0, 3]), 'return', null) === 0;
}
}