Function update_vm_env

scripts/setup_slack.py:124–186

Update config/vm.env with Slack tokens, bot name, and slash command.

(
    bot_token: str, app_token: str, bot_name: str = "", slash_command: str = ""
)

def update_vm_env(
    bot_token: str, app_token: str, bot_name: str = "", slash_command: str = ""
) -> bool:
    """Update config/vm.env with Slack tokens, bot name, and slash command."""
    try:
        # Read existing content
        if VM_ENV_PATH.exists():
            with open(VM_ENV_PATH, "r") as f:
                content = f.read()
        else:
            content = ""

        # Update or add SLACK_BOT_TOKEN
        if "SLACK_BOT_TOKEN=" in content:
            content = re.sub(
                r"SLACK_BOT_TOKEN=.*",
                f"SLACK_BOT_TOKEN={bot_token}",
                content,
            )
        else:
            content += f"\nSLACK_BOT_TOKEN={bot_token}"

        # Update or add SLACK_APP_TOKEN
        if "SLACK_APP_TOKEN=" in content:
            content = re.sub(
                r"SLACK_APP_TOKEN=.*",
                f"SLACK_APP_TOKEN={app_token}",
                content,
            )
        else:
            content += f"\nSLACK_APP_TOKEN={app_token}"

        # Update or add SLACK_BOT_NAME
        if bot_name:
            if "SLACK_BOT_NAME=" in content:
                content = re.sub(
                    r"SLACK_BOT_NAME=.*",
                    f"SLACK_BOT_NAME={bot_name}",
                    content,
                )
            else:
                content += f"\nSLACK_BOT_NAME={bot_name}"

        # Update or add SLACK_SLASH_COMMAND
        if slash_command:
            if "SLACK_SLASH_COMMAND=" in content:
                content = re.sub(
                    r"SLACK_SLASH_COMMAND=.*",
                    f"SLACK_SLASH_COMMAND={slash_command}",
                    content,
                )
            else:
                content += f"\nSLACK_SLASH_COMMAND={slash_command}"

        # Write back
        with open(VM_ENV_PATH, "w") as f:
            f.write(content)