from flask import Blueprint, render_template, redirect, url_for, session, flash
from .models import DeliveryOrder

main = Blueprint('main', __name__)


@main.route('/')
def home():
    return """
        <h1>DO Tracker Online</h1>
        <p><a href='/login'>Login as Store</a></p>
        <p><a href='/store'>Store Dashboard</a></p>
    """


@main.route('/store')
def store_dashboard():
    if 'store_id' not in session:
        flash("You must be logged in to access the store dashboard.", "warning")
        return redirect(url_for('auth.login'))

    store_name = session.get('store_name')

    return f"""
        <h1>Welcome, {store_name}</h1>
        <p><a href='/logout'>Logout</a></p>
        <ul>
            <li><a href='/store/do-entry'>Enter New DO</a> (coming next)</li>
            <li><a href='/store/track'>Track DOs</a> (coming soon)</li>
        </ul>
    """