import discord from discord.ext import commands from discord import app_commands from config import Config SCRIPT_TEMPLATES: dict[str, dict[str, str]] = { "python": { "description": "Python script with argparse and logging", "extension": ".py", "template": """#!/usr/bin/env python3 \"\"\" {description} Usage: python {name}.py [options] \"\"\" import argparse import logging import sys logging.basicConfig( level=logging.INFO, format="%(asctime)s [%(levelname)s] %(message)s", ) log = logging.getLogger("{name}") def parse_args() -> argparse.Namespace: parser = argparse.ArgumentParser(description="{description}") parser.add_argument( "-v", "--verbose", action="store_true", help="Enable debug logging", ) return parser.parse_args() def main() -> int: args = parse_args() if args.verbose: logging.getLogger().setLevel(logging.DEBUG) log.info("Script started — add your logic here") # --- CUSTOM CODE BELOW --- return 0 if __name__ == "__main__": sys.exit(main()) """, }, "bash": { "description": "Bash script with usage, error handling, and color output", "extension": ".sh", "template": """#!/usr/bin/env bash set -euo pipefail # {description} # Usage: ./{name}.sh [options] SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)" RED='\\033[0;31m' GREEN='\\033[0;32m' YELLOW='\\033[1;33m' NC='\\033[0m' usage() { cat <&2; } VERBOSE=false while [[ $# -gt 0 ]]; do case "$1" in -h|--help) usage ;; -v|--verbose) VERBOSE=true; shift ;; *) error "Unknown option: $1"; usage ;; esac done info "Script started — add your logic here" # --- CUSTOM CODE BELOW --- """, }, "javascript": { "description": "Node.js script with argument parsing and error handling", "extension": ".js", "template": """#!/usr/bin/env node /** * {description} * Usage: node {name}.js [options] */ const process = require("node:process"); function parseArgs() { const args = { verbose: false, }; for (let i = 2; i < process.argv.length; i++) { switch (process.argv[i]) { case "-h": case "--help": console.log(`Usage: node ${process.argv[1].split("/").pop()} [options]`); console.log(`\\n{description}\\n`); console.log("Options:"); console.log(" -h, --help Show this help"); console.log(" -v, --verbose Verbose output"); process.exit(0); case "-v": case "--verbose": args.verbose = true; break; default: console.error(`Unknown option: ${process.argv[i]}`); process.exit(1); } } return args; } function main() { const args = parseArgs(); if (args.verbose) console.log("[DEBUG] Verbose mode enabled"); console.log("[INFO] Script started — add your logic here"); // --- CUSTOM CODE BELOW --- } main(); """, }, "powershell": { "description": "PowerShell script with advanced functions and parameter binding", "extension": ".ps1", "template": """<# .SYNOPSIS {description} .DESCRIPTION {description} .PARAMETER Verbose Enable verbose output #> param( [Parameter(Mandatory = $false)] [switch]$Verbose ) function Write-Info { param([string]$Message) Write-Host "[INFO] $Message" -ForegroundColor Green } function Write-Warn { param([string]$Message) Write-Host "[WARN] $Message" -ForegroundColor Yellow } function Write-Error { param([string]$Message) Write-Host "[ERROR] $Message" -ForegroundColor Red } if ($Verbose) { $VerbosePreference = "Continue" Write-Verbose "Verbose mode enabled" } Write-Info "Script started — add your logic here" # --- CUSTOM CODE BELOW --- """, }, } class Scripts(commands.Cog): """Generate script templates for various languages.""" def __init__(self, bot: commands.Bot) -> None: self.bot = bot @commands.group(name="script", invoke_without_command=True) async def script_group(self, ctx: commands.Context) -> None: """Script generation commands. Use `!script list` to see available languages.""" embed = discord.Embed( title="ScriptForge — Help", description=( f"Generate ready-to-use script scaffolds. Use:\n" f"`{Config.PREFIX}script list` — show available languages\n" f"`{Config.PREFIX}script generate ` — create a script\n" f"`{Config.PREFIX}script info ` — details about a language" ), color=discord.Color.blue(), ) await ctx.send(embed=embed) @script_group.command(name="list") async def list_languages(self, ctx: commands.Context) -> None: """List all available script languages.""" embed = discord.Embed( title="Available Script Languages", color=discord.Color.green(), ) for lang, data in SCRIPT_TEMPLATES.items(): embed.add_field( name=f"`{lang}` ({data['extension']})", value=data["description"], inline=False, ) await ctx.send(embed=embed) @script_group.command(name="info") async def language_info(self, ctx: commands.Context, language: str) -> None: """Show details for a specific language.""" lang = language.lower() data = SCRIPT_TEMPLATES.get(lang) if not data: available = ", ".join(f"`{k}`" for k in SCRIPT_TEMPLATES) embed = discord.Embed( title="Unknown Language", description=( f"`{lang}` is not available. Choose from: {available}\n" f"Use `{Config.PREFIX}script list` to see all options." ), color=discord.Color.red(), ) await ctx.send(embed=embed) return embed = discord.Embed( title=f"`{lang}` Script Template", description=data["description"], color=discord.Color.blue(), ) embed.add_field(name="File Extension", value=data["extension"]) embed.add_field( name="How to generate", value=f"`{Config.PREFIX}script generate {lang} my_script 'does something'`", inline=False, ) await ctx.send(embed=embed) @script_group.command(name="generate") async def generate_script( self, ctx: commands.Context, language: str, name: str, *, description: str = "A script generated by ScriptForge", ) -> None: """Generate a script template. Usage: !script generate """ lang = language.lower() data = SCRIPT_TEMPLATES.get(lang) if not data: available = ", ".join(f"`{k}`" for k in SCRIPT_TEMPLATES) embed = discord.Embed( title="Unknown Language", description=( f"`{lang}` is not available. Choose from: {available}\n" f"Use `{Config.PREFIX}script list` to see all options." ), color=discord.Color.red(), ) await ctx.send(embed=embed) return content = data["template"].format(name=name, description=description) filename = f"{name}{data['extension']}" embed = discord.Embed( title=f"Generated `{filename}`", description=f"Language: **{lang}** | Lines: **{content.count(chr(10)) + 1}**", color=discord.Color.green(), ) embed.add_field( name="Usage", value=f"`{Config.PREFIX}script generate {lang} `", inline=False, ) embed.set_footer(text="Reply below to refine or ask for modifications") await ctx.send(embed=embed) await ctx.send(f"```{lang}\n{content}\n```") @script_group.error async def script_error(self, ctx: commands.Context, error: Exception) -> None: if isinstance(error, commands.MissingRequiredArgument): embed = discord.Embed( title="Missing Arguments", description=( f"Usage: `{Config.PREFIX}script generate `\n" f"Example: `{Config.PREFIX}script generate python backup_tool 'Backs up files to S3'`" ), color=discord.Color.red(), ) await ctx.send(embed=embed) else: embed = discord.Embed( title="Error", description=f"Something went wrong: {error}", color=discord.Color.red(), ) await ctx.send(embed=embed) async def setup(bot: commands.Bot) -> None: await bot.add_cog(Scripts(bot))