#!/usr/bin/env python
#
# Copyright (C) 2011-2012 ABINIT Group (Yann Pouillon)
#
# This file is part of the ABINIT software package. For license information,
# please see the COPYING file in the top-level directory of the ABINIT source
# distribution.
#

from ConfigParser import ConfigParser
from time import gmtime,strftime

import os
import re
import sys

class MyConfigParser(ConfigParser):

  def optionxform(self,option):
    return str(option)



# ---------------------------------------------------------------------------- #

#
# Main program
#

# Initial setup
my_name     = "make-data-tests"
my_config   = "config/specs/buildsys.conf"
my_macro    = "config/m4/auto-tests.m4"
my_makefile = "Makefile.am"

# Check if we are in the top of the ABINIT source tree
if ( not os.path.exists("configure.ac") or
     not os.path.exists("src/98_main/abinit.F90") ):
  print "%s: You must be in the top of an ABINIT source tree." % my_name
  print "%s: Aborting now." % my_name
  sys.exit(1)

# Check if we have a config file
if ( os.path.exists(my_config) ):
  cnf = MyConfigParser()
  cnf.read(my_config)
else:
  print "%s: Could not find config file (%s)." % (my_name,my_config)
  print "%s: Aborting now." % my_name
  sys.exit(2)

# What time is it?
now = strftime("%Y/%m/%d %H:%M:%S +0000",gmtime())

# Init macro
tests_macro = """\
# Generated by %s on %s

#
# ABINIT Test Suite support for the "configure" script
#

#
# IMPORTANT NOTE
#
# This file has been automatically generated by the %s
# script. If you try to edit it, your changes will systematically be
# overwritten.
#



# ABI_TESTS_INIT()
# --------------
#
# Defines whether the Abinit Documentation may be built along with the package.
#
AC_DEFUN([ABI_TESTS_INIT],[
  dnl Init
  abi_tests_mode="@TESTS_MODE@"
@TESTS_INIT@

  case "${abi_tests_mode}" in
    data)
      AC_MSG_NOTICE([the Abinit Documentation will never be built])
      ;;
    subsystem)
      AC_MSG_NOTICE([the Abinit Documentation may be built upon request])
      ;;
  esac

  AM_CONDITIONAL([DO_BUILD_TESTS],[test "${enable_tests_build}" = "yes"])
]) # ABI_TESTS_INIT
""" % (my_name,now,my_name)

# Extract information from config file
tests_mode = cnf.get("tests","mode")

# Generate Autoconf macro
if ( tests_mode == "data" ):
  tests_init = "  AC_CONFIG_FILES([tests/Makefile])"
elif ( tests_mode == "subsystem" ):
  tests_init = "  AC_CONFIG_SUBDIRS([tests])\n"

tests_macro = re.sub("@TESTS_MODE@",tests_mode,tests_macro)
tests_macro = re.sub("@TESTS_INIT@",tests_init,tests_macro)
file(my_macro,"w").write(tests_macro)

# Generate makefile snippets
if ( tests_mode == "data" ):
  tests_files = list()
  os.system("cd tests && ./wipeout.sh >/dev/null 2>&1")
  os.system("cd tests && ./autogen.sh >/dev/null 2>&1")
  for root,dirs,files in os.walk("tests"):
    if ( "autom4te.cache" in dirs ):
      dirs.remove("autom4te.cache")
    tests_files += [os.path.join(root,item) for item in files]
  tests_files.sort()
  file(my_makefile,"a").write("\nEXTRA_DIST += \\\n\t%s\n" % \
    " \\\n\t".join(tests_files))
elif ( tests_mode == "subsystem" ):
  file(my_makefile,"a").write("\nSUBDIRS += tests\n")
  os.system("cd tests && ./autogen.sh >/dev/null 2>&1")
