#!/usr/bin/env python3

# Ancillary wrapper around gdk-pixbuf-print-mime-types that sets up a

# modified environment in order to use the tools that we just built

# instead of the system ones

import argparse

import os

import subprocess

import sys

argparser = argparse.ArgumentParser(description='Compile resources')

argparser.add_argument('--printer', metavar='PATH', help='Path to gdk-pixbuf-print-mime-types')

argparser.add_argument('--pixdata', metavar='PATH', help='Path to gdk-pixbuf-pixdata')

argparser.add_argument('--loaders', metavar='PATH', help='Path to the loaders.cache file')

argparser.add_argument('--bindir', metavar='PATH', help='Path to the source directory')

argparser.add_argument('input', help='Template file')

argparser.add_argument('output', help='Output file')

args = argparser.parse_args()

newenv = os.environ.copy()

newenv['GDK_PIXBUF_PIXDATA'] = args.pixdata

newenv['GDK_PIXBUF_MODULE_FILE'] = args.loaders

# 'nt': NT-based Windows, see https://docs.python.org/3/library/os.html

if os.name == 'nt':

    gdk_pixbuf_dll_buildpath = os.path.dirname(args.pixdata)

    newenv['PATH'] = gdk_pixbuf_dll_buildpath + os.pathsep + newenv['PATH']

cmd = args.printer

mimetypes_out = subprocess.Popen(cmd, env=newenv, stdout=subprocess.PIPE).communicate()[0]

if not mimetypes_out:

    sys.exit(1)

infile = open(args.input, 'r')

outfile = open(args.output, 'w')

for line in infile:

    line = line.replace('@bindir@', args.bindir)

    line = line.replace('@mimetypes@', mimetypes_out.decode('ascii'))

    outfile.write(line)

infile.close()

outfile.close()

sys.exit(0)