#!/usr/bin/python
# -*- coding: utf8 -*-

# © 2010 Rafaël Carré <rafael.carre@gmail>

import sys
import os
import subprocess
import tempfile


def run_gcc(args):
    os.execv(args[0], args) # run real gcc

def get_output(args):
    output = False
    for i in args:
        if output == True:
            return i
        elif i == '-o':
            output = True


def try_thumb(args, output):
    thumb_args = args + ['-mthumb']
    thumb_args[thumb_args.index('-o') + 1] = output
    thumb_gcc = subprocess.Popen(thumb_args, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    (stdout, stderr) = thumb_gcc.communicate()

    if thumb_gcc.returncode != 0: # building failed
        return False

    # building with thumb succeeded, show our output
    sys.stderr.write(stderr)
    sys.stdout.write(stdout)
    return True

##### main


args=sys.argv[1:]   # remove script path

for opt in ['-E', '-MM', '-v']:
    if opt in args:
        run_gcc(args)

output = get_output(args)
split = output.rsplit('.o', 1)

if len(split) == 1: # output doesn't end in .o
    run_gcc(args)

dirname = os.path.dirname(output)
thumb_output = tempfile.mktemp(suffix='.o', prefix=split[0], dir=dirname)

args.append('-mthumb-interwork')

if try_thumb(args, thumb_output):
    os.rename(thumb_output, output)
    sys.exit(0)
else:
    #sys.stderr.write('skipped ' + os.path.basename(output) + '\n')
    run_gcc(args)