# Copyright 2015 Google Inc. All rights reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#    http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import os
from setuptools import setup
from setuptools import Extension
from setuptools.command.build_ext import build_ext as BuildExt
from subprocess import Popen

DIR = os.path.abspath(os.path.dirname(__file__))
# NOTE: If you are editing the array below then you probably also need
# to change MANIFEST.in.
LIB_OBJECTS = [
    'core/desugarer.o',
    'core/formatter.o',
    'core/libjsonnet.o',
    'core/lexer.o',
    'core/parser.o',
    'core/pass.o',
    'core/static_analysis.o',
    'core/string_utils.o',
    'core/vm.o',
    'third_party/md5/md5.o',
    'third_party/rapidyaml/rapidyaml/ext/c4core/src/c4/char_traits.o',
    'third_party/rapidyaml/rapidyaml/ext/c4core/src/c4/base64.o',
    'third_party/rapidyaml/rapidyaml/ext/c4core/src/c4/language.o',
    'third_party/rapidyaml/rapidyaml/ext/c4core/src/c4/memory_util.o',
    'third_party/rapidyaml/rapidyaml/ext/c4core/src/c4/format.o',
    'third_party/rapidyaml/rapidyaml/ext/c4core/src/c4/time.o',
    'third_party/rapidyaml/rapidyaml/ext/c4core/src/c4/memory_resource.o',
    'third_party/rapidyaml/rapidyaml/ext/c4core/src/c4/error.o',
    'third_party/rapidyaml/rapidyaml/src/c4/yml/parse.o',
    'third_party/rapidyaml/rapidyaml/src/c4/yml/preprocess.o',
    'third_party/rapidyaml/rapidyaml/src/c4/yml/common.o',
    'third_party/rapidyaml/rapidyaml/src/c4/yml/tree.o',
]

MODULE_SOURCES = ['python/_jsonnet.c']

def get_version():
    """
    Parses the version out of libjsonnet.h
    """
    with open(os.path.join(DIR, 'include/libjsonnet.h')) as f:
        for line in f:
            if '#define' in line and 'LIB_JSONNET_VERSION' in line:
                v_code = line.partition('LIB_JSONNET_VERSION')[2].strip('\n "')
                if v_code[0] == "v":
                    v_code = v_code[1:]
                return v_code

class BuildJsonnetExt(BuildExt):
    def run(self):
        p = Popen(['make'] + LIB_OBJECTS, cwd=DIR)
        p.wait()
        if p.returncode != 0:
            raise Exception('Could not build %s' % (', '.join(LIB_OBJECTS)))
        BuildExt.run(self)

jsonnet_ext = Extension(
    '_jsonnet',
    sources=MODULE_SOURCES,
    extra_objects=LIB_OBJECTS,
    include_dirs = ['include'],
    language='c++'
)

setup(name='jsonnet',
      url='https://jsonnet.org',
      project_urls={
        'Source': 'https://github.com/google/jsonnet',
      },
      description='Python bindings for Jsonnet - The data templating language ',
      license="Apache License 2.0",
      author='David Cunningham',
      author_email='dcunnin@google.com',
      version=get_version(),
      cmdclass={
          'build_ext': BuildJsonnetExt,
      },
      ext_modules=[jsonnet_ext],
      test_suite="python._jsonnet_test",
)