main.py

import asyncio
import sys

import discord
from discord.ext import commands, tasks

import ast
import os
import requests
import signal
import random
import time

import httplist

owners = [401849772157435905, 876488885419520020] # Owner account IDs


with open('token.txt', 'r') as f:
    bottoken = f.readline().strip()

totalmessages = 0


with open('userdata.txt', 'r') as f:
    userdata = ast.literal_eval(f.read())


def save_all():
    with open('userdata.txt', 'w') as f:
        f.write(repr(userdata))


def timeout_handler(signal, frame):
    print("BOT: Timeout!") # this shouldn't exit!
    #print(f'BOT: Timeout! Saving data and exiting...')
    #save_all()
    #sys.exit(0)


signal.signal(signal.SIGALRM, timeout_handler)



intents = discord.Intents.default()
intents.members = True

client = commands.Bot(command_prefix = '$', help_command=None, intents=intents) #Makes the bot prefix.

    # https://discord.com/api/oauth2/authorize?client_id=1079242361491693658&permissions=8&scope=applications.commands%20bot

class LeaderBoardPosition:
    def __init__(self, user, xp):
        self.user = user
        self.xp = xp

@client.event
async def on_ready():
    print("========")
    print(f"current UNIX time is {time.time()}.")
    print("========")
    print('Logged in as')
    print(client.user.name)
    print(client.user.id)
    print('========')


@client.event
async def on_message(message):
    global totalmessages
    if message.author.id not in userdata.keys():
        userdata[message.author.id] = {'xp': 0, 'time': -1}
    if userdata[message.author.id]['time'] + 30 < time.time():
        userdata[message.author.id]['time'] = time.time()
        userdata[message.author.id]['xp'] += random.randint(5, 10)
    totalmessages += 1

    save_all()

    await client.process_commands(message)


@client.command()
async def betatestr(ctx):
    user = ctx.message.author
    await user.add_roles(1088968688683720745)
    await ctx.send("You should now have the Beta Testr role!")
    
    
@client.command()
async def explain(ctx):
    embed = discord.Embed(title="Explanation of This:tm:", description="", color=0xFF0000)
    embed.add_field(name="What is this server?", value="This Discord Server is the community area for Surviv Reloaded, open-source server for the defunct online game surviv.io.")
    embed.add_field(name="What is this bot?", value="This bot was made by Killaship to save the hassle of explaining what this is to everyone.")
    embed.add_field(name="What is Surviv Reloaded?", value="It's an open-source server hosting the original client. In other words, it's the original surviv.io, just hosted by a different server. It's not a clone of Surviv.io.")
    embed.add_field(name="Where can I get more info?", value="https://github.com/SurvivReloaded")
    await ctx.send(embed=embed)


@client.command()
async def shell(ctx,cmd):
    if ctx.author.id in owners:
       out = os.popen(str(cmd))
       try:
           await ctx.send(str(out.read()))
       except:
          os.system(cmd)


@client.command()
async def awardxp(ctx, user: discord.User, amount = 0):
    if ctx.message.author.id in owners:
        if user.id not in userdata:
            userdata[user.id] = {'xp': 0, 'time': -1}
        userdata[user.id]['xp'] += amount
        await ctx.send(f'Gave {user.mention} {amount} XP!')
    
    

  
@client.command()
async def links(ctx):
    embed = discord.Embed(title="Links", description="", color=0xFF0000)
    embed.add_field(name="Test Server", value="https://test.resurviv.io")
    embed.add_field(name="Play Stable Version", value="https://resurviv.io")
    embed.add_field(name="Discord Perma Invite", value="https://discord.resurviv.io")
    embed.add_field(name="Subreddit", value="https://reddit.com/r/survivreloaded")
    embed.add_field(name="Github (Organization)", value="https://github.com/SurvivReloaded")
    embed.add_field(name="GitLab (deprecated)", value="https://gitlab.com/hasanger/survivreloaded")
    embed.add_field(name="Bot GitHub", value="https://github.com/Killaship/survivbot")
    await ctx.send(embed=embed)


@client.command()
async def getxp(ctx, user: discord.User = None):
    user = ctx.author if not user else user
    if user.id not in userdata:
        await ctx.send(f'{user.mention} has 0 XP!')
    await ctx.send(f'{user.mention} has {userdata[user.id]["xp"]} XP!')


@client.command()
async def getleaderboard(ctx):
    leaderboards = []
    for key in userdata:
        leaderboards.append(LeaderBoardPosition(key, userdata[key]['xp']))
    top = sorted(leaderboards, key=lambda x: x.xp, reverse=True)
    text = ''
    for i in range(0, 6):
        try:
            text += f'{i + 1}. {client.get_user(top[i].user)} - {top[i].xp} XP\n'
        except IndexError:
            break
    embed = discord.Embed(title=text, description="Top 6 XP counts!", color=0xFF0000)
    await ctx.send(embed=embed)


@client.event
async def on_message_delete(message):
    deleted = discord.Embed(description="Message deleted in {msgchannel}".format(msgchannel=message.channel.mention), color=0xFF0000)
    channel=client.get_channel(1092435780095451236)
    if(message.channel.id != 1092435780095451236 and message.channel.id != 1091118930749309008):
        
        if message.attachments:
            if(len(message.attachments) == 1):
                if message.attachments[0].url.endswith(('.jpg', '.png', '.jpeg', '.gif')):
                    deleted.set_image(url=message.attachments[0].url)
                else:
                    deleted.add_field(name="Attachment", value=message.attachments[0].url) # No attachment or unsupported file     
        deleted.add_field(name="Author", value=message.author)
        deleted.add_field(name="Message", value=message.content)
        deleted.timestamp = message.created_at
        await channel.send(embed=deleted)
    
    
@client.event
async def on_message_edit(message_before, message_after):
    edited = discord.Embed(description="Message edited in {msgchannel}".format(msgchannel=message_before.channel.mention), color=0xFFFF00)
    channel=client.get_channel(1092435780095451236)
    if(message_before.channel.id != 1092435780095451236 and message_before.channel.id != 1091118930749309008):
        
        if message_before.attachments:
            if(len(message_before.attachments) == 1):
                if message_before.attachments[0].url.endswith(('.jpg', '.png', '.jpeg', '.gif')):
                    edited.set_image(url=message_before.attachments[0].url)
                else:
                    edited.add_field(name="Attachment", value=message_before.attachments[0].url) # No attachment or unsupported file     
        edited.add_field(name="Author", value=message_before.author)
        edited.add_field(name="Message Before", value=message_before.content)
        edited.add_field(name="Message After", value=message_after.content)
        edited.timestamp = message_before.created_at
        await channel.send(embed=edited)
    
    

@client.command()
async def help(ctx):
   
  

    embed = discord.Embed(title="help", description="", color=0xFF0000)#Declaring the help command is an embed.

    

    embed.add_field(name="$explain", value="Explains this server and lists FAQ.")

    embed.add_field(name="$links", value="Lists various links related to the project.")

    embed.add_field(name="$serverstatus", value="Checks whether the game server (or at least website) is up. It checks all websites on which the game is hosted. Not too reliable, might return 502 errors.")

    embed.add_field(name="$getxp", value="This command shows the amount of XP the sender has.")
    
    embed.add_field(name="$getleaderboard", value="This command lists the 5 members of the server with the most XP!. (6 including bot)")

    embed.add_field(name="$checkurl", value="Checks the connectivity of any URL.")

    embed.add_field(name="$help", value="This command.")

    embed.add_field(name="Bot GitHub", value="https://github.com/Killaship/survivbot")

    await ctx.send(embed=embed)#sends the embed.


@client.command()
async def serverstatus(ctx):
    text = ["Checking 3 site(s)\n", "Note: This may or may not be accurate. Do not trust these results.\n",
            "*https://taskjourney.org:449*\n"]
    code = urlcheck("https://taskjourney.org:449/")
    if code != 200:
        text.append("The server sent an abnormal response. If it's 301 or 302, the server redirected the bot. If it's not those, the server might be down. The HTTP code sent was: {http}. ({phrase})\n".format(http=str(code), phrase=httplist.httplist[code]))
    else:
        text.append("The server is currently up. (It sent a response code of 200 OK)\n")
        text.append("If your game is frozen, it's most likely that the client froze or crashed. The game is still relatively unstable, you'll have to reload the game.\n")
        code = urlcheck("https://survivreloaded.com/")
    text.append("*https://survivreloaded.com/*\n")
    code = urlcheck("https://survivreloaded.com/")
    if code != 200:
        text.append("The server sent an abnormal response. If it's 301 or 302, the server redirected the bot. If it's not those, the server might be down. The HTTP code sent was: {http}. ({phrase})\n".format(http=str(code), phrase=httplist.httplist[code]))
    else:
        text.append("The server is currently up. (It sent a response code of 200 OK)\n")
        text.append("If your game is frozen, it's most likely that the client froze or crashed. The game is still relatively unstable, you'll have to reload the game.\n")  
    code = urlcheck("https://resurviv.io/")
    text.append("*https://resurviv.io/*\n")
    if code != 200:
        text.append("The server sent an abnormal response. If it's 301 or 302, the server redirected the bot. If it's not those, the server might be down. The HTTP code sent was: {http}. ({phrase})\n".format(http=str(code), phrase=httplist.httplist[code]))
    else:
        text.append("The server is currently up. (It sent a response code of 200 OK)\n")
        text.append("If your game is frozen, it's most likely that the client froze or crashed. The game is still relatively unstable, you'll have to reload the game.\n")
    embed = discord.Embed(title="Surviv Reloaded Status", description=''.join(text), color=0x00FF00)
    await ctx.send(embed=embed)


@client.command()
async def checkurl(ctx,site):
    code = urlcheck(site)
    if code != 200:
        await ctx.send("The server is currently down or unresponsive. The HTTP code sent was: {http}. ({phrase})".format(http=str(code), phrase=httplist.httplist[code]))
    else:
        await ctx.send("The server is currently up. (It sent a response code of 200 OK)")

@client.command()
async def resetbot(ctx):
    if ctx.message.author.id in owners:
        await ctx.send("Bot is reloading, please wait a few seconds before sending commands.")
        exit()
    else:
        await ctx.send("hey, wait a minute, you're not the owner! you can't do that! >:(")        

@client.command()
async def ownersonly(ctx):
    if ctx.message.author.id in owners:
        await ctx.send("You are the owner of this application.")
        exit()
    else:
        await ctx.send("You're not the owner of this application.")         
        
        
def urlcheck(url):
    signal.alarm(httplist.TIMEOUT)
    try:
        r = requests.head(url, timeout=3)
        return r.status_code
    except:
        return 999

@tasks.loop(minutes=10)
async def save():
    save_all()



save.start()
client.run(bottoken)