#include <boost/foreach.hpp>

#include <dns/rdataclass.h>
#include <dns/rrttl.h>

#include "data_source_plot.h"

namespace isc {
namespace dns {

// this implementation returns fixed records,
// and does not allow update statements

using namespace isc::dns;
using namespace isc::dns::rdata;
using namespace isc::data;

namespace {
const Name authors_name("authors.bind");
const Name version_name("version.bind");
}

void
DataSourceParkingLot::serve(std::string zone_name) {
    zones.serve(zone_name);
}

void
DataSourceParkingLot::addARecord(std::string data) {
    a_records.push_back(RdataPtr(new in::A(data)));
}

void
DataSourceParkingLot::addAAAARecord(std::string data) {
    aaaa_records.push_back(RdataPtr(new in::AAAA(data)));
}

void
DataSourceParkingLot::addNSRecord(std::string data) {
    ns_records.push_back(RdataPtr(new generic::NS(data)));
}

void
DataSourceParkingLot::setSOARecord(RdataPtr soa_record) {
}

void
DataSourceParkingLot::setDefaultZoneData() {
    clearARecords();
    clearAAAARecords();
    clearNSRecords();

    addARecord("127.0.0.1");
    addAAAARecord("::1");
    addNSRecord("ns1.parking.example");
    addNSRecord("ns2.parking.example");
    addNSRecord("ns3.parking.example");
}

DataSourceParkingLot::DataSourceParkingLot() {
    setDefaultZoneData();
    soa = RdataPtr(new generic::SOA(Name("parking.example"),
                                    Name("noc.parking.example"),
                                    1, 1800, 900, 604800, 86400));
}

bool
DataSourceParkingLot::hasZoneFor(const Name& name, Name &zone_name)
{
    if (name == authors_name) {
        zone_name = authors_name;
        return (true);
    } else if (name == version_name) {
        zone_name = version_name;
        return (true);
    }
    return zones.findClosest(name, zone_name);
}

SearchResult
DataSourceParkingLot::findRRsets(const isc::dns::Name& zone_name,
                                 const isc::dns::Name& name,
                                 const isc::dns::RRClass& clas,
                                 const isc::dns::RRType& type) const
{
    SearchResult result;
    
    if (clas == RRClass::CH()) {
        if (type == RRType::TXT()) {
            if (name == authors_name) {
                RRsetPtr rrset = RRsetPtr(new RRset(authors_name, RRClass::CH(),
                                                    RRType::TXT(), RRTTL(0)));
                rrset->addRdata(generic::TXT("Han Feng"));
                rrset->addRdata(generic::TXT("Kazunori Fujiwara"));
                rrset->addRdata(generic::TXT("Michael Graff"));
                rrset->addRdata(generic::TXT("Evan Hunt"));
                rrset->addRdata(generic::TXT("Jelte Jansen"));
                rrset->addRdata(generic::TXT("Jin Jian"));
                rrset->addRdata(generic::TXT("JINMEI Tatuya"));
                rrset->addRdata(generic::TXT("Naoki Kambe"));
                rrset->addRdata(generic::TXT("Shane Kerr")); 
                rrset->addRdata(generic::TXT("Zhang Likun"));
                rrset->addRdata(generic::TXT("Jeremy C. Reed")); 

                result.addRRset(rrset);
                result.setStatus(SearchResult::success);
            } else if (name == version_name) {
                RRsetPtr rrset = RRsetPtr(new RRset(version_name, RRClass::CH(),
                                                    RRType::TXT(), RRTTL(0)));
                rrset->addRdata(generic::TXT("BIND10 0.0.1"));
                result.addRRset(rrset);
                result.setStatus(SearchResult::success);
            } else {
                result.setStatus(SearchResult::name_not_found);
            }
        } else if (type == RRType::NS()) {
            if (name == authors_name || name == version_name) {
                RRsetPtr rrset = RRsetPtr(new RRset(name, RRClass::CH(),
                                                    RRType::NS(),
                                                    RRTTL(0)));
                rrset->addRdata(generic::NS(name));
                result.addRRset(rrset);
                result.setStatus(SearchResult::success);
            } else {
                result.setStatus(SearchResult::name_not_found);
            }
        } else {
            result.setStatus(SearchResult::name_not_found);
        }
    } else if (clas == RRClass::IN()) {
        if (zones.contains(name)) {
            RRsetPtr rrset = RRsetPtr(new RRset(name, clas, type, RRTTL(3600)));
            result.setStatus(SearchResult::success);
            if (type == RRType::A()) {
                BOOST_FOREACH(RdataPtr a, a_records) {
                    rrset->addRdata(a);
                }
            } else if (type == RRType::AAAA()) {
                BOOST_FOREACH(RdataPtr aaaa, aaaa_records) {
                    rrset->addRdata(aaaa);
                }
            } else if (type == RRType::NS()) {
                BOOST_FOREACH(RdataPtr ns, ns_records) {
                    rrset->addRdata(ns);
                }
            } else if (type == RRType::SOA()) {
                rrset->addRdata(soa);
            }
            result.addRRset(rrset);
        } else {
            // we don't have the name itself. Do we have the zone?
            if (zones.contains(zone_name)) {
                result.setStatus(SearchResult::name_not_found);
            } else {
                result.setStatus(SearchResult::zone_not_found);
            }
        }
    } else {
        result.setStatus(SearchResult::zone_not_found);
    }
    return result;
}

/// Do direct 'search' in database, no extra processing,
/// and add the resulting rrsets to the specified section
/// in the given message
/// returns the status code of the searchresult
/// Once the dns logic is moved from parkinglot to this class,
/// we should probably make this private
SearchResult::status_type
DataSourceParkingLot::addToMessage(Message& msg,
             const Section& section,
             const Name& zone_name,
             const Name& name,
             const RRClass& clas,
             const RRType& type) const
{
    SearchResult result = findRRsets(zone_name, name, clas, type);
    BOOST_FOREACH(RRsetPtr rrset, result) {
        msg.addRRset(section, rrset);
    }
    return result.getStatus();
}


}
}