from flask import Blueprint, render_template, request

import crudkit

from crudkit.ui.fragments import render_table

from ..models import Inventory, User, WorkLog

bp_search = Blueprint("search", __name__)

def init_search_routes(app):
    @bp_search.get("/search")
    def search():
        q = request.args.get("q")
        if not q:
            return "Oh no!"

        inventory_service = crudkit.crud.get_service(Inventory)
        user_service = crudkit.crud.get_service(User)
        worklog_service = crudkit.crud.get_service(WorkLog)

        inventory_columns = [
            {"field": "label"},
            {"field": "barcode", "label": "Bar Code #"},
            {"field": "name"},
            {"field": "serial", "label": "Serial #"},
            {"field": "brand.name", "label": "Brand"},
            {"field": "model"},
            {"field": "device_type.description", "label": "Device Type"},
            {"field": "owner.label", "label": "Contact",
             "link": {"endpoint": "entry.entry", "params": {"id": "{owner.id}", "model": "user"}}},
            {"field": "location.label", "label": "Location"},
        ]
        inventory_results = inventory_service.list({
            'notes|label|owner.label__icontains': q,
            'fields': [
                "label",
                "name",
                "barcode",
                "serial",
                "brand.name",
                "model",
                "device_type.description",
                "owner.label",
                "location.label",
            ]
        })

        user_columns = [
            {"field": "last_name"},
            {"field": "first_name"},
            {"field": "title"},
            {"field": "supervisor.label", "label": "Supervisor",
             "link": {"endpoint": "entry.entry", "params": {"id": "{supervisor.id}", "model": "user"}}},
            {"field": "location.label", "label": "Location"},
        ]
        user_results = user_service.list({
            'supervisor.label|label__icontains': q,
            'fields': [
                "last_name",
                "first_name",
                "title",
                "supervisor.label",
                "location.label",
            ]
        })

        worklog_columns = [
            {"field": "contact.label", "label": "Contact",
             "link": {"endpoint": "entry.entry", "params": {"id": "{contact.id}", "model": "user"}}},
            {"field": "work_item.label", "label": "Work Item",
             "link": {"endpoint": "entry.entry", "params": {"id": "{work_item.id}", "model": "inventory"}}},
            {"field": "complete", "format": "yesno"},
            {"field": "start_time", "format": "datetime"},
            {"field": "end_time", "format": "datetime"},
            {"field": "updates", "format": lambda x: len(x)},
        ]
        worklog_results = worklog_service.list({
            'contact.label|work_item.label|updates.content__icontains': q,
            'fields': [
                "contact.label",
                "work_item.label",
                "complete",
                "start_time",
                "end_time",
                "updates",
            ]
        })

        inventory_results = render_table(inventory_results, inventory_columns, opts={"object_class": "inventory"})
        user_results = render_table(user_results, user_columns, opts={"object_class": "user"})
        worklog_results = render_table(worklog_results, worklog_columns, opts={"object_class": "worklog"})

        return render_template('search.html', q=q, inventory_results=inventory_results, user_results=user_results, worklog_results=worklog_results)

    app.register_blueprint(bp_search)