import random
from flask import Flask, jsonify, render_template

app = Flask(__name__)

# 数据源：用于随机生成的素材库
NAMES = [
    "星野", "云舟", "夜阑", "墨白", "子衿", "清和", "若曦", "南风",
    "朝歌", "暮雪", "青岚", "白露", "寒山", "流萤", "听澜", "望舒"
]

STORIES = [
    "在一个被星光笼罩的夜晚，{name}发现了一扇通往未知世界的门。",
    "{name}在旧书店的角落里找到了一本会说话的日记，从此命运改变。",
    "传说中，{name}拥有一颗能听见风之低语的心，于是踏上了寻找故乡的旅程。",
    "当城市沉入梦境，{name}却在屋顶上遇见了来自百年前的旅人。",
    "{name}在海边捡到一枚发光的贝壳，里面藏着一整个消失的王国。"
]

COLORS = [
    "#FF6B6B", "#4ECDC4", "#FFE66D", "#1A535C", "#FF9F1C",
    "#2EC4B6", "#E71D36", "#7768AE", "#F4A261", "#264653"
]

QUOTES = [
    "生活不是等待暴风雨过去，而是学会在雨中跳舞。",
    "你今天的努力，是幸运的伏笔。",
    "所有的光芒，都需要时间才能被看到。",
    "心之所向，素履以往；生如逆旅，一苇以航。",
    "不要害怕走得慢，只要方向是对的。",
    "世界很大，请带着梦想一起奔跑。"
]


def generate_name():
    """随机生成一个名字"""
    return random.choice(NAMES) + "·" + random.choice(NAMES)


def generate_story():
    """随机生成一个故事"""
    name = random.choice(NAMES)
    return random.choice(STORIES).format(name=name)


def generate_color():
    """随机生成一个颜色"""
    return random.choice(COLORS)


def generate_quote():
    """随机生成一句名言"""
    return random.choice(QUOTES)


def generate_password(length=16):
    """随机生成一个密码"""
    chars = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789!@#$%^&*"
    return "".join(random.choice(chars) for _ in range(length))


@app.route("/")
def index():
    """首页"""
    return render_template("index.html")


@app.route("/api/generate/<kind>")
def generate(kind):
    """根据类型返回随机生成结果"""
    generators = {
        "name": generate_name,
        "story": generate_story,
        "color": generate_color,
        "quote": generate_quote,
        "password": generate_password
    }

    if kind not in generators:
        return jsonify({"error": "未知的生成类型"}), 400

    result = generators[kind]()
    return jsonify({"type": kind, "result": result})


if __name__ == "__main__":
    app.run(debug=True)