/*
 * trigger-command.c - trigger command
 *
 * Copyright (C) 2014 Sébastien Helleu <flashcode@flashtux.org>
 *
 * This file is part of WeeChat, the extensible chat client.
 *
 * WeeChat is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3 of the License, or
 * (at your option) any later version.
 *
 * WeeChat is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with WeeChat.  If not, see <http://www.gnu.org/licenses/>.
 */

#include <stdlib.h>
#include <stdio.h>
#include <string.h>

#include "../weechat-plugin.h"
#include "trigger.h"
#include "trigger-buffer.h"
#include "trigger-config.h"


/*
 * Set "enabled" value in a trigger.
 *
 * Argument "enable" can be:
 *   -1: toggle trigger
 *    0: disable trigger
 *    1: enable trigger
 */

void
trigger_command_set_enabled (struct t_trigger *trigger, int enable)
{
    if (enable < 0)
    {
        enable = weechat_config_boolean (trigger->options[TRIGGER_OPTION_ENABLED]) ?
            0 : 1;
    }
    weechat_config_option_set (trigger->options[TRIGGER_OPTION_ENABLED],
                               (enable) ? "on" : "off", 1);
    weechat_printf_tags (NULL, "no_trigger",
                         (enable) ?
                         _("Trigger \"%s\" enabled") :
                         _("Trigger \"%s\" disabled"),
                         trigger->name);
}

/*
 * Renames a trigger and displays error if the rename is not possible.
 *
 * This function is called by commands:
 *   /trigger set OLD name NEW
 *   /trigger rename OLD NEW
 */

void
trigger_command_rename (struct t_trigger *trigger, const char *new_name)
{
    char *name, *name2;

    name = strdup (trigger->name);
    name2 = weechat_string_remove_quotes (new_name, "'\"");

    if (name && name2)
    {
        /* check that new name is valid */
        if (!trigger_name_valid (name2))
        {
            weechat_printf_tags (NULL, "no_trigger",
                                 _("%sError: invalid name for trigger"),
                                 weechat_prefix ("error"));
            goto end;
        }
        /* check that no trigger already exists with the new name */
        if (trigger_search (name2))
        {
            weechat_printf_tags (NULL, "no_trigger",
                                 _("%sError: trigger \"%s\" already "
                                   "exists"),
                                 weechat_prefix ("error"), name2);
            goto end;
        }
        /* rename the trigger */
        if (trigger_rename (trigger, name2))
        {
            weechat_printf_tags (NULL, "no_trigger",
                                 _("Trigger \"%s\" renamed to \"%s\""),
                                 name, trigger->name);
        }
        else
        {
            weechat_printf_tags (NULL, "no_trigger",
                                 _("%sError: failed to rename trigger "
                                   "\"%s\""),
                                 weechat_prefix ("error"), name);
        }
    }

end:
    if (name)
        free (name);
    if (name2)
        free (name2);
}

/*
 * Callback for command "/trigger": manage triggers.
 */

int
trigger_command_trigger (void *data, struct t_gui_buffer *buffer, int argc,
                         char **argv, char **argv_eol)
{
    struct t_trigger *ptr_trigger;
    const char *option;
    char *value, **sargv;
    int i, type, count, index_option, enable, sargc;

    /* make C compiler happy */
    (void) data;
    (void) buffer;

    sargv = NULL;

    /* list all triggers */
    if ((argc == 1)
        || ((argc == 2) && (weechat_strcasecmp (argv[1], "list") == 0)))
    {
        if (triggers)
        {
            weechat_printf_tags (NULL, "no_trigger", "");
            weechat_printf_tags (NULL, "no_trigger", _("List of triggers:"));
            for (ptr_trigger = triggers; ptr_trigger;
                 ptr_trigger = ptr_trigger->next_trigger)
            {
                weechat_printf_tags (NULL, "no_trigger",
                                     "  %s: %s, \"%s\" (%d hooks, %d/%d) %s",
                                     ptr_trigger->name,
                                     weechat_config_string (ptr_trigger->options[TRIGGER_OPTION_HOOK]),
                                     weechat_config_string (ptr_trigger->options[TRIGGER_OPTION_ARGUMENTS]),
                                     ptr_trigger->hooks_count,
                                     ptr_trigger->hook_count_cb,
                                     ptr_trigger->hook_count_cmd,
                                     weechat_config_boolean (ptr_trigger->options[TRIGGER_OPTION_ENABLED]) ?
                                     "" : _("(disabled)"));
                option = weechat_config_string (ptr_trigger->options[TRIGGER_OPTION_CONDITIONS]);
                if (option && option[0])
                {
                    weechat_printf_tags (NULL, "no_trigger",
                                         "      conditions: \"%s\"", option);
                }
                for (i = 0; i < ptr_trigger->regex_count; i++)
                {
                    weechat_printf_tags (NULL, "no_trigger",
                                         "      regex %d%s%s%s: "
                                         "%s\"%s%s%s\" --> \"%s%s%s\"",
                                         i + 1,
                                         (ptr_trigger->regex[i].variable) ? " (" : "",
                                         (ptr_trigger->regex[i].variable) ? ptr_trigger->regex[i].variable : "",
                                         (ptr_trigger->regex[i].variable) ? ")" : "",
                                         weechat_color ("chat_delimiters"),
                                         weechat_color (weechat_config_string (trigger_config_color_regex)),
                                         ptr_trigger->regex[i].str_regex,
                                         weechat_color ("chat_delimiters"),
                                         weechat_color (weechat_config_string (trigger_config_color_replace)),
                                         ptr_trigger->regex[i].replace,
                                         weechat_color ("chat_delimiters"));
                }
                option = weechat_config_string (ptr_trigger->options[TRIGGER_OPTION_COMMAND]);
                if (option && option[0])
                {
                    weechat_printf_tags (NULL, "no_trigger",
                                         "      command: \"%s\"", option);
                }
            }
        }
        else
        {
            weechat_printf_tags (NULL, "no_trigger", _("No trigger defined"));
        }
        goto end;
    }

    /* add a trigger */
    if (weechat_strcasecmp (argv[1], "add") == 0)
    {
        sargv = weechat_string_split_shell (argv_eol[2], &sargc);
        if (!sargv || (sargc < 2))
        {
            weechat_printf_tags (NULL, "no_trigger",
                                 _("%sError: missing arguments for \"%s\" "
                                   "command"),
                                 weechat_prefix ("error"), "trigger");
            goto end;
        }
        if (!trigger_name_valid (sargv[0]))
        {
            weechat_printf_tags (NULL, "no_trigger",
                                 _("%sError: invalid name for trigger"),
                                 weechat_prefix ("error"));
            goto end;
        }
        type = trigger_search_hook_type (sargv[1]);
        if (type < 0)
        {
            weechat_printf_tags (NULL, "no_trigger",
                                 _("%sError: invalid hook type \"%s\""),
                                 weechat_prefix ("error"), sargv[1]);
            goto end;
        }
        if ((sargc > 6) && (trigger_search_return_code (sargv[6]) < 0))
        {
            weechat_printf_tags (NULL, "no_trigger",
                                 _("%sError: invalid return code \"%s\""),
                                 weechat_prefix ("error"), sargv[6]);
            goto end;
        }
        ptr_trigger = trigger_alloc (sargv[0]);
        if (!ptr_trigger)
        {
            weechat_printf_tags (NULL, "no_trigger",
                                 _("%sError: failed to create trigger \"%s\""),
                                 weechat_prefix ("error"), sargv[0]);
            goto end;
        }
        ptr_trigger = trigger_new (
            sargv[0],                      /* name */
            "on",                          /* enabled */
            sargv[1],                      /* hook */
            (sargc > 2) ? sargv[2] : "",   /* arguments */
            (sargc > 3) ? sargv[3] : "",   /* conditions */
            (sargc > 4) ? sargv[4] : "",   /* regex */
            (sargc > 5) ? sargv[5] : "",   /* command */
            (sargc > 6) ? sargv[6] : "");  /* return code */
        if (ptr_trigger)
        {
            weechat_printf_tags (NULL, "no_trigger",
                                 _("Trigger \"%s\" created"), sargv[0]);
        }
        else
        {
            weechat_printf_tags (NULL, "no_trigger",
                                 _("%sError: failed to create trigger \"%s\""),
                                 weechat_prefix ("error"), sargv[0]);
        }
        goto end;
    }

    /* set option in a trigger */
    if (weechat_strcasecmp (argv[1], "set") == 0)
    {
        if (argc < 5)
        {
            weechat_printf_tags (NULL, "no_trigger",
                                 _("%sError: missing arguments for \"%s\" "
                                   "command"),
                                 weechat_prefix ("error"), "trigger");
            goto end;
        }
        ptr_trigger = trigger_search (argv[2]);
        if (!ptr_trigger)
        {
            weechat_printf_tags (NULL, "no_trigger",
                                 _("%sError: trigger \"%s\" not found"),
                                 weechat_prefix ("error"), argv[2]);
            goto end;
        }
        if (weechat_strcasecmp (argv[3], "name") == 0)
        {
            trigger_command_rename (ptr_trigger, argv[4]);
            goto end;
        }
        value = weechat_string_remove_quotes (argv_eol[4], "'\"");
        if (value)
        {
            index_option = trigger_search_option (argv[3]);
            if (index_option >= 0)
            {
                weechat_config_option_set (ptr_trigger->options[index_option],
                                           value, 1);
                weechat_printf_tags (NULL, "no_trigger",
                                     _("Trigger \"%s\" updated"),
                                     ptr_trigger->name);
            }
            else
            {
                weechat_printf_tags (NULL, "no_trigger",
                                     _("%sError: trigger option \"%s\" not "
                                       "found"),
                                     weechat_prefix ("error"), argv[3]);
            }
            free (value);
        }
        goto end;
    }

    /* rename a trigger */
    if (weechat_strcasecmp (argv[1], "rename") == 0)
    {
        if (argc < 4)
        {
            weechat_printf_tags (NULL, "no_trigger",
                                 _("%sError: missing arguments for \"%s\" "
                                   "command"),
                                 weechat_prefix ("error"), "trigger");
            goto end;
        }
        ptr_trigger = trigger_search (argv[2]);
        if (!ptr_trigger)
        {
            weechat_printf_tags (NULL, "no_trigger",
                                 _("%sError: trigger \"%s\" not found"),
                                 weechat_prefix ("error"), argv[2]);
            goto end;
        }
        trigger_command_rename (ptr_trigger, argv[3]);
        goto end;
    }

    /* enable/disable/toggle trigger(s) */
    if ((weechat_strcasecmp (argv[1], "enable") == 0)
        || (weechat_strcasecmp (argv[1], "disable") == 0)
        || (weechat_strcasecmp (argv[1], "toggle") == 0))
    {
        if (argc < 3)
        {
            weechat_printf_tags (NULL, "no_trigger",
                                 _("%sError: missing arguments for \"%s\" "
                                   "command"),
                                 weechat_prefix ("error"), "trigger");
            goto end;
        }
        enable = -1;
        if (weechat_strcasecmp (argv[1], "enable") == 0)
            enable = 1;
        else if (weechat_strcasecmp (argv[1], "disable") == 0)
            enable = 0;
        if (weechat_strcasecmp (argv[2], "-all") == 0)
        {
            for (ptr_trigger = triggers; ptr_trigger;
                 ptr_trigger = ptr_trigger->next_trigger)
            {
                trigger_command_set_enabled (ptr_trigger, enable);
            }
        }
        else
        {
            for (i = 2; i < argc; i++)
            {
                ptr_trigger = trigger_search (argv[i]);
                if (ptr_trigger)
                    trigger_command_set_enabled (ptr_trigger, enable);
                else
                {
                    weechat_printf_tags (NULL, "no_trigger",
                                         _("%sTrigger \"%s\" not found"),
                                         weechat_prefix ("error"), argv[i]);
                }
            }
        }
        goto end;
    }

    /* delete trigger(s) */
    if (weechat_strcasecmp (argv[1], "del") == 0)
    {
        if (argc < 3)
        {
            weechat_printf_tags (NULL, "no_trigger",
                                 _("%sError: missing arguments for \"%s\" "
                                   "command"),
                                 weechat_prefix ("error"), "trigger");
            goto end;
        }
        if (weechat_strcasecmp (argv[2], "-all") == 0)
        {
            count = triggers_count;
            trigger_free_all ();
            if (count > 0)
                weechat_printf_tags (NULL, "no_trigger",
                                     _("%d triggers removed"), count);
        }
        else
        {
            for (i = 2; i < argc; i++)
            {
                ptr_trigger = trigger_search (argv[i]);
                if (ptr_trigger)
                {
                    trigger_free (ptr_trigger);
                    weechat_printf_tags (NULL, "no_trigger",
                                         _("Trigger \"%s\" removed"), argv[i]);
                }
                else
                {
                    weechat_printf_tags (NULL, "no_trigger",
                                         _("%sTrigger \"%s\" not found"),
                                         weechat_prefix ("error"), argv[i]);
                }
            }
        }
        goto end;
    }

    /* open the trigger monitor buffer */
    if (weechat_strcasecmp (argv[1], "monitor") == 0)
    {
        trigger_buffer_open (1);
        goto end;
    }

end:
    if (sargv)
        weechat_string_free_split (sargv);

    return WEECHAT_RC_OK;
}

/*
 * Hooks trigger commands.
 */

void
trigger_command_init ()
{
    weechat_hook_command (
        "trigger",
        N_("manage triggers"),
        N_("list"
           " || add <name> <hook> [\"<arguments>\" [\"<conditions>\" "
           "[\"<regex>\" [\"<command>\" [\"<return_code>\"]]]]]"
           " || set <name> <option> <value>"
           " || rename <name> <new_name>"
           " || enable|disable|toggle <name>|-all [<name>...]"
           " || del <name>|-all [<name>...]"
           " || monitor"),
        N_("      add: add a trigger\n"
           "     name: name of trigger\n"
           "     hook: signal, hsignal, modifier, print, timer\n"
           "arguments: arguments for the hook, depending on hook (separated by "
           "semicolons):\n"
           "           signal: name(s) of signal\n"
           "           hsignal: name(s) of hsignal\n"
           "           modifier: name(s) of modifier\n"
           "           print: buffer, tags, message, strip_colors\n"
           "           timer: interval, align_second, max_calls\n"
           "      set: set an option in a trigger\n"
           "   option: name of option: name, hook, arguments, conditions, regex, "
           "command, return_code\n"
           "           (for help on option, you can do /help "
           "trigger.trigger.<name>.<option>)\n"
           "    value: new value for the option\n"
           "   rename: rename a trigger\n"
           "   enable: enable trigger(s)\n"
           "  disable: disable trigger(s)\n"
           "   toggle: toggle trigger(s)\n"
           "      del: delete a trigger\n"
           "     -all: do action on all triggers\n"
           "  monitor: open the trigger monitor buffer\n"
           "\n"
           "When a trigger callback is called, following actions are performed, "
           "in this order:\n"
           "  1. if no regex/command is defined, exit\n"
           "  2. check conditions; if false, exit\n"
           "  3. replace text using regex (if defined in trigger)\n"
           "  4. execute command (if defined in trigger)\n"
           "Note: on steps 1 and 2, the exit is made with the return code "
           "defined in trigger (or NULL for a modifier).\n"
           "\n"
           "Examples:\n"
           "  send alert (BEL) on highlight or private message:\n"
           "    /trigger add beep print \"\" \"${tg_highlight} || ${tg_msg_pv}\" "
           "\"\" \"/print -stderr \\a\"\n"
           "  replace password with '*' in /oper command (in command line and "
           "command history):\n"
           "    /trigger add oper modifier input_text_display;history_add "
           "\"\" \"==^(/oper +\\S+ +)(.*)==\\1\\*2\"\n"
           "  add text attributes in *bold*, _underline_ and /italic/:\n"
           "    /trigger add effects modifier weechat_print \"\" "
           "\"==\\*(\\S+)\\*==*${color:bold}\\1${color:-bold}*== "
           "==_(\\S+)_==_${color:underline}\\1${color:-underline}_== "
           "==/(\\S+)/==/${color:italic}\\1${color:-italic}/\"\n"
           "  silently save config each hour:\n"
           "    /trigger add cfgsave timer 3600000;0;0 \"\" \"\" \"/mute /save\""),
        "list"
        " || add %(trigger_names) %(trigger_hooks)"
        " || set %(trigger_names) %(trigger_options)|name %(trigger_option_value)"
        " || rename %(trigger_names) %(trigger_names)"
        " || enable|disable|toggle|del %(trigger_names)|-all %(trigger_names)|%*"
        " || monitor",
        &trigger_command_trigger, NULL);
}