cimg

#!/usr/local/bin/python

# -*- coding: utf-8 -*-

################################################################################

# nif_stub.py

# Description:  

#

# Author:       shozo fukuda

# Date:         Mon Dec  6 07:36:25 2021

# Last revised: $Date$

# Application:  Python 3.9

################################################################################


#<IMPORT>

import os,sys
import argparse
import re

#<CLASS>########################################################################

# Description:  

# Dependencies: 

################################################################################

class NifStub:
    def __init__(self, module):
        self.module = module
        self.app    = ':'+module.split('.')[0].lower()
        self.nif    = module.lower().replace('.', '_')      # restrict nif name

        self.func   = []

    def parse(self, file):
        for line in file:
            match = re.search(r'^{"(.*)",\s*(\d+),', line)
            if match:
                name  = match.group(1)
                ality = int(match.group(2))
                self.func.append((name, ality))

    def mk_niftbl(self, output):
        print('defmodule {module} do\n'
               '  @moduledoc false\n'
               '  @compile {{:autoload, false}}\n'
               '\n'
               '  #loading NIF library\n'
               '  @on_load :load_nif\n'
               '  def load_nif do\n'
               '    nif_file = Application.app_dir({app}, "priv/{nif}")\n'
               '    :erlang.load_nif(nif_file, 0)\n'
               '  end\n'
               '\n'
               '  # stub implementations for NIFs (fallback)\n'
               .format(
                   module=self.module,
                   app=self.app,
                   nif=self.nif),
               end='',
               file=output)

        for name, ality in self.func:
            print('  def {name}({prms}),\n    do: raise("NIF {name}/{ality} not implemented")'
                   .format(
                       name=name,
                       ality=ality,
                       prms=', '.join(["_%d"%(x+1) for x in range(ality)])),
                   file=output)

        print('end', file=output)

#<MAIN>#########################################################################

# Function:     

# Description:  

# Dependencies: 

################################################################################

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description="Generate NIFs table")
    parser.add_argument('module',
        help="module name")
    parser.add_argument('srcs', nargs='+', type=argparse.FileType('r'),
        help="src files")
    parser.add_argument('-o', '--output', type=argparse.FileType('w'), default=sys.stdout,
        help="output file")
    args = parser.parse_args()

    nif_stub = NifStub(args.module)
    for src in args.srcs:
        nif_stub.parse(src)
    nif_stub.mk_niftbl(args.output)

# nif_stub.py