chimera.pyw

# Basic bot dependencies
import discord
from discord.ext.commands import Bot
import platform
import os
import asyncio
from threading import Thread

# Import configurations
import configs

# Import logger
from lib.helpers import Logger

# Imports for system tray icon
from pystray import Icon, Menu, MenuItem
from PIL import Image
import webbrowser

# Modules import - this imports all modules under the modules directory
# IDEs will complain about unresolved references, but it runs as intended
from modules import *

# Create a bot client with a description and a command prefix
client = Bot(description="A remote system administration bot for discord", command_prefix=configs.BOT_PREFIX)


@client.event
async def on_ready():
    print('--------')
    print('Chimera Remote Administration Bot by CedArctic')
    print('--------')
    print('Logged in as ' + client.user.name + ' (ID:' + str(client.user.id) + ') | Connected to '
          + str(len(client.guilds)) + ' servers | Connected to ' + str(len(set(client.get_all_members()))) + ' users')
    print('--------')
    print(f'Current Discord.py Version: {discord.__version__} | Current Python Version: {platform.python_version()}')
    print('--------')
    print(f'Use this link to invite {client.user.name}:')
    print(f'https://discordapp.com/oauth2/authorize?client_id={client.user.id}&scope=bot&permissions=8')
    print('--------')
    print('Github Link: https://github.com/CedArctic/Chimera')
    print('--------')
    return await client.change_presence(activity=discord.Game(name='with your PC'))


# Module: openurl
# Description: Opens url in default browser
# Usage: !openurl http://example.com/
@client.command()
@Logger(client)
async def openurl(ctx, url):
    await openurl_module.openurl(ctx, url)


# Module: cmd
# Description: Executes cmd command
# Usage: !cmd "command"
@client.command()
@Logger(client)
async def cmd(ctx, cmnd):
    await cmd_module.cmd(ctx, cmnd)


# Module: powershell
# Description: Executes powershell command
# Usage: !powershell "command"
@client.command()
@Logger(client)
async def powershell(ctx, cmnd):
    await powershell_module.powershell(ctx, cmnd)


# Module: lock
# Description: Locks system
# Usage: !lock or !lock secondsToLock
@client.command()
@Logger(client)
async def lock(ctx, seconds=0):
    await lock_module.lock(ctx, seconds)

# Module: appQuitter
# Description: Quits the application
# Usage: !appquitter "Application Name" or !appquitter "Application Name" minutesToQuit
@client.command()
@Logger(client)
async def appquitter(ctx, appName,minutes=0):
    await appQuitter_module.appquitter(ctx,appName, minutes)
    
    
# Module: sleep
# Description: Puts system to sleep
# Usage: !sleep or !sleep secondsToSleep
@client.command()
@Logger(client)
async def sleep(ctx, seconds=0):
    await sleep_module.sleep(ctx, seconds)


# Module: shutdown
# Description: Shuts system down
# Usage: !shutdown or !shutdown secondsToShutdown
@client.command()
@Logger(client)
async def shutdown(ctx, seconds=0):
    await shutdown_module.shutdown(ctx, seconds)


# Module: restart
# Description: Restarts system
# Usage: !restart or !restart secondsToRestart
@client.command()
@Logger(client)
async def restart(ctx, seconds=0):
    await restart_module.restart(ctx, seconds)


# Module: hibernate
# Description: Hibernates the system
# Usage: !hibernate or !hibernate secondsToHibernation
@client.command()
@Logger(client)
async def hibernate(ctx, seconds=0):
    await hibernate_module.hibernate(ctx, seconds)


# Module: logoff
# Description: Logs the user out of the system
# Usage: !logoff or !logoff secondsToLogoff
@client.command()
@Logger(client)
async def logoff(ctx, seconds=0):
    await logoff_module.logoff(ctx, seconds)


# Module: screenshot
# Description: Takes a screenshot and sends it back
# Usage: !screenshot or !screenshot secondsToScreenshot
@client.command()
@Logger(client)
async def screenshot(ctx, seconds=0):
    await screenshot_module.screenshot(ctx, seconds)


# Module: say
# Description: Uses powershell and a TTS engine to make your computer say something
# Usage: !say "Something to say"
@client.command()
@Logger(client)
async def say(ctx, txt):
    await say_module.say(ctx, txt)


# Module: media
# Description: Controls Media Features
# Usage: !media command or !media command times
@client.command()
@Logger(client)
async def media(ctx, command, times=1):
    await media_module.media(ctx, command, times)


# Module: camera
# Description: Records a video or takes a photo (no audio)
# Usage: !camera command time
@client.command()
@Logger(client)
async def camera(ctx, command, time=5):
    await camera_module.camera(ctx, command, time)


# Module: echo
# Description: Turns command output display to discord chat on and off (works for !cmd and !powershell)
# Usage: !echo off or !echo on
@client.command()
@Logger(client)
async def echo(ctx, status):
    await echo_module.echo(ctx, status)


# Module: log
# Description: Turns on of off logs in chat. Also can be used to retrieve Chimera execution logs
# Usage: !log [off|on] | [show] [date (format: YYYY-MM-DD)]
@client.command()
@Logger(client)
async def log(ctx, param, date=None):
    await log_module.log(ctx, param, date)


# Module: file
# Description: Allows file download, upload and system navigation
# Usage: !file [command] [[path]|[times]]
@client.command()
@Logger(client)
async def file(ctx, command, *args):
    await file_module.file(ctx, command, *args)


# Module: launch
# Description: Launches a shortcut in the shortcuts directory
# Usage: !launch [shortcut]
@client.command()
@Logger(client)
async def launch(ctx, shortcut):
    await launch_module.launch(ctx, shortcut)


# Module: helpme
# Description: Allows file download, upload and system navigation
# Usage: !helpme [command]
@client.command()
@Logger(client)
async def helpme(ctx, command=None):
    await helpme_module.helpme(ctx, command)


# Module: notification
# Description: Sends a notification to the computer
# Usage: !notification "Notification Content"
@client.command()
@Logger(client)
async def notification(ctx, txt):
    await notification_module.notification(ctx, txt)



# System Tray menu functions

# Starts the bot client
def iconRun():
    loop = asyncio.get_event_loop()
    loop.create_task(client.start(configs.BOT_TOKEN))
    t1=Thread(target=loop.run_forever)
    t1.start()


# Shows logs folder
def showLogs(): os.startfile("logs")


# Shows shortcuts folder
def showShortcuts(): os.startfile("shortcuts")


# Opens bot invitation link in the browser
def connectInfo(): webbrowser.open(
    f'https://discordapp.com/oauth2/authorize?client_id={client.user.id}&scope=bot&permissions=8')


# Exits the application
def applicationExit():
    # This doesn't quit the bot client. To do that you need to call client.logout or .close
    # This merely stops the icon from showing on the taskbar. A full solution will be implemented later
    icon.visible = False
    icon.stop()


# About
def about(): webbrowser.open('https://github.com/CedArctic/Chimera/blob/master/README.md')


# Instructions
def instructions(): webbrowser.open('https://github.com/CedArctic/Chimera/blob/master/README.md')


# Create system tray icon and start running the client
def iconSetup():
    iconImage = Image.open("Chimera_logo.png")
    iconMenu = Menu(
        MenuItem("Connect", action=connectInfo, default=True),
        MenuItem("Instructions", action=instructions),
        MenuItem("Show Logs", action=showLogs),
        MenuItem("Show Shortcuts", action=showShortcuts),
        MenuItem("About", action=about),
        MenuItem("Exit", action=applicationExit),
    )
    icon = Icon('Chimera', icon=iconImage, menu=iconMenu)
    return icon


# Application Entry Point - starts icon and bot client
icon = iconSetup()
iconRun()
icon.run()