main.py

import os
from dotenv import load_dotenv
import time
from random import randint
import discord
from discord.ext import tasks
import subprocess
import shlex
from platform import system
import requests

import check_folder_size as cfs
import change_autoplaylist as catpl
import log_writter
from youtube_to_mp3 import main_dl
import detect_pc_status as dps
import update
import user_exp

intents = discord.Intents.all()
intents.members = True
client = discord.Client(intents=intents)
localtime = time.localtime()
base_dir = os.path.abspath(os.path.dirname(__file__))


async def check_voice_channel():
    # 列出所有語音頻道
    voice_channel_lists = []
    for server in client.guilds:
        for channel in server.channels:
            if channel.type == discord.ChannelType.voice:
                voice_channel_lists.append(channel)
                print(server.name + "/" + channel.name)
                members = channel.members
                msg = ""
                # 列出所有語音頻道的成員
                for member in members:
                    print("   ⌊" + member.name)
                    if member == client.get_user(885723595626676264) or member == client.get_user(657519721138094080):
                        # 若找到Allen Music Bot或Allen Why，則嘗試加入該語音頻道
                        try:
                            await client.get_channel(channel.id).connect(self_mute=True, self_deaf=True)
                            msg = "加入語音頻道：" + server.name + "/" + channel.name
                            log_writter.write_log(msg)
                            return "加入語音頻道：" + channel.name
                        except Exception as e:
                            msg = "加入語音頻道失敗：" + server.name + "/" + channel.name + "(" + str(e) + ")"
                            log_writter.write_log(msg)
                            if str(e) == "Already connected to a voice channel.":
                                return "已經連線至語音頻道。"
                            else:
                                return str(e)


@tasks.loop(seconds=10)
async def give_voice_exp():  # 給予語音經驗
    voice_channel_lists = []
    for server in client.guilds:
        for channel in server.channels:
            if channel.type == discord.ChannelType.voice:
                voice_channel_lists.append(channel)
                members = channel.members
                for member in members:
                    if not member.bot:
                        user_exp.add_exp(member.id, "voice", 1)


@client.event
async def on_ready():
    music = discord.Activity(type=discord.ActivityType.playing, name="修正完成！(狂喜)")
    await client.change_presence(status=discord.Status.online, activity=music)
    log_writter.write_log("-------------------------------------------------------------\n", True)
    log_writter.write_log("\n登入成功！\n目前登入身份：" +
                          str(client.user) + "\n以下為使用紀錄(只要開頭訊息有\"a!\"，則這則訊息和系統回應皆會被記錄)：\n\n")
    await check_voice_channel()
    for guild in client.guilds:
        for member in guild.members:
            date = member.joined_at.strftime("%Y-%m-%d %H:%M:%S")
            user_exp.set_join_date(member.id, date)
            print(f"{member.name} 加入於 {date}")
    await give_voice_exp.start()  # 開始偵測語音經驗


@client.event
async def on_member_join(member):
    msg = "歡迎 **" + member.name + "** 加入 __" + member.guild.name + "__ ！"
    await member.guild.system_channel.send(msg)
    log = str(member.guild.system_channel) + "/" + str(client.user) + ":\n" + msg
    log_writter.write_log(log)
    user_exp.set_join_date(member.id, member.joined_at.strftime("%Y-%m-%d %H:%M:%S"))
    new_member = await client.fetch_user(member.id)
    embed = discord.Embed(
        title="歡迎加入 " + member.guild.name + " ！",
        description="請到[這裡](https://discord.com/channels/857996539262402570/858373026960637962)查看頻道介紹。",
        color=0x57c2ea)
    await new_member.send(embed=embed)
    embed = discord.Embed(
        title="在開始之前...",
        description="什麼頻道都沒看到嗎？這是因為你**並未被分配身分組**。但是放心，我們會盡快確認你的身分，到時你就能加入我們了！",
        color=0x57c2ea)
    await new_member.send(embed=embed)


@client.event
async def on_member_remove(member):
    msg = "**" + member.name + "** 離開了 __" + member.guild.name + "__ ..."
    await member.guild.system_channel.send(msg)
    log = str(member.guild.system_channel + "/" + str(client.user) + ":\n" + msg)
    log_writter.write_log(log)


final_msg = []
msg_author = ""
msg_is_file = False
msg_send_channel = ""
testing = False


@client.event
async def on_message(message):  # 有訊息時
    global final_msg, msg_author, msg_is_file, msg_send_channel, testing
    msg_in = message.content
    if not message.author.bot and isinstance(msg_in, str):
        user_exp.add_exp(message.author.id, "text", len(msg_in))
    elif not message.author.bot and isinstance(msg_in, discord.File):
        user_exp.add_exp(message.author.id, "text", 1)
    if message.author == client.user:  # 排除自己的訊息，避免陷入無限循環
        return
    elif msg_in == "a!test" and message.author == client.get_user(657519721138094080):
        ip = ""
        try:
            ip = requests.get("https://api.ipify.org").text
        except Exception as e:
            ip = "無法取得IP：" + str(e)
        if testing:
            testing = False
            final_msg.append("測試模式已關閉。({0})".format(ip))
        else:
            testing = True
            final_msg.append("測試模式已開啟。({0})".format(ip))
        use_log = str(message.channel) + "/" + str(message.author) + ":\n" + msg_in + "\n\n"
        log_writter.write_log(use_log)
    elif testing:
        return
    elif msg_in[:2] == "a!":
        embed = discord.Embed(
            title="感謝你的使用！",
            description="你好，感謝你使用Allen Bot！\n提醒你，雖然本機器人即將被棄用，但是我們也準備好了替代品——<@1059102973998936137>，來繼承Allen Bot的所有功能！",
            color=0x57c2ea)
        embed.set_author(name="Allen Why", icon_url=client.get_user(657519721138094080).display_avatar)
        embed.add_field(name="為何要棄用Allen Bot？", value="Allen Bot是我的第一個機器人，但是它的程式碼寫得很糟糕，毫無條理"
                        "(你也能在[這裡](https://github.com/Alllen95Wei/My-Discord-Bot)"
                        "看到這場噩夢🤯)；\n"
                        "加上Discord發布了新版的「斜線指令」功能，讓使用機器人變得更加簡單。\n"
                        "所以我們決定要重寫一個新的機器人，來取代Allen Bot。", inline=False)
        embed.add_field(name="Allen Bot的替代品是什麼？", value="Allen Bot的替代品是<@1059102973998936137>，它的程式碼寫得"
                        "比Allen Bot好多了，而且也支援所有Allen Bot的功能(甚至更多！)，更重要的是：支援斜線指令(超級重要)！", inline=False)
        embed.add_field(name="那麼，Allen Bot會怎麼樣？", value="在<@1059102973998936137>的設計完成後，"
                        "Allen Bot的原始碼將會於GitHub封存，並轉為read-only。", inline=False)
        embed.add_field(name="我應該如何使用新版的機器人？", value="請使用</help:1069227660816957491>來取得協助。",
                        inline=False)
        embed.set_footer(text="Allen Bot停止服務通知")
        await message.channel.send(embed=embed)
        use_log = str(message.channel) + "/" + str(message.author) + ":\n" + msg_in + "\n\n"
        log_writter.write_log(use_log)
        if len(msg_in) == 2:
            final_msg.append("我在這！\n如果需要指令協助，請輸入`a!help`")
        elif msg_in[2:5] == "say":
            if msg_in[6:] == "":
                final_msg.append("```參數:\nsay <文字>：用大聲公📣+粗體+斜體+底線說出文字```")
            else:
                msg = "📣📣___**" + str(msg_in[6:]) + "**___"
                final_msg.append(msg)
        elif msg_in[2:6] == "help":  # a! 指令說明
            final_msg.append("呼叫機器人：`a!`\n\n參數:\n"
                             "`say <文字>`：用大聲公📣+**粗體**+___斜體+底線___說出文字\n"
                             "`ama <問題>`：給你這個問題的隨機回答\n"
                             "`random <範圍>`：在指定數字範圍隨機取得一數\n"
                             "`qrcode <文字>`：將輸入的文字轉換為QR Code\n"
                             # "`daily901 <new/channel/fb>`：得到關於「日常901」的資訊\n"
                             "`rickroll`：？？？\n"
                             "`ytdl <YouTube連結>`：下載YouTube的影片為mp3\n"
                             "`rc [ID]`：重新連接至語音頻道。可指定頻道ID，否則將自動檢測音樂機器人及Allen Why在哪個頻道\n"
                             "`dc`：嘗試從目前的語音頻道中斷連接\n"
                             "`ping`：查詢機器人的延遲(毫秒)\n"
                             "`dps`：查詢伺服器電腦的CPU及記憶體使用率\n"
                             "`sizecheck`：檢查`\"C:\\MusicBot\\audio_cache\"`的大小\n"
                             "`changeatpl <bgm/normal>`：更換Allen Music Bot的自動播放清單\n"
                             "`cmd <指令>`：在伺服器端執行指令並傳回結果"
                             "\n想得到更詳細的指令參數說明，直接輸入指令而不加參數即可\n試試看吧！")
        elif msg_in[2:5] == "ama":
            if len(msg_in) == 5:
                final_msg.append("```參數：\nama <問題>：就是8號球，給你這個問題的隨機回答```")
            else:
                ans1 = ("g", "s", "b")
                ans_g = ("看起來不錯喔", "肯定的", "我覺得可行", "絕對OK", "是的", "確定", "200 OK", "100 Continue",
                         "Just do it")
                ans_s = (
                    "現在別問我", "404 Not Found", "你的問題超出宇宙的範圍了", "答案仍在變化", "400 Bad Request",
                    "這問題實在沒人答得出來",
                    "Answer=A=Ans=答案",
                    "最好不要現在告訴你", "300 Multiple Choices", "去問瑪卡巴卡更快",
                    "您撥的電話無人接聽，嘟聲後開始計費。", "對不起，您播的號碼是空號，請查明後再撥。")

                ans_b = (
                    "不可能", "否定的", "不值得", "等等等等", "No no no", "我拒絕", "我覺得不行耶",
                    "https://cdn2.ettoday.net/images/4945/4945172.jpg",
                    "403 Forbidden", "這樣不好")

                ball_result1 = ans1[randint(0, 2)]
                if ball_result1 == "g":
                    ball_result2 = ans_g[randint(0, len(ans_g) - 1)]
                    ans2 = "```diff\n+✓" + ball_result2
                elif ball_result1 == "s":
                    ball_result2 = ans_s[randint(0, len(ans_s) - 1)]
                    ans2 = "```fix\n□" + ball_result2
                else:
                    ball_result2 = ans_b[randint(0, len(ans_b) - 1)]
                    ans2 = "```diff\n-✗" + ball_result2
                final_msg.append(ans2 + "```")
        elif msg_in[2:8] == "random":
            if len(msg_in) == 8:
                final_msg.append("```參數：\nrandom <範圍>：在指定數字範圍隨機取得一數\n"
                                 "       <範圍>：輸入2個數值，中間以逗號做分隔\n(提示：參數輸入\"%\"就會自動設定範圍為0~100)```")
            else:
                if msg_in[9:] == "%":
                    a = 0
                    b = 100
                else:
                    a = int(msg_in[9:msg_in.find(",")])
                    b = int(msg_in[msg_in.find(",") + 1:])
                r_range_result = randint(a, b)
                final_msg.append("結果：```fix\n" + str(r_range_result) + "```")
        elif msg_in[2:8] == "qrcode":
            if len(msg_in) == 8:
                final_msg.append("```參數：\nqrcode <文字>：將輸入的文字轉換為QR Code```")
            else:
                import urllib.parse
                text = urllib.parse.quote(msg_in[9:])
                final_msg.append(
                    "https://chart.apis.google.com/chart?cht=qr&chs=500x500&choe=UTF-8&chld=H|1&chl=" + text)
        elif msg_in[2:10] == "rickroll":
            channel = message.author.voice.channel
            try:
                await channel.connect()
            except Exception as e:
                final_msg.append("```" + str(e) + "```")
            final_msg.append("ap!p never gonna give you up")
            final_msg.append("ap!skip f")
            msg_send_channel = client.get_channel(891665312028713001)
        elif msg_in[2:11] == "sizecheck":
            if "Direct Message" in str(message.channel):
                final_msg.append("不能在私人訊息使用此指令。請至伺服器使用此指令。")
            elif message.author == client.get_user(657519721138094080) or message.guild.owner:
                final_msg.append(cfs.check_size())
                msg_author = message.author
            else:
                final_msg.append("你無權使用此指令。")
        elif msg_in[2:12] == "changeatpl":
            if "Direct Message" in str(message.channel):
                final_msg.append("不能在私人訊息使用此指令。請至伺服器使用此指令。")
            else:
                if message.author == client.get_user(657519721138094080) or message.guild.owner:
                    if msg_in[13:16] == "bgm":
                        final_msg.append(catpl.change_atpl_to_bgm())
                    elif msg_in[13:19] == "normal":
                        final_msg.append(catpl.change_atpl_to_normal())
                    else:
                        final_msg.append("```參數：\nchangeatpl bgm：將Allen Music Bot的自動播放清單換為BGM playlist。\n"
                                         "           normal：將Allen Music Bot的自動播放清單回歸原狀。```")
                else:
                    final_msg.append("你無權使用此指令。")
        elif msg_in[2:6] == "ytdl":
            if len(msg_in) == 6:
                final_msg.append("```參數：\nytdl <YouTube連結>：將該YouTube影片下載為mp3，再傳回Discord。由於Discord有"
                                 "檔案大小限制，因此有時可能會失敗。```")
            else:
                yt_url = msg_in[7:]
                file_name = str(message.author) + yt_url[-11:]
                if main_dl(yt_url, file_name, file_name + ".mp3") == "finished":
                    final_msg.append(discord.File(file_name + ".mp3"))
                    msg_is_file = True
        elif msg_in[2:4] == "rc":
            channel_id = msg_in[5:]
            if channel_id != "":
                try:
                    vc = client.get_channel(int(channel_id))
                    try:
                        await vc.connect(self_mute=True, self_deaf=True)
                        final_msg.append("加入語音頻道：{0}".format(client.get_channel(int(channel_id))))
                    except Exception as e:
                        if str(e) == "Already connected to a voice channel.":
                            final_msg.append("已經連線至語音頻道。")
                        else:
                            final_msg.append("```" + str(e) + "```")
                except ValueError as VE:
                    final_msg.append("參數錯誤：請貼上該頻道的ID！")
                    final_msg.append("```" + str(VE) + "```")
                except Exception as e:
                    final_msg.append("```" + str(e) + "```")
            else:
                final_msg.append(await check_voice_channel())
        elif msg_in[2:5] == "dps":
            act_msg = dps.pc_status()
            final_msg.append(act_msg)
        elif msg_in[2:6] == "ping":
            final_msg.append("目前延遲：" + str(round(client.latency * 1000)) + "ms")
        elif msg_in[2:5] == "cmd":
            if message.author == client.get_user(657519721138094080):
                if len(msg_in) == 5:
                    final_msg.append("```參數：\ncmd <指令>：在伺服器端執行指令並傳回結果。```")
                else:
                    command = msg_in[6:]
                    if command == "cmd":
                        final_msg.append("不能使用`cmd`指令。")
                    else:
                        try:
                            command = shlex.split(command)
                            command_output = str(subprocess.run(command, capture_output=True, text=True).stdout)
                            if command_output != "":
                                final_msg.append("```" + command_output + "```")
                            else:
                                final_msg.append("終端未傳回回應。")
                        except WindowsError as e:
                            if "WinError 2" in str(e):
                                final_msg.append("似乎沒有這個指令，或指令無法透過Python執行。")
                                final_msg.append("錯誤內容：\n```" + str(e) + "```")
                            else:
                                final_msg.append("```" + str(e) + "```")
            else:
                final_msg.append("你無權使用此指令。")
        elif msg_in[2:4] == "dc":
            try:
                await message.guild.voice_client.disconnect()
                final_msg.append("已嘗試離開語音頻道。")
            except Exception as e:
                if str(e) == "'NoneType' object has no attribute 'disconnect'":
                    final_msg.append("機器人目前沒有連接到語音頻道。")
                else:
                    final_msg.append("中斷連接時發生問題。")
                    final_msg.append("```" + str(e) + "```")
        elif msg_in[2:8] == "update":
            if message.author == client.get_user(657519721138094080):
                owner = client.get_user(657519721138094080)
                await owner.send("更新流程啟動。")
                event = discord.Activity(type=discord.ActivityType.playing, name="更新中...")
                await client.change_presence(status=discord.Status.do_not_disturb, activity=event)
                update.update(os.getpid(), system())
            else:
                final_msg.append("你無權使用此指令。")
        else:
            final_msg.append("參數似乎無效...\n輸入`a!help`獲得說明")
    elif message.channel == client.get_channel(891665312028713001):
        if "https://www.youtube.com" == msg_in[:23] or "https://youtu.be" == msg_in[:16] or "https://open.spotify.com" \
                == msg_in[:24]:
            if "&list=" in msg_in:
                msg_in = msg_in[:msg_in.find("&list=")]
                final_msg.append("<@{0}> 已將清單連結轉換為單一影片連結。".format(message.author.id))
            ap_cmd = "ap!p " + msg_in
            final_msg.append(ap_cmd)
            use_log = str(message.channel) + "/" + str(message.author) + ":\n" + msg_in + "\n\n"
            log_writter.write_log(use_log)
    if msg_send_channel == "":
        msg_send_channel = message.channel
    for i in range(len(final_msg)):
        current_msg = final_msg[i]
        try:
            if isinstance(current_msg, str):
                await msg_send_channel.send(current_msg)
            elif isinstance(current_msg, discord.File):
                await msg_send_channel.send(file=current_msg)
            else:
                await msg_send_channel.send("```" + str(current_msg) + "```")
            new_log = str(msg_send_channel) + "/" + str(client.user) + ":\n" + str(final_msg[i]) + "\n\n"
            log_writter.write_log(new_log)
        except Exception as e:
            if "or fewer in length." in str(e):
                txt_file_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'full_msg.txt')
                open(txt_file_path, "w").write(str(final_msg[i]))
                await msg_send_channel.send("由於訊息長度過長，因此改以文字檔方式呈現。",
                                            file=discord.File(txt_file_path))
                os.remove(txt_file_path)
                new_log = str(msg_send_channel) + "/" + str(
                    client.user) + ":\n" + "由於訊息長度過長，因此改以文字檔方式呈現。" + "\n\n"
                log_writter.write_log(new_log)
            else:
                final_msg = "發生錯誤。錯誤內容如下：\n```" + str(e) + "```"
                await msg_send_channel.send(final_msg)
                new_log = str(msg_send_channel) + "/" + str(client.user) + ":\n" + str(final_msg) + "\n\n"
                log_writter.write_log(new_log)
    final_msg = []
    msg_is_file = False
    msg_send_channel = ""


# 取得TOKEN
load_dotenv(dotenv_path=os.path.join(base_dir, "TOKEN.env"))
TOKEN = str(os.getenv("TOKEN"))
client.run(TOKEN)