#!/usr/bin/env python3

"""
Filters/Prints record objects


faiyaza at cs dot princeton dot edu
Copyright (c) 2009 Board of Trustees, Princeton University

"""

import sys
import os
from optparse import OptionParser
from pprint import pprint
from xml.parsers.expat import ExpatError
from sfa.util.xml import XML


def create_parser():
    command = sys.argv[0]
    argv = sys.argv[1:]
    usage = "%(command)s [options]" % locals()
    description = """getRecord will parse a supplied (via stdin) record and print all values or key/values, and filter results based on a given key or set of keys."""
    parser = OptionParser(usage=usage, description=description)
    parser.add_option("-d", "--debug", dest="DEBUG", action="store_true",
                      default=False, help="record file path")
    parser.add_option("-k", "--key", dest="withkey", action="store_true",
                      default=False, help="print SSH keys and certificates")
    parser.add_option("-p", "--plinfo", dest="plinfo", action="store_true",
                      default=False, help="print PlanetLab specific internal fields")

    return parser


def printRec(record_dict, filters, options):
    line = ""
    if len(filters):
        for filter in filters:
            if options.DEBUG:
                print("Filtering on %s" % filter)
            line += "%s: %s\n" % (filter,
                                  printVal(record_dict.get(filter, None)))
        print(line)
    else:
        # print the wole thing
        for (key, value) in record_dict.items():
            if (not options.withkey and key in ('gid', 'keys')) or\
                    (not options.plinfo and key == 'pl_info'):
                continue
            line += "%s: %s\n" % (key, printVal(value))
        print(line)


# fix the iteratable values
def printVal(value):
    line = ""
    if type(value) in (tuple, list):
        for i in value:
            line += "%s " % i
    elif value != None:
        line += value
    return line.rstrip("\n")


def main():
    parser = create_parser()
    (options, args) = parser.parse_args()

    stdin = sys.stdin.read()

    record = XML(stdin)
    record_dict = record.todict()

    if options.DEBUG:
        pprint(record.toxml())
        print("#####################################################")

    printRec(record_dict, args, options)

if __name__ == '__main__':
    try:
        main()
    except ExpatError as e:
        print("RecordError.  Is your record valid XML?")
        print(e)
    except Exception as e:
        print(e)