import re
import django.utils.html
from django.shortcuts import render_to_response as render
from ietf.idtracker.models import IETFWG, InternetDraft, Rfc
from ietf.ipr.models import IprRfc, IprDraft, IprDetail
from ietf.ipr.related import related_docs
from ietf.utils import log


def search(request, type="", q="", id=""):
    wgs = IETFWG.objects.filter(group_type__group_type_id=1).exclude(group_acronym__acronym='2000').select_related().order_by('acronym.acronym')
    args = request.REQUEST.items()
    if args:
        for key, value in args:
            if key == "option":
                type = value
            if re.match(".*search", key):
                q = value
            if re.match(".*id", key):
                id = value
        if type and q or id:
            log("Got query: type=%s, q=%s, id=%s" % (type, q, id))

            # Search by RFC number or draft-identifier
            if type in ["document_search", "rfc_search"]:
                if type == "document_search":
                    if q:
                        start = InternetDraft.objects.filter(filename__contains=q)
                    if id:
                        start = InternetDraft.objects.filter(id_document_tag=id)
                if type == "rfc_search":
                    if q:
                        start = Rfc.objects.filter(rfc_number=q)
                if start.count() == 1:
                    first = start[0]
                    # get all related drafts, then search for IPRs on all

                    docs = related_docs(first, [])
                    #docs = get_doclist.get_doclist(first)
                    iprs = []
                    for doc in docs:
                        if isinstance(doc, InternetDraft):
                            disclosures = [ item.ipr for item in IprDraft.objects.filter(document=doc, ipr__status__in=[1,3]) ]
                        elif isinstance(doc, Rfc):
                            disclosures = [ item.ipr for item in IprRfc.objects.filter(document=doc, ipr__status__in=[1,3]) ]
                        else:
                            raise ValueError("Doc type is neither draft nor rfc: %s" % doc)
                        if disclosures:
                            doc.iprs = disclosures
                            iprs += disclosures
                    iprs = list(set(iprs))
                    return render("ipr/search_doc_result.html", {"q": q, "first": first, "iprs": iprs, "docs": docs})
                elif start.count():
                    return render("ipr/search_doc_list.html", {"q": q, "docs": start })
                else:
                    raise ValueError("Missing or malformed search parameters, or internal error")

            # Search by legal name
            elif type == "patent_search":
                iprs = IprDetail.objects.filter(legal_name__icontains=q, status__in=[1,3]).order_by("-submitted_date", "-ipr_id")
                count = iprs.count()
                iprs = [ ipr for ipr in iprs if not ipr.updated_by.all() ]
                # Some extra information, to help us render 'and' between the
                # last two documents in a sequence
                for ipr in iprs:
                    if ipr.drafts.count():
                        ipr.last_draft = ipr.drafts.all()[int(ipr.drafts.count())-1]
                    if ipr.rfcs.count():
                        ipr.last_rfc = ipr.rfcs.all()[int(ipr.rfcs.count())-1]
                return render("ipr/search_holder_result.html", {"q": q, "iprs": iprs, "count": count } )

            # Search by content of email or pagent_info field
            elif type == "patent_info_search":
                pass

            # Search by wg acronym
            elif type == "wg_search":
                pass

            # Search by rfc and id title
            elif type == "title_search":
                pass

            # Search by title of IPR disclosure
            elif type == "ipr_title_search":
                pass
            else:
                raise ValueError("Unexpected search type in IPR query: %s" % type)
        return django.http.HttpResponseRedirect(request.path)
    return render("ipr/search.html", {"wgs": wgs})