mirror of
https://github.com/Murasame-Dev/McStatus-API.git
synced 2025-12-14 12:18:03 +00:00
ギリ猫
f6b9b36713
* Add module mc-status-img * delete: more submodule dir * feat: compact pull #5 and delete unuseable things. * chore: move the functions and modules to great place. * add: ignore the .vscode directory * chore: add line break in the end. --------- Co-authored-by: quqiOnfree <quqiOnfree@outlook.com>
116 lines
3.5 KiB
Python
116 lines
3.5 KiB
Python
# -*- coding: utf-8 -*-
|
||
# 重写 Flask-MCMOTD,早期版本用的是面向过程的方式进行写的，一个文件写了400多行，真是要爆了T.T
|
||
# 2025/8/28 by Murasame:使用 *Claude Sonnet 4* 重写成了 FastAPI
|
||
|
||
# 基础模块
|
||
import asyncio
|
||
import base64
|
||
import uvicorn
|
||
from io import BytesIO
|
||
|
||
# API
|
||
from fastapi import FastAPI, HTTPException, Query, Response
|
||
from fastapi.middleware.cors import CORSMiddleware
|
||
|
||
# Java版查询模块
|
||
from JavaServerStatus import java_status
|
||
# 基岩版查询模块
|
||
from BedrockServerStatus import bedrock_status
|
||
# 此API优先解析 srv 记录
|
||
from dnslookup import dns_lookup
|
||
|
||
# 格式化文本
|
||
from FormatData import format_java_data, format_bedrock_data, format_index, format_java_index, format_bedrock_index, format_img_index
|
||
|
||
# 图片生成模块
|
||
from generate_img import get_icon_image, generate_java_status_image, generate_bedrock_status_image, get_background_image
|
||
|
||
app = FastAPI(
|
||
title="McStatus API",
|
||
description="Minecraft服务器状态查询API",
|
||
version="3.0.0"
|
||
)
|
||
|
||
# 添加CORS中间件
|
||
app.add_middleware(
|
||
CORSMiddleware,
|
||
allow_origins=["*"],
|
||
allow_credentials=True,
|
||
allow_methods=["*"],
|
||
allow_headers=["*"],
|
||
)
|
||
|
||
@app.get("/")
|
||
async def index():
|
||
message = format_index()
|
||
return message
|
||
|
||
# Java 服务器状态查询
|
||
@app.get("/java")
|
||
async def get_java_status(ip: str = Query(None, description="服务器IP地址或域名")):
|
||
# 空值输出 API 用法
|
||
if not ip:
|
||
message = format_java_index()
|
||
raise HTTPException(status_code=400, detail=message)
|
||
|
||
try:
|
||
ip, type = dns_lookup(ip)
|
||
print(f"解析Java版IP: {ip}, 是否为 SRV: {type}")
|
||
status = java_status(ip)
|
||
|
||
data = format_java_data(ip, type, status)
|
||
|
||
return data
|
||
|
||
except Exception as e:
|
||
raise HTTPException(status_code=500, detail=str(e))
|
||
|
||
# 基岩版服务器状态查询
|
||
@app.get("/bedrock")
|
||
async def get_bedrock_status(ip: str = Query(None, description="服务器IP地址或域名")):
|
||
# 空值输出 API 用法
|
||
if not ip:
|
||
message = format_bedrock_index()
|
||
raise HTTPException(status_code=400, detail=message)
|
||
|
||
try:
|
||
print(f"解析基岩版IP: {ip}")
|
||
status = bedrock_status(ip)
|
||
|
||
data = format_bedrock_data(ip, status)
|
||
|
||
return data
|
||
|
||
except Exception as e:
|
||
raise HTTPException(status_code=500, detail=str(e))
|
||
|
||
# 生成服务器状态图片
|
||
@app.get("/img")
|
||
async def get_status_image(ip: str = Query(None, description="服务器IP地址或域名"), type: str = Query(None, description="服务器版本，java 或 bedrock")):
|
||
# 空值输出 API 用法
|
||
if not ip or not type or type not in ["java", "bedrock"]:
|
||
message = format_img_index()
|
||
raise HTTPException(status_code=400, detail=message)
|
||
|
||
try:
|
||
# 背景图获取方法
|
||
background_data = await get_background_image()
|
||
|
||
# java 服务器的方法
|
||
if type == "java":
|
||
image = await generate_java_status_image(ip)
|
||
|
||
# bedrock 服务器的方法
|
||
if type == "bedrock":
|
||
image = await generate_bedrock_status_image(ip)
|
||
|
||
img_io = BytesIO()
|
||
image.save(img_io, 'JPEG')
|
||
img_io.seek(0)
|
||
return Response(content=img_io.getvalue(), media_type="image/jpeg")
|
||
|
||
except Exception as e:
|
||
raise HTTPException(status_code=500, detail=str(e))
|
||
|
||
if __name__ == '__main__':
|
||
uvicorn.run("app:app", host="0.0.0.0", port=5000, reload=True)
|