+++ /dev/null
-<?php
-/**
- * @file
- * Add system notifications as a new drush option.
- */
-
-/**
- * @todo there are no hooks fired after a command errors out.
- */
-register_shutdown_function('drush_notify_shutdown');
-
-/**
- * Implements hook_drush_help_alter().
- */
-function notify_drush_help_alter(&$command) {
- if ($command['command'] == 'global-options') {
- // Do not include these in options in standard help.
- if ($command['#brief'] === FALSE) {
- $command['options']['notify'] = array(
- 'description' => 'Use system notifications to signal command completion. If set to a number, commands that finish in fewer seconds will not trigger a notification.',
- 'example-value' => 60,
- 'never-propagate' => TRUE,
- );
- $command['options']['notify-audio'] = array(
- 'description' => 'Trigger an audio alert to signal command completion. If set to a number, commands that finish in fewer seconds will not trigger a notification.',
- 'example-value' => 60,
- 'never-propagate' => TRUE,
- );
- $command['sub-options']['notify']['notify-cmd'] = array(
- 'description' => 'Specify the shell command to trigger the notification.',
- 'never-propagate' => TRUE,
- );
- $command['sub-options']['notify']['notify-cmd-audio'] = array(
- 'description' => 'Specify the shell command to trigger the audio notification.',
- 'never-propagate' => TRUE,
- );
- }
- }
-}
-
-/**
- * Implements hook_drush_help().
- */
-function notify_drush_help($section) {
- switch ($section) {
- case 'notify:cache-clear':
- return dt('Caches have been cleared.');
- case 'notify:site-install:error':
- return dt('Failed on site installation');
- }
-}
-
-/**
- * Shutdown function to signal on errors.
- */
-function drush_notify_shutdown() {
- $cmd = drush_get_command();
-
- if (empty($cmd['command'])) {
- return;
- }
-
- // pm-download handles its own notification.
- if ($cmd['command'] != 'pm-download' && drush_notify_allowed($cmd['command'])) {
- $msg = dt("Command '!command' completed.", array('!command' => $cmd['command']));
- drush_notify_send(drush_notify_command_message($cmd['command'], $msg));
- }
-
- if (drush_get_option('notify', FALSE) && drush_get_error()) {
- // If the only error is that notify failed, do not try to notify again.
- $log = drush_get_error_log();
- if (count($log) == 1 && array_key_exists('NOTIFY_COMMAND_NOT_FOUND', $log)) {
- return;
- }
-
- // Send an alert that the command failed.
- if (drush_notify_allowed($cmd['command'])) {
- $msg = dt("Command '!command' failed.", array('!command' => $cmd['command']));
- drush_notify_send(drush_notify_command_message($cmd['command'] . ':error', $msg));
- }
- }
-}
-
-/**
- * Determine the message to send on command completion.
- *
- * @param string $command
- * Name of the Drush command for which we check message overrides.
- * @param string $default
- * (Default: NULL) Default message to use if there are not notification message overrides.
- *
- * @return string
- * Message to use for notification.
- */
-function drush_notify_command_message($command, $default = NULL) {
- if ($msg = drush_command_invoke_all('drush_help', 'notify:' . $command)) {
- $msg = implode("\n", $msg);
- }
- else {
- $msg = $default ? $default : $msg = $command . ': No news is good news.';
- }
-
- return $msg;
-}
-
-/**
- * Prepares and dispatches notifications to delivery mechanisms.
- *
- * You may avoid routing a message to secondary messaging mechanisms (e.g. audio),
- * by direct use of the delivery functions.
- *
- * @param string $msg
- * Message to send via notification.
- */
-function drush_notify_send($msg) {
- drush_notify_send_text($msg);
- if (drush_get_option('notify-audio', FALSE)) {
- drush_notify_send_audio($msg);
- }
-}
-
-/**
- * Send text-based system notification.
- *
- * This is the automatic, default behavior. It is intended for use with tools
- * such as libnotify in Linux and Notification Center on OSX.
- *
- * @param string $msg
- * Message text for delivery.
- *
- * @return bool
- * TRUE on success, FALSE on failure
- */
-function drush_notify_send_text($msg) {
- $override = drush_get_option('notify-cmd', FALSE);
-
- if (!empty($override)) {
- $cmd = $override;
- }
- else {
- switch (PHP_OS) {
- case 'Darwin':
- $cmd = 'terminal-notifier -message %s -title Drush';
- $error_message = dt('terminal-notifier command failed. Please install it from https://github.com/alloy/terminal-notifier.');
- break;
- case 'Linux':
- default:
- $icon = drush_normalize_path(DRUSH_BASE_PATH . '/drush_logo-black.png');
- $cmd = "notify-send %s -i $icon";
- $error_message = dt('notify-send command failed. Please install it as per http://coderstalk.blogspot.com/2010/02/how-to-install-notify-send-in-ubuntu.html.');
- break;
- }
- }
-
- if (!drush_shell_exec($cmd, $msg)) {
- return drush_set_error('NOTIFY_COMMAND_NOT_FOUND', $error_message . ' ' . dt('Or you may specify an alternate command to run by specifying --notify-cmd=<my_command>'));
- }
-
- return TRUE;
-}
-
-/**
- * Send an audio-based system notification.
- *
- * This function is only automatically invoked with the additional use of the
- * --notify-audio flag or configuration state.
- *
- * @param $msg
- * Message for audio recital.
- *
- * @return bool
- * TRUE on success, FALSE on failure
- */
-function drush_notify_send_audio($msg) {
- $override = drush_get_option('notify-cmd-audio', FALSE);
-
- if (!empty($override)) {
- $cmd = $override;
- }
- else {
- switch (PHP_OS) {
- case 'Darwin':
- $cmd = 'say %s';
- break;
- case 'Linux':
- default:
- $cmd = drush_get_option('notify-cmd-audio', 'spd-say') . ' %s';
- }
- }
-
- if (!drush_shell_exec($cmd, $msg)) {
- return drush_set_error('NOTIFY_COMMAND_NOT_FOUND', dt('The third party notification utility failed.'));
- }
-}
-
-/**
- * Identify if the given Drush request should trigger a notification.
- *
- * @param $command
- * Name of the command.
- *
- * @return
- * Boolean
- */
-function drush_notify_allowed($command) {
- $notify = drush_get_option(array('notify', 'notify-audio'), FALSE);
- $execution = time() - $_SERVER['REQUEST_TIME'];
-
- return ($notify === TRUE ||
- (is_numeric($notify) && $notify > 0 && $execution > $notify));
-}
-