#!/usr/bin/python3
import os
import re
import sys
if len (sys.argv) < 3:
    print ("Specify hpcups.drv and hpijs.drv pathnames")
    sys.exit (1)

hpcups_drv = sys.argv[1]
hpijs_drv = sys.argv[2]

# Match e.g.      Model "ModelName"
# and catch 'ModelName' in group 0
mn_re = re.compile ('^\s*ModelName\s+"(.*)"\s*$')

# Match e.g.      Attribute "1284DeviceID" "" "blah"
# and catch everything before 'blah' in group 0, 'blah' in group 1,
# trailing characters in group 2
devid_re = re.compile ('^(\s*Attribute\s+"1284DeviceID"\s+""\s+")(.*)("\s*)$')

# Match e.g.   }
end_re = re.compile ('^\s*}')

devid_by_mn = dict()

hpcups_lines = open (hpcups_drv, "r").readlines ()
current_mn = None
for line in hpcups_lines:
    if current_mn == None:
        match = mn_re.match (line)
        if match == None:
            continue

        current_mn = match.groups ()[0]
    else:
        match = devid_re.match (line)
        if match:
            devid_by_mn[current_mn] = match.groups ()[1]
            continue

    if end_re.match (line):
        current_mn = None

print("%d IEEE 1284 Device IDs loaded from %s" % (len (devid_by_mn),
                                                os.path.basename (hpcups_drv)),
      file=sys.stderr)

replaced = 0
hpijs_lines = open (hpijs_drv, "r").readlines ()
current_mn = None
for line in hpijs_lines:
    if current_mn == None:
        match = mn_re.match (line)
        if match:
            current_mn = match.groups ()[0]
            if current_mn.endswith (" hpijs"):
                current_mn = current_mn[:-6]
    else:
        match = devid_re.match (line)
        if match:
            devid = devid_by_mn.get (current_mn)
            if devid:
                line = (match.groups ()[0] + devid + match.groups ()[2])
                replaced += 1
            else:
                print ("Not matched: %s" % current_mn, file=sys.stderr)

    if end_re.match (line):
        current_mn = None

    print (line.rstrip ("\n"))

print("%d IEEE 1284 Device IDs loaded in %s" % (replaced,
                                                os.path.basename (hpijs_drv)),
      file=sys.stderr)