#!/usr/bin/python

import os

licenses = ['BSD', 'FDL', 'LGPL']

lfiles = []
for root, subdirs, files in os.walk(os.curdir):
    for sfile in files:
        if sfile.endswith(('.cpp', '.h', '.js', '.qs', '.qml', '.ui', \
            '.g', '.sh', '.cmake', '.pod', '.qdoc', '.c', '.xsl', '.xml', \
            '.cmake')):
            lfiles.append('%s/%s' % (root, sfile))

def readlicense(sfile, replacedashes, replacehashes):
    sheader = ''
    with open(sfile, 'r') as f:
        shouldappend = False
        for sline in f.readlines():
            if 'QT_BEGIN_LICENSE' in sline:
                shouldappend = True
            elif 'QT_END_LICENSE' in sline:
                sheader = '%s%s' % (sheader, sline)
                shouldappend = False
            if shouldappend:
                sheader = '%s%s' % (sheader, sline)
        if replacedashes:
            sheader = sheader.replace('**', '--')
        if replacehashes:
            sheader = sheader.replace('**', '##')
    return sheader

for sfile in lfiles:
    with open(sfile, 'r') as f:
        scontent = f.read()
    for license in licenses:
        if ('$QT_BEGIN_LICENSE:%s$' % license) in scontent:
            snewheader = readlicense('%s/header.%s' % (os.curdir, license),
                sfile.endswith('.g'), sfile.endswith('.sh'))
            soldheader = readlicense(sfile, False, False)
            snewcontent = scontent.replace(soldheader, snewheader)
            if not snewcontent == scontent:
                with open(sfile, 'w') as f:
                    f.write(snewcontent)