from . import db
from flask_login import UserMixin
from datetime import datetime

String_size = db.String(255)

class User(UserMixin, db.Model):
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(String_size, unique=True, nullable=False)
    email = db.Column(String_size, unique=True, nullable=False)
    password_hash = db.Column(String_size, nullable=False)
    created_at = db.Column(db.DateTime, default=datetime.utcnow)

    def set_password(self, password):
        from werkzeug.security import generate_password_hash
        self.password_hash = generate_password_hash(password)

    def check_password(self, password):
        from werkzeug.security import check_password_hash
        return check_password_hash(self.password_hash, password)

    def __repr__(self):
        return f'<User {self.username}>'

# ==========================
# Your existing models below
# ==========================

class Task(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    district = db.Column(String_size)
    block_name = db.Column(String_size)
    village_name = db.Column(String_size)
    serial_number = db.Column(String_size)
    parent_id = db.Column(String_size, nullable=True)
    parent_task_name = db.Column(db.Text)
    task_name = db.Column(db.Text)
    unit = db.Column(String_size)
    qty = db.Column(String_size)
    rate = db.Column(String_size)
    boq_amount = db.Column(String_size)
    previous_billed_qty = db.Column(String_size)
    previous_billing_amount = db.Column(String_size)
    remaining_amount = db.Column(String_size)
    in_this_ra_bill_qty = db.Column(String_size)
    in_this_ra_billing_amount = db.Column(String_size)
    cumulative_billed_qty = db.Column(String_size)
    cumulative_billed_amount = db.Column(String_size)
    variation_qty = db.Column(String_size)
    variation_amount = db.Column(String_size)
    remark = db.Column(String_size, nullable=True)
    uploaded_at = db.Column(db.DateTime, default=datetime.utcnow)

class WorkDetail(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    name_of_work = db.Column(String_size)
    cover_agreement_no = db.Column(String_size)
    name_of_contractor = db.Column(String_size)
    name_of_tpi_agency = db.Column(String_size)
    name_of_division = db.Column(String_size)
    name_of_village = db.Column(String_size)
    block = db.Column(String_size)
    scheme_id = db.Column(String_size)
    measurement_book = db.Column(String_size)
    date_of_billing = db.Column(String_size)
    uploaded_at = db.Column(db.DateTime, default=datetime.utcnow)
    district = db.Column(db.String(255))

class ActivityLog(db.Model):  # ✅ OUTSIDE WorkDetail
    id = db.Column(db.Integer, primary_key=True)
    user = db.Column(db.String(100))  # can link with User model later
    action = db.Column(db.String(255))
    details = db.Column(db.Text)
    timestamp = db.Column(db.DateTime, default=datetime.utcnow)