from flask_sqlalchemy import SQLAlchemy
from datetime import datetime
from enum import Enum

db = SQLAlchemy()


class UserRole(Enum):
    ADMIN = 'Admin'
    USER = 'User'


class User(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(80), nullable=False, unique=True)
    nickname = db.Column(db.String(80), nullable=True)
    password = db.Column(db.String(256), nullable=False)
    role = db.Column(db.Enum(UserRole), nullable=False, default=UserRole.USER)
    created_at = db.Column(db.DateTime, default=datetime.now)
    phone = db.Column(db.String(11), nullable=True)
    email = db.Column(db.String(120), nullable=True)
    qq = db.Column(db.String(10), nullable=True)


class PaperPassagesRelation(db.Model):
    paper_id = db.Column(db.Integer, db.ForeignKey('paper.id'), nullable=False, primary_key=True)
    passage_id = db.Column(db.Integer, db.ForeignKey('passage.id'), nullable=False, primary_key=True)


class PendingPassage(db.Model):
    # 基本信息
    id = db.Column(db.Integer, primary_key=True)
    sender = db.relationship('User', foreign_keys='PendingPassage.sender_id')
    sender_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
    created_at = db.Column(db.DateTime, default=datetime.now)

    # 内容
    title = db.Column(db.String(200), nullable=False)
    content = db.Column(db.Text, nullable=False)
    image_url = db.Column(db.String(500), nullable=True)
    authors = db.Column(db.String(60), nullable=False)
    is_outlink = db.Column(db.Boolean, default=False)


class Passage(db.Model):
    # 基本信息
    id = db.Column(db.Integer, primary_key=True)
    sender = db.relationship('User', foreign_keys='Passage.sender_id')
    sender_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
    created_at = db.Column(db.DateTime, default=datetime.now)

    # 审核
    processed_at = db.Column(db.DateTime, nullable=True)
    processor = db.relationship('User', foreign_keys='Passage.processor_id')
    processor_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=True)
    is_posted = db.Column(db.Boolean, default=False)

    # 内容
    title = db.Column(db.String(200), nullable=False)
    content = db.Column(db.Text, nullable=False)
    image_url = db.Column(db.String(500), nullable=True)
    authors = db.Column(db.String(60), nullable=False)
    is_outlink = db.Column(db.Boolean, default=False)


class Paper(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    title = db.Column(db.String(200), nullable=False)
    description = db.Column(db.Text, nullable=False)
    image_url = db.Column(db.String(500), nullable=True)
    created_at = db.Column(db.DateTime, default=datetime.now)
    passages = db.relationship('Passage', secondary='paper_passages_relation', backref=db.backref('papers', lazy=True))