# -*- coding: utf-8 -*-

import json
from datetime import datetime

from elasticsearch import Elasticsearch

from libnmap.plugins.backendplugin import NmapBackendPlugin
from libnmap.reportjson import ReportEncoder


class NmapElasticsearchPlugin(NmapBackendPlugin):
    """
        This class enables the user to store and manipulate nmap reports \
        in a elastic search db.
    """

    def __init__(self, index=None):
        if index is None:
            self.index = "nmap.{0}".format(datetime.now().strftime("%Y-%m-%d"))
        else:
            self.index = index
        self._esapi = Elasticsearch()

    def insert(self, report, doc_type=None):
        """
        insert NmapReport in the backend
        :param NmapReport:
        :return: str the ident of the object in the backend for
        future usage
        or None
        """
        if doc_type is None:
            doc_type = "NmapReport"
        j = json.dumps(report, cls=ReportEncoder)
        res = self._esapi.index(
            index=self.index, doc_type=doc_type, body=json.loads(j)
        )
        rc = res["_id"]
        return rc

    def delete(self, id):
        """
        delete NmapReport if the backend
        :param id: str
        """
        raise NotImplementedError

    def get(self, id):
        """
        retrieve a NmapReport from the backend
        :param id: str
        :return: NmapReport
        """
        res = self._esapi.get(index=self.index, doc_type="NmapReport", id=id)
        rc = res["_source"]
        return rc

    def getall(self, filter=None):
        """
        :return: collection of tuple (id,NmapReport)
        :param filter: Nice to have implement a filter capability
        """
        rsearch = self._esapi.search(
            index=self.index, body={"query": {"match_all": {}}}
        )
        print("--------------------")
        print(type(rsearch))
        print(rsearch)
        print("------------")