#! /usr/bin/env python

import sys, os, re

## Get output filename
OUTFILE = "analyses.json"
if len(sys.argv) >= 2:
    OUTFILE = sys.argv[1]
if not OUTFILE.endswith(".json"):
    OUTFILE += ".json"
print("Using output file name '%s'" % OUTFILE)

# ## Add the rivet Python module build dir(s) to the Python module path, then import
# pybuild = os.path.abspath(os.path.join(os.getcwd(), "..", "..", "pyext", "build"))
# pydirs = [os.path.join(pybuild, d) for d in os.listdir(pybuild)
#           if re.match(r"lib\..*-.*-%d\.%d" % (sys.version_info[0], sys.version_info[1]), d)]
# sys.path = pydirs + sys.path
import rivet

## Add info file locations
from glob import glob
dirpatt = os.path.join(os.getcwd(), "..", "..", "analyses", "plugin*")
for d in glob(dirpatt):
    #print(d)
    rivet.addAnalysisDataPath(os.path.abspath(d))

## Build analysis Inspire -> [Ana, Ana, ...] dict
# TODO: any papers without Inspire IDs?
ins_anas = {}
for aname in sorted(rivet.AnalysisLoader.analysisNames()): #< NOTE: no alias entries
    ana = rivet.AnalysisLoader.getAnalysis(aname)
    iid = ana.inspireID()
    #print(aname, ":", iid, ":", ana.summary())
    try:
        x = int(iid)
        ins_anas.setdefault(iid, []).append(aname)
    except:
        pass

# ## Remove invalid
# if "" in ins_anas:
#     del ins_anas[""]

## Export to JSON
dictstrs = []
for ins, anas in sorted(ins_anas.items()):
    #if len(anas) > 1: print(ins, anas)
    anasstr = ", ".join('"{0}"'.format(a) for a in anas)
    dictstrs.append('"{0}": [{1}]'.format(ins, anasstr))
outstr = "{" + ", ".join(dictstrs) + "}\n"

## Write out
if OUTFILE == "-":
    sys.stdout.write(outstr)
else:
    f = open(OUTFILE, "w")
    f.write(outstr)
    f.close()
