#
# Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies).
# All rights reserved.
# This component and the accompanying materials are made available
# under the terms of "Eclipse Public License v1.0"
# which accompanies this distribution, and is available
# at the URL "http://www.eclipse.org/legal/epl-v10.html".
#
# Initial Contributors:
# Nokia Corporation - initial contribution.
#
# Contributors:
#
# Description: 
#

import re, sys, glob, os.path
from optparse import OptionParser

# When defreezing, these entries can exist, but only at the beginning of the exports. 
# Anything after these will be defrozen.
ALLOWEDENTRIES = set(['_Z10jni_lookupPKc', '_Z13findDllMethodPKc'])

# Pattern for entries
RE_ENTRY = re.compile(r"^\s+([^ ]+) +@ +\d+ +NONAME( +ABSENT)?( *;.*)?$")

# Transient exports which will be marked as absent.
# See http://www.codesourcery.com/public/cxx-abi/abi.html#mangling for detailed 
# de-mangling instructions
TRANSIENTS = set([
    # TI = typeinfo structures
    '_ZTISi', # Si == ::std::basic_istream<char,  std::char_traits<char> >
    '_ZTISo', # So == ::std::basic_ostream<char,  std::char_traits<char> >
    '_ZTISd', # Sd == ::std::basic_iostream<char, std::char_traits<char> >
    '_ZTISt13basic_istreamIwSt11char_traitsIwEE',
    '_ZTISt13basic_ostreamIwSt11char_traitsIwEE',
    '_ZTISt14basic_iostreamIwSt11char_traitsIwEE',
    '_ZTISt15basic_streambufIwSt11char_traitsIwEE',
    '_ZTISt15basic_streambufIcSt11char_traitsIcEE',
    '_ZTISt13basic_filebufIcSt11char_traitsIcEE',
    '_ZTISt15basic_stringbufIcSt11char_traitsIcESaIcEE',
    '_ZTISt15basic_stringbufIwSt11char_traitsIwESaIwEE',
    '_ZTISt18basic_stringstreamIcSt11char_traitsIcESaIcEE',
    '_ZTISt18basic_stringstreamIwSt11char_traitsIwESaIwEE',
    '_ZTISt19basic_ostringstreamIcSt11char_traitsIcESaIcEE',
    '_ZTISt19basic_ostringstreamIwSt11char_traitsIwESaIwEE',
    '_ZTISt19basic_istringstreamIcSt11char_traitsIcESaIcEE',
    '_ZTISt19basic_istringstreamIwSt11char_traitsIwESaIwEE',

    # TV = virtual tables
    '_ZTVSi',
    '_ZTVSo',
    '_ZTVSd',
    '_ZTVSt13basic_istreamIwSt11char_traitsIwEE',
    '_ZTVSt13basic_ostreamIwSt11char_traitsIwEE',
    '_ZTVSt14basic_iostreamIwSt11char_traitsIwEE',
    '_ZTVSt15basic_streambufIwSt11char_traitsIwEE',
    '_ZTVSt15basic_streambufIcSt11char_traitsIcEE',
    '_ZTVSt13basic_filebufIcSt11char_traitsIcEE',
    '_ZTVSt15basic_stringbufIcSt11char_traitsIcESaIcEE',
    '_ZTVSt15basic_stringbufIwSt11char_traitsIwESaIwEE',
    '_ZTVSt18basic_stringstreamIcSt11char_traitsIcESaIcEE',
    '_ZTVSt18basic_stringstreamIwSt11char_traitsIwESaIwEE',
    '_ZTVSt19basic_ostringstreamIcSt11char_traitsIcESaIcEE',
    '_ZTVSt19basic_ostringstreamIwSt11char_traitsIwESaIwEE',
    '_ZTVSt19basic_istringstreamIcSt11char_traitsIcESaIcEE',
    '_ZTVSt19basic_istringstreamIwSt11char_traitsIwESaIwEE',

    # VTT structure (construction vtable index)
    '_ZTTSd',
    '_ZTTSi',
    '_ZTTSo',
    '_ZTTSt13basic_istreamIwSt11char_traitsIwEE',
    '_ZTTSt13basic_ostreamIwSt11char_traitsIwEE',
    '_ZTTSt14basic_ifstreamIcSt11char_traitsIcEE',
    '_ZTTSt14basic_iostreamIwSt11char_traitsIwEE',
    '_ZTTSt14basic_ofstreamIcSt11char_traitsIcEE',
    '_ZTTSt18basic_stringstreamIcSt11char_traitsIcESaIcEE',
    '_ZTTSt18basic_stringstreamIwSt11char_traitsIwESaIwEE',
    '_ZTTSt19basic_ostringstreamIcSt11char_traitsIcESaIcEE',
    '_ZTTSt19basic_ostringstreamIwSt11char_traitsIwESaIwEE',
    '_ZTTSt19basic_istringstreamIcSt11char_traitsIcESaIcEE',
    '_ZTTSt19basic_istringstreamIwSt11char_traitsIwESaIwEE'
])

# These are produced only by 9.2, not by 5.0
TRANSIENTS_92 = set([
    '_ZTTSd',
    '_ZTTSi',
    '_ZTTSo',
    '_ZTTSt13basic_istreamIwSt11char_traitsIwEE',
    '_ZTTSt13basic_ostreamIwSt11char_traitsIwEE',
    '_ZTTSt14basic_ifstreamIcSt11char_traitsIcEE',
    '_ZTTSt14basic_iostreamIwSt11char_traitsIwEE',
    '_ZTTSt14basic_ofstreamIcSt11char_traitsIcEE'
])

# Only produced by stl4 - not by stl5
TRANSIENTS_STL4 = set([
    '_ZTISi',
    '_ZTISt13basic_istreamIwSt11char_traitsIwEE',
    '_ZTVSi',
    '_ZTVSt13basic_istreamIwSt11char_traitsIwEE'
])


# Only produced by stl5
TRANSIENTS_STL5 = set([
    '_ZTISt15basic_streambufIcSt11char_traitsIcEE',
    '_ZTISt15basic_stringbufIcSt11char_traitsIcESaIcEE',
    '_ZTISt15basic_stringbufIwSt11char_traitsIwESaIwEE',
    '_ZTISt18basic_stringstreamIcSt11char_traitsIcESaIcEE',
    '_ZTISt18basic_stringstreamIwSt11char_traitsIwESaIwEE',
    '_ZTISt19basic_ostringstreamIcSt11char_traitsIcESaIcEE',
    '_ZTISt19basic_ostringstreamIwSt11char_traitsIwESaIwEE',
    '_ZTISt19basic_istringstreamIcSt11char_traitsIcESaIcEE',
    '_ZTISt19basic_istringstreamIwSt11char_traitsIwESaIwEE',

    '_ZTTSt18basic_stringstreamIcSt11char_traitsIcESaIcEE',
    '_ZTTSt18basic_stringstreamIwSt11char_traitsIwESaIwEE',
    '_ZTTSt19basic_ostringstreamIcSt11char_traitsIcESaIcEE',
    '_ZTTSt19basic_ostringstreamIwSt11char_traitsIwESaIwEE',
    '_ZTTSt19basic_istringstreamIcSt11char_traitsIcESaIcEE',
    '_ZTTSt19basic_istringstreamIwSt11char_traitsIwESaIwEE'

    '_ZTVSt15basic_streambufIcSt11char_traitsIcEE',
    '_ZTVSt15basic_stringbufIcSt11char_traitsIcESaIcEE',
    '_ZTVSt15basic_stringbufIwSt11char_traitsIwESaIwEE',
    '_ZTVSt18basic_stringstreamIcSt11char_traitsIcESaIcEE',
    '_ZTVSt18basic_stringstreamIwSt11char_traitsIwESaIwEE',
    '_ZTVSt19basic_ostringstreamIcSt11char_traitsIcESaIcEE',
    '_ZTVSt19basic_ostringstreamIwSt11char_traitsIwESaIwEE',
    '_ZTVSt19basic_istringstreamIcSt11char_traitsIcESaIcEE',
    '_ZTVSt19basic_istringstreamIwSt11char_traitsIwESaIwEE'
])

TRANSIENTS_SPECIFIC = TRANSIENTS_92 | TRANSIENTS_STL4 | TRANSIENTS_STL5
TRANSIENTS_COMMON = TRANSIENTS - TRANSIENTS_SPECIFIC

def readExports(file):
    lines = [l.rstrip() for l in open(file).readlines()]
    if lines.pop(0) != "EXPORTS":
        raise Exception("File %s is not a deffile" % file)

    exports = []
    
    for l in lines:
        # Skip empty lines
        if len(l.strip()) == 0:
            continue

        # Read lines
        match = RE_ENTRY.match(l)
        if not match:
            raise Exception("Unrecognized line on file %s: %s" % (file, l))

        (export, absent, comment) = match.groups()
        exports.append((export, absent, comment))

        
    return exports

def movelastTransients(exports):
    return  [e for e in exports if e[0] not in TRANSIENTS] + \
            [e for e in exports if e[0] in TRANSIENTS]

def removeSpecific(exports):
    return  [e for e in exports if e[0] not in TRANSIENTS_SPECIFIC]

def absentTransients(exports):
    newExports = []
    for e in exports:
        if e[0] in TRANSIENTS:
            e = (e[0], " ABSENT", e[2])
        newExports.append(e)
    return newExports

def removeTransients(exports):
    return  [e for e in exports if e[0] not in TRANSIENTS]

def defreeze(exports):
    defrozen = []
    for export, absent, comment in exports:
        if export in ALLOWEDENTRIES:
            defrozen.append((export, absent, comment))
        else:
            break
    return defrozen

def generate(exports):
    output = "EXPORTS\n"

    for index in range(len(exports)):
        (export, absent, comment) = exports[index]
        output = output + "\t%s @ %i NONAME%s%s\n" % (
            export, index + 1, absent or "", comment or "")
    return output + "\n"

def list(exports):
    return "\n    ".join(["'%s'," % e[0] for e in exports])


def main():
    parser = OptionParser()
    parser.add_option("-f", dest = "outputFile")
    parser.add_option("-d", dest = "outputDir")
    parser.add_option("-o", dest = "overwrite",
                      action = "store_true", default = False)
    parser.add_option("--reorder", dest = "reorder",
                      action = "store_true", default = False,
                      help = "reorder transient exports to last")
    parser.add_option("--removeSpecific", dest = "removeSpecific",
                      action = "store_true", default = False,
                      help = "remove exports which are specific to some environment")
    parser.add_option("--removeTransients", dest = "removeTransients",
                      action = "store_true", default = False,
                      help = "remove all exports which are transients")
    parser.add_option("--absentify", dest = "absentify",
                      action = "store_true", default = False,
                      help = "make absent transients exports")
    parser.add_option("--defreeze", dest = "defreeze",
                      action = "store_true", default = False,
                      help = "defreeze all except configured exports")
    parser.add_option("--list", dest = "list",
                      action = "store_true", default = False)
    (options, args) = parser.parse_args()

    # Read
    for file in sum([glob.glob(arg) for arg in args], []):
        exports = readExports(file)

        # Modify
        if options.reorder:
            exports = movelastTransients(exports)
        if options.removeSpecific:
            exports = removeSpecific(exports)
        if options.absentify:
            exports = absentTransients(exports)
        if options.defreeze:
            exports = defreeze(exports)
        if options.removeTransients:
            exports = removeTransients(exports)

        # Generate / list
        if options.list:
            output = list(exports)
        else:
            output = generate(exports)

        # Write / print
        if options.outputFile:
            open(options.outputFile, "w").write(output)
        elif options.outputDir:
            open(os.path.join(options.outputDir, os.path.basename(file)), 
                 "w").write(output)
        elif options.overwrite:
            open(file, "w").write(output)
            print "Modified %s" % file
        else:
            print output        

if __name__ == "__main__":
    main()