###
# Copyright (c) 2023, John Burwell
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#
#   * Redistributions of source code must retain the above copyright notice,
#     this list of conditions, and the following disclaimer.
#   * Redistributions in binary form must reproduce the above copyright notice,
#     this list of conditions, and the following disclaimer in the
#     documentation and/or other materials provided with the distribution.
#   * Neither the name of the author of this software nor the name of
#     contributors to this software may be used to endorse or promote products
#     derived from this software without specific prior written consent.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.

###

from supybot import conf, registry
try:
    from supybot.i18n import PluginInternationalization
    _ = PluginInternationalization('Chat')
except ImportError:
    # Placeholder that allows the plugin to run without the i18n module
    _ = lambda x: x


def configure(advanced):
    # This will be called by supybot to configure this module.  advanced is
    # a bool that specifies whether the user identified themself as an advanced
    # user or not.  You should effect your configuration by manipulating the
    # registry as appropriate.
    from supybot.questions import expect, anything, something, yn
    conf.registerPlugin('Chat', True)


Chat = conf.registerPlugin('Chat')

# Configuration Parameters

# API key for OpenAI
conf.registerGlobalValue(
    Chat,
    "api_key",
    registry.String(
        "",
        _("""Your ChatGPT API key. This must be set for the plugin to work."""),
        private=True,
    )
)

# Default model to use
conf.registerGlobalValue(
    Chat,
    "model",
    registry.String(
        "gpt-4",
        _("""The OpenAI model to use for generating responses. Default is 'gpt-4'."""),
    )
)

# Maximum tokens for responses
conf.registerGlobalValue(
    Chat,
    "max_tokens",
    registry.Integer(
        256,
        _("""The maximum number of tokens to include in the response. Default is 256."""),
    )
)

# System prompt for the assistant (when using single line)
conf.registerGlobalValue(
    Chat,
    "system_prompt",
    registry.String(
        "You are an IRC bot participating in a channel conversation. Keep your responses brief, informal, and conversational. Avoid overly technical language unless asked, and ensure your tone matches the casual nature of IRC discussions.",
        _("""The system prompt to guide the assistant's behavior. Default provides context about being an IRC bot and guidance on interaction style."""),
    )
)

# System prompt file for the assistant (when using a file instead of single line)
conf.registerGlobalValue(
    Chat,
    "system_prompt_file",
    registry.String(
        "",
        _("""Specifies the name of a file (e.g., 'prompt.txt') in the plugin directory to use as the system prompt for the assistant, instead of the single-line 'system_prompt' setting. If this is set, its contents will take precedence over 'system_prompt' when guiding the assistant's behavior."""),
    )
)

# Number of lines to include from scrollback
conf.registerGlobalValue(
    Chat,
    "scrollback_lines",
    registry.Integer(
        10,
        _("""The number of recent lines from the channel to include as context. Default is 10."""),
    )
)

# String to join multi-line responses
conf.registerGlobalValue(
    Chat,
    "join_string",
    registry.String(
        " / ",
        _("""The string used to join multi-line responses into a single line. Default is ' / '."""),
    )
)

# Passive interaction settings
conf.registerGlobalValue(
    Chat,
    "passive_mode",
    registry.String(
        "off",
        _("""Controls passive participation. Options: off, mention, smart."""),
    )
)

conf.registerGlobalValue(
    Chat,
    "passive_probability",
    registry.Float(
        0.35,
        _("""When passive_mode is 'smart', probability (0-1) that the bot considers chiming in on eligible lines."""),
    )
)

conf.registerGlobalValue(
    Chat,
    "passive_max_replies",
    registry.Integer(
        3,
        _("""Maximum number of passive replies per thread (-1 for no cap)."""),
    )
)

conf.registerGlobalValue(
    Chat,
    "passive_engagement_timeout",
    registry.Integer(
        180,
        _("""Seconds after which an ongoing thread expires if the bot stays silent."""),
    )
)

conf.registerGlobalValue(
    Chat,
    "passive_cooldown",
    registry.Integer(
        120,
        _("""Seconds to stay quiet after ending a passive thread before starting a new one."""),
    )
)

conf.registerGlobalValue(
    Chat,
    "passive_trigger_words",
    registry.SpaceSeparatedListOfStrings(
        [],
        _("""Extra keywords that make the bot more likely to respond in smart mode."""),
    )
)

conf.registerGlobalValue(
    Chat,
    "passive_prompt_addendum",
    registry.String(
        "Stay aware of the channel like a quiet regular. Only jump in when someone addresses $bot_name, when you are already in a thread, or when you are very confident a brief reply will help. Let conversations end naturally if no one follows up.",
        _("""Additional guidance appended to the system prompt when passive mode is enabled."""),
    )
)

# History service integration
conf.registerGlobalValue(
    Chat,
    "history_service_url",
    registry.String(
        "http://127.0.0.1:8901",
        _("""Base URL for the optional history service (e.g. http://127.0.0.1:8901). Leave blank to disable."""),
    )
)

conf.registerGlobalValue(
    Chat,
    "history_service_token",
    registry.String(
        "",
        _("""Bearer token to authenticate with the history service (if required)."""),
        private=True,
    )
)

conf.registerGlobalValue(
    Chat,
    "history_service_timeout",
    registry.Float(
        1.5,
        _("""Timeout in seconds for history service HTTP requests."""),
    )
)

conf.registerGlobalValue(
    Chat,
    "history_include_files",
    registry.Integer(
        2,
        _("""How many rotated log files the history service should scan (passed as include_files parameter)."""),
    )
)

conf.registerGlobalValue(
    Chat,
    "history_result_limit",
    registry.Integer(
        60,
        _("""Maximum number of log lines to request from the history service."""),
    )
)

conf.registerGlobalValue(
    Chat,
    "history_max_chars",
    registry.Integer(
        1800,
        _("""Maximum number of characters of history context to include in the prompt."""),
    )
)

conf.registerGlobalValue(
    Chat,
    "history_max_lines",
    registry.Integer(
        80,
        _("""Maximum number of history lines to include in the prompt."""),
    )
)

conf.registerGlobalValue(
    Chat,
    "history_trigger_words",
    registry.SpaceSeparatedListOfStrings(
        ["remember", "earlier", "history", "logs", "recap", "summary", "yesterday", "before"],
        _("""Words or phrases that should cause the bot to consult the history service."""),
    )
)

# Logging level for the plugin
conf.registerGlobalValue(
    Chat,
    'log_level',
    registry.String(
        'INFO',
        _("""The logging level for the Chat plugin. Options: DEBUG, INFO, WARNING, ERROR, CRITICAL.""")
    )
)

# vim:set shiftwidth=4 tabstop=4 expandtab textwidth=79: