#!/usr/bin/env python
# -*- coding: utf-8 -*-

import json

from libnmap.objects import NmapHost, NmapReport, NmapService
from libnmap.objects.os import (
    CPE,
    NmapOSClass,
    NmapOSFingerprint,
    NmapOSMatch,
    OSFPPortUsed,
)
from libnmap.parser import NmapParser


class ReportEncoder(json.JSONEncoder):
    """
    ReportEncoder is a internal class used mostly by plugins to convert
    NmapReport objects in json format.
    e.g.:
        nmapreport_obj = NmapParser.parse_fromfile(
            "libnmap/test/files/1_hosts.xml"
        )
        nmapreport_json = json.dumps(nmapreport_obj, cls=ReportEncoder)
    """

    def default(self, obj):
        otype = {
            "NmapHost": NmapHost,
            "NmapOSFingerprint": NmapOSFingerprint,
            "NmapOSMatch": NmapOSMatch,
            "NmapOSClass": NmapOSClass,
            "CPE": CPE,
            "OSFPPortUsed": OSFPPortUsed,
            "NmapService": NmapService,
            "NmapReport": NmapReport,
        }
        if isinstance(obj, tuple(otype.values())):
            key = ("__{0}__").format(obj.__class__.__name__)
            return {key: obj.__dict__}
        return json.JSONEncoder.default(self, obj)


class ReportDecoder(json.JSONDecoder):
    """
    ReportDecoder is a internal class used mostly by plugins to convert
    json nmap report in to NmapReport objects.
    e.g.:
        nmap_report_obj = json.loads(nmap_report_json, cls=ReportDecoder)
    """

    def decode(self, json_str):
        r = NmapParser.parse_fromdict(json.loads(json_str))
        return r