#!/usr/bin/env python

# A DiGIR client

import os

import os.path

import sys

import urllib

import urllib2

sys.path.append(os.path.dirname(__file__)+"/../lib")

import dates

import opts

import streams

import util

# Config

timeout = 20 # sec

default_schema = 'http://digir.net/schema/conceptual/darwin/full/2003/1.0/darwin2full.xsd'

request_xml_template = '''\

<?xml version="1.0" encoding="UTF-8"?>

<request

    xmlns="http://digir.net/schema/protocol/2003/1.0"

    xmlns:xsd="http://www.w3.org/2001/XMLSchema"

    xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"

    xmlns:digir="http://digir.net/schema/protocol/2003/1.0"

    xmlns:darwin="http://digir.net/schema/conceptual/darwin/2003/1.0"

    xmlns:dwc="http://digir.net/schema/conceptual/darwin/2003/1.0"

    xsi:schemaLocation="http://digir.net/schema/protocol/2003/1.0 

      http://digir.sourceforge.net/schema/protocol/2003/1.0/digir.xsd 

      http://digir.net/schema/conceptual/darwin/2003/1.0 

      http://digir.sourceforge.net/schema/conceptual/darwin/2003/1.0/darwin2.xsd">

    <header>

        <version>1.0</version>

        <sendTime>[time]</sendTime>

        <source>[source]</source>

        <destination resource="[resource]">[url]</destination>

        <type>search</type>

    </header>

    <search>

        <filter>

            <equals>

                <darwin:Kingdom>plantae</darwin:Kingdom>

            </equals>

        </filter>

        <records limit="[count]" start="[start]">

            <structure schemaLocation="[schema]"/>

        </records>

        <count>true</count>

    </search>

</request>

'''

diags_start = '<diagnostics>'

diags_end = '</diagnostics>'

class InputException(Exception): pass

def main():

    env_names = []

    def usage_err():

        raise SystemExit('Usage: '+opts.env_usage(env_names, True)+' '

            +sys.argv[0]+' 2>>log')

    # Get config from env vars

    url = opts.get_env_var('url', None, env_names)

    resource = opts.get_env_var('resource', None, env_names)

    schema = opts.get_env_var('schema', default_schema, env_names)

    start = util.cast(int, opts.get_env_var('start', 0, env_names))

    count = util.cast(int, opts.get_env_var('n', 1, env_names))

    debug = opts.env_flag('debug', False, env_names)

    if url == None or resource == None: usage_err()

    def clear_line(): sys.stderr.write('\n')

    log_indent = 0

    def log(msg, line_ending='\n'):

        sys.stderr.write(('    '*log_indent)+msg+line_ending)

    def debug_log(str_, label=None):

        if debug:

            if label != None: sys.stderr.write(label+':\n')

            sys.stderr.write(str_+'\n')

    self_dir = os.path.dirname(__file__)

    source = os.popen(self_dir+"/local_ip").read().strip()

    this_request_xml_template = (request_xml_template

        .replace('[source]', source)

        .replace('[url]', url)

        .replace('[resource]', resource)

        .replace('[schema]', schema)

        .replace('[count]', str(count))

        )

    time = dates.strftime('%Y-%m-%d %H:%M:%S %Z', dates.now())

    request_xml = (this_request_xml_template

        .replace('[start]', str(start))

        .replace('[time]', time)

        )

    debug_log(request_xml, 'request')

    this_url = url+'?'+urllib.urlencode({'request': request_xml})

    diag_capture = streams.CaptureStream(streams.TimeoutInputStream(

        urllib2.urlopen(this_url), timeout), diags_start, diags_end)

    stream = streams.StreamIter(diag_capture)

    try:

        # Copy lines

        for line in stream:

            sys.stdout.write(line)

    finally: # still run if break is called

        stream.close()

        debug_log(diag_capture.match, 'diagnostics')

main()