mirror of
https://github.com/TeamPGM/PagerMaid_Plugins_Pyro.git
synced 2024-11-25 11:58:06 +00:00
71 lines
2.1 KiB
Python
71 lines
2.1 KiB
Python
from pagermaid.listener import listener
|
|
from pagermaid.enums import Message
|
|
from pagermaid.services import client as requests
|
|
|
|
import socket
|
|
|
|
|
|
def is_ip(ip):
|
|
try:
|
|
socket.inet_aton(ip)
|
|
return True
|
|
except Exception:
|
|
return False
|
|
|
|
|
|
def get_ip(domain):
|
|
try:
|
|
ip = socket.gethostbyname(domain)
|
|
except Exception:
|
|
return None
|
|
return ip
|
|
|
|
|
|
async def post(host):
|
|
url = "https://api.potatonet.idc.wiki/network/simple_health_check/scripts/gfw_check"
|
|
data = {"host": host}
|
|
return await requests.post(url, data=data)
|
|
|
|
|
|
@listener(command="gfw", parameters="[text]", description="查询是否被墙")
|
|
async def gfw(message: Message):
|
|
text = None
|
|
if message.arguments:
|
|
text = message.arguments
|
|
elif message.reply_to_message:
|
|
text = message.reply_to_message.text
|
|
if not text:
|
|
return await message.edit("请输入或回复一个IP或域名。")
|
|
try:
|
|
if not is_ip(text):
|
|
if get_ip(text) is None:
|
|
return await message.edit("出错了呜呜呜 ~ 无效的参数。")
|
|
else:
|
|
text = get_ip(text)
|
|
try:
|
|
message: Message = await message.edit(f"{text} 查询中...")
|
|
except Exception:
|
|
return await message.edit("出错了呜呜呜 ~ 无效的参数。")
|
|
data = await post(text)
|
|
data = data.json()
|
|
except Exception as e:
|
|
return await message.edit(f"出错了呜呜呜 ~ 查询失败。{e}")
|
|
|
|
if data["success"]:
|
|
if (
|
|
data["data"]["tcp"]["cn"] == data["data"]["tcp"]["!cn"]
|
|
and data["data"]["icmp"]["cn"] == data["data"]["icmp"]["!cn"]
|
|
):
|
|
if (
|
|
data["data"]["tcp"]["cn"] == False
|
|
and data["data"]["icmp"]["cn"] == False
|
|
):
|
|
reply = f"IP: {text}\n状态: 全球不通，不能判断是否被墙"
|
|
else:
|
|
reply = f"IP: {text}\n状态: 未被墙"
|
|
else:
|
|
reply = f"IP: {text}\n状态: 被墙"
|
|
else:
|
|
reply = f"IP: {text}\n状态: 查询失败"
|
|
return await message.edit(reply)
|