#!/usr/bin/env python

# This script is a workaround the RVCT compiler and linker crashes while compiling WebKit in debug

import os

import os.path

import re

# Do not aglomerate these directories

exceptionDirs = [

    '../JavaScriptCore/API',

    '../WebKit/qt/Api',

    '../WebKit/qt/WebCoreSupport',

    ]

# In these directories, create separate aglomerate files for sources matching each pattern

dirExtraSplits = {

    'generated': (# These splits are trial/error until armcc stoped reporting Internal faults...

                  re.compile(r'^JSHTML'),

                  re.compile(r'^JSCss'),

                  re.compile(r'^JSSVGAnimated'),

                  re.compile(r'^JSSVGFE'),

                  re.compile(r'^JSSVGPathSeg'),

                  re.compile(r'^JSSVG'),

                  re.compile(r'^JSD'),

                  re.compile(r'^JSX'),

                  re.compile(r'^JS'),

                  ),

    # Because of Token being defined twice

    'xml': (re.compile(r'^XSL'),)

    }

# Compile these files alone

exceptionFiles = [

    '../JavaScriptCore/wtf/CurrentTime.cpp',

    # Bacause of WebCore::finalize

    'bindings/js/JavaScriptProfileNode.cpp',

    # Because of WebCore::handleException

    'bindings/js/ScriptObject.cpp',

    # Because of JSType being defined in JSC/API and JSC/runtime

    'bindings/js/JavaScriptProfile.cpp',

    'bindings/js/ScriptValue.cpp',

    # Because FontFallbackList.h have expicitely no include gard

    'platform/graphics/FontFallbackList.cpp',

    'platform/graphics/Font.cpp',

    'platform/graphics/FontCache.cpp',

    # Stand-alone files with no documented reason

    'accessibility/qt/AccessibilityObjectQt.cpp',

    'bindings/js/ScriptControllerQt.cpp',

    'bridge/IdentifierRep.cpp',

    'bridge/NP_jsobject.cpp',

    'bridge/c/c_instance.cpp',

    'bridge/c/c_runtime.cpp',

    'bridge/c/c_utility.cpp',

    'bridge/qt/qt_class.cpp',

    'bridge/qt/qt_instance.cpp',

    'bridge/qt/qt_runtime.cpp',

    'css/FontFamilyValue.cpp',

    'css/MediaFeatureNames.cpp',

    'css/SVGCSSStyleSelector.cpp',

    'dom/NamedAttrMap.cpp',

    'dom/NamedMappedAttrMap.cpp',

    'dom/NamedNodeMap.cpp',

    'dom/Node.cpp',

    'dom/QualifiedName.cpp',

    'editing/qt/EditorQt.cpp',

    'generated/HTMLElementFactory.cpp',

    'generated/HTMLEntityNames.cpp',

    'generated/HTMLNames.cpp',

    'generated/JSElement.cpp',

    'generated/JSHTMLElementWrapperFactory.cpp',

    'generated/SVGNames.cpp',

    'generated/XLinkNames.cpp',

    'generated/XPathGrammar.cpp',

    'generated/XPathGrammar.tab.c',

    'history/BackForwardList.cpp',

    'history/CachedFrame.cpp',

    'history/CachedPage.cpp',

    'html/File.cpp',

    'html/FileList.cpp',

    'html/FormDataList.cpp',

    'html/HTMLAllCollection.cpp',

    'html/HTMLCollection.cpp',

    'html/HTMLDataListElement.cpp',

    'html/HTMLDocument.cpp',

    'html/HTMLFormCollection.cpp',

    'html/HTMLImageLoader.cpp',

    'html/HTMLNameCollection.cpp',

    'html/HTMLOptionsCollection.cpp',

    'html/HTMLParser.cpp',

    'html/HTMLParserErrorCodes.cpp',

    'html/HTMLTableRowsCollection.cpp',

    'html/HTMLTokenizer.cpp',

    'html/HTMLViewSourceDocument.cpp',

    'html/ImageData.cpp',

    'html/LegacyHTMLDocumentParser.cpp',

    'html/LegacyPreloadScanner.cpp',

    'html/PreloadScanner.cpp',

    'html/TimeRanges.cpp',

    'html/ValidityState.cpp',

    'inspector/InspectorFrontendClientLocal.cpp',

    'loader/FrameLoader.cpp',

    'loader/FTPDirectoryDocument.cpp',

    'loader/ImageDocument.cpp',

    'loader/MediaDocument.cpp',

    'loader/PluginDocument.cpp',

    'loader/TextResourceDecoder.cpp',

    'page/Frame.cpp',

    'page/qt/DragControllerQt.cpp',

    'page/qt/EventHandlerQt.cpp',

    'page/qt/FrameQt.cpp',

    'platform/ContextMenu.cpp',

    'platform/KURL.cpp',

    'platform/Scrollbar.cpp',

    'platform/ScrollbarThemeComposite.cpp',

    'platform/graphics/Color.cpp',

    'platform/graphics/MediaPlayer.cpp',

    'platform/graphics/SimpleFontData.cpp',

    'platform/qt/PasteboardQt.cpp',

    'platform/qt/PlatformQtAllInOne.cpp',

    'plugins/qt/PluginDataQt.cpp',

    'rendering/MediaControlElements.cpp',

    'rendering/PointerEventsHitRules.cpp',

    'rendering/RenderListItem.cpp',

    'rendering/RenderImage.cpp',

    'rendering/RenderLayer.cpp',

    'rendering/RenderLayerBacking.cpp',

    'rendering/RenderVideo.cpp',

    'rendering/RenderObjectChildList.cpp',

    'rendering/RenderingSVGAllInOne.cpp',

    'rendering/SVGRenderTreeAsText.cpp',

    'svg/SVGDocumentExtensions.cpp',

    'svg/SVGFontFaceElement.cpp',

    'svg/SVGGlyphElement.cpp',

    'svg/SVGHKernElement.cpp',

    'svg/SVGImageLoader.cpp',

    'svg/SVGStyleElement.cpp',

    'svg/SVGStyledElement.cpp',

    ]

# Find WebCore's directory

print('... Seeking the WebCore directory')

while not os.path.exists('WebCore/WebCore.pro'):

    if os.path.realpath('.') == os.path.realpath('..'):

        raise Exception("Could not find WebKit's root directory within parent directories. Are you in WebKit's tree?") 

    os.chdir('..')

os.chdir('WebCore')

# Read the .mmp file name from bld.inf

mmpFileNamePattern = re.compile(r'.*\.mmp$', re.IGNORECASE)

if not os.path.exists('bld.inf'):

    raise Exception('Could not find WebCore/bld.inf. Did you run build-webkit or qmake using Qt configured for symbian-sbsv2?')

with open('bld.inf') as bldFile:

    for line in bldFile:

        m = mmpFileNamePattern.match(line)

        if m:

            mmpFileName = m.group()

            break

newContentLines = []

sourcePathPattern = re.compile(r'SOURCEPATH\s+(.+)$')

sourcePattern = re.compile(r'SOURCE\s+(.+)$')

allInOneFilePattern = re.compile(r'SOURCE\s+AllInOne_')

pathChunkPattern = re.compile(r'\w+')

dirCombinedFilesSubGroups = [] # Each group is a tuple of its file pattern along with the opened aglomerate file

dirIsExcluded = False

if not os.path.exists(mmpFileName):

    raise Exception('Could not find the {0} file specified in WebCore/bld.inf. Did you run build-webkit or qmake using Qt configured for symbian-sbsv2?'.format(mmpFileName))

with open(mmpFileName) as mmpFile:

    for line in mmpFile:

        if allInOneFilePattern.match(line):

            raise Exception("This script has already been ran. Please re-run qmake or restore a backup copy of the WebCore's mmp file if you want to force a refresh.")

    mmpFile.seek(0)

    for line in mmpFile:

        # Look for SOURCEPATH...-like lines

        pathMatch = sourcePathPattern.match(line)

        if pathMatch:

            currentSourcePath = pathMatch.group(1)

            newContentLines.append(line)

            if currentSourcePath in exceptionDirs:

                dirIsExcluded = True

            else:

                dirIsExcluded = False

                sourceSuffix = '_'.join(pathChunkPattern.findall(currentSourcePath))

                # Close aglomerate files and reset the sub groups list

                for group in dirCombinedFilesSubGroups:

                    group[1].close()

                dirCombinedFilesSubGroups = []

                # The additional subgroup, if specified. armcc doesn't seem to like huge files

                if currentSourcePath in dirExtraSplits.keys():

                    patterns = dirExtraSplits[currentSourcePath]

                    for i in range(len(patterns)):

                        allInOneFileName = 'AllInOne_{0}{1}.cpp'.format(sourceSuffix, i)

                        print('--- Creating aglomerate file [{0}]'.format(currentSourcePath + '/' + allInOneFileName))

                        dirCombinedFilesSubGroups.append((patterns[i], open(currentSourcePath + '/' + allInOneFileName, 'w')))

                        newContentLines.append("SOURCE\t\t{0}\n".format(allInOneFileName))

                # The default aglomerate source file which include cpp files

                # Append it as the last group with no pattern to be used when no other sub-group pattern matched

                allInOneFileName = 'AllInOne_{0}.cpp'.format(sourceSuffix)

                print('--- Creating aglomerate file [{0}]'.format(currentSourcePath + '/' + allInOneFileName))

                dirCombinedFilesSubGroups.append((None, open(currentSourcePath + '/' + allInOneFileName, 'w')))

                newContentLines.append("SOURCE\t\t{0}\n".format(allInOneFileName))

            continue # for line in mmpFile:

        # Look for SOURCE...-like lines

        sourceMatch = sourcePattern.match(line)

        if not dirIsExcluded and sourceMatch:

            sourceFileName = sourceMatch.group(1)

            sourceFilePath = currentSourcePath + '/' + sourceFileName

            if not sourceFilePath in exceptionFiles:

                for group in dirCombinedFilesSubGroups:

                    if group[0] is None or group[0].match(sourceFileName):

                        group[1].write("#include \"{0}\"\n".format(sourceFileName))

                        break # for group in dirCombinedFilesSubGroups: 

                continue # for line in mmpFile:

        # Else

        newContentLines.append(line)

for group in dirCombinedFilesSubGroups:

    group[1].close()

print('=== Patching the MMP file [{0}]'.format(mmpFileName))

mmpFile = open(mmpFileName, 'w')

mmpFile.writelines(newContentLines)

mmpFile.close()