#!/usr/bin/env python3

##
 # Copyright (c) 2013-2015, Roland Bock
 # Copyright (c) 2018, Egor Pugin
 # All rights reserved.
 #
 # Redistribution and use in source and binary forms, with or without modification,
 # are permitted provided that the following conditions are met:
 #
 #  * Redistributions of source code must retain the above copyright notice,
 #    this list of conditions and the following disclaimer.
 #  * Redistributions in binary form must reproduce the above copyright notice,
 #    this list of conditions and the following disclaimer in the documentation
 #    and/or other materials provided with the distribution.
 #
 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
 # ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 # WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 # IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
 # INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
 # BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
 # LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
 # OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
 # OF THE POSSIBILITY OF SUCH DAMAGE.
 ##

import argparse
import os
import re
import sqlite3
import sys

INCLUDE = 'sqlpp11'
NAMESPACE = 'sqlpp'

# map sqlite3 types
types = {
    'integer': 'integer',
    'text': 'text',
    'blob': 'blob',
    'real': 'floating_point',
}

def main():
    parser = argparse.ArgumentParser(description='sqlpp11 cpp schema generator')
    parser.add_argument('ddl', help='path to ddl')
    parser.add_argument('target', help='path to target')
    parser.add_argument('namespace', help='namespace')
    parser.add_argument('-identity-naming',
                        help='Use table and column names from the ddl '
                        '(defaults to UpperCamelCase for tables and '
                        'lowerCamelCase for columns)',
                        action='store_true')
    args = parser.parse_args()

    pathToHeader = args.target + '.h'

    # execute schema scripts
    conn = sqlite3.connect(':memory:')
    conn.executescript(open(args.ddl).read())

    # set vars
    toClassName = identity_naming_func if args.identity_naming else class_name_naming_func
    toMemberName = identity_naming_func if args.identity_naming else member_name_naming_func
    DataTypeError = False

    header = open(pathToHeader, 'w')
    namespace = args.namespace
    nsList = namespace.split('::')

    # start printing
    print('#pragma once', file=header)
    print('', file=header)
    print('// generated by ' + ' '.join(sys.argv), file=header)
    print('', file=header)
    print('#include ', file=header)
    print('#include ', file=header)
    print('#include ', file=header)
    print('', file=header)
    for ns in nsList:
        print('namespace ' + ns, file=header)
        print('{', file=header)

    cursor = conn.cursor()
    cursor.execute("SELECT name FROM sqlite_master WHERE type='table';")
    tables = cursor.fetchall()
    for table_name in tables:
        table_name = table_name[0]

        if table_name.startswith('sqlite_'):
            continue

        sqlTableName = table_name
        tableClass = toClassName(sqlTableName)
        tableMember = toMemberName(sqlTableName)
        tableNamespace = tableClass + '_'
        tableTemplateParameters = tableClass
        print('  namespace ' + tableNamespace, file=header)
        print('  {', file=header)

        columns = cursor.execute("PRAGMA table_info('%s')" % table_name).fetchall()
        for column in columns:
            sqlColumnName = column[1]
            columnClass = toClassName(sqlColumnName)
            tableTemplateParameters += ',\n               ' + tableNamespace + '::' + columnClass
            columnMember = toMemberName(sqlColumnName)
            sqlColumnType = column[2].lower()
            sqlCanBeNull = column[3] == 0 or column[3] == '0'
            print('    struct ' + columnClass, file=header)
            print('    {', file=header)
            print('      struct _alias_t', file=header)
            print('      {', file=header)
            print('        static constexpr const char _literal[] =  "' + escape_if_reserved(sqlColumnName) + '";', file=header)
            print('        using _name_t = sqlpp::make_char_sequence;', file=header)
            print('        template', file=header)
            print('        struct _member_t', file=header)
            print('          {', file=header)
            print('            T ' + columnMember + ';', file=header)
            print('            T& operator()() { return ' + columnMember + '; }', file=header)
            print('            const T& operator()() const { return ' + columnMember + '; }', file=header)
            print('          };', file=header)
            print('      };', file=header)
            try:
                traitslist = [NAMESPACE + '::' + types[sqlColumnType]]
            except KeyError as e:
                print ('Error: datatype "'  + sqlColumnType + '"" is not supported.')
                DataTypeError = True
            requireInsert = True
            hasAutoValue = sqlColumnName == 'id'
            if hasAutoValue:
                traitslist.append(NAMESPACE + '::tag::must_not_insert')
                traitslist.append(NAMESPACE + '::tag::must_not_update')
                requireInsert = False
            if sqlCanBeNull:
                traitslist.append(NAMESPACE + '::tag::can_be_null')
                requireInsert = False
            if column[4]:
                requireInsert = False
            if requireInsert:
                traitslist.append(NAMESPACE + '::tag::require_insert')
            print('      using _traits = ' + NAMESPACE + '::make_traits;', file=header)
            print('    };', file=header)

        print('  } // namespace ' + tableNamespace, file=header)
        print('', file=header)

        print('  struct ' + tableClass + ': ' + NAMESPACE + '::table_t', file=header)
        print('  {', file=header)
        print('    struct _alias_t', file=header)
        print('    {', file=header)
        print('      static constexpr const char _literal[] =  "' + sqlTableName + '";', file=header)
        print('      using _name_t = sqlpp::make_char_sequence;', file=header)
        print('      template', file=header)
        print('      struct _member_t', file=header)
        print('      {', file=header)
        print('        T ' + tableMember + ';', file=header)
        print('        T& operator()() { return ' + tableMember + '; }', file=header)
        print('        const T& operator()() const { return ' + tableMember + '; }', file=header)
        print('      };', file=header)
        print('    };', file=header)
        print('  };', file=header)


    for ns in reversed(nsList):
        print('} // namespace ' + ns, file=header)

    if (DataTypeError):
        print("Error: unsupported datatypes." )
        print("Possible solutions:")
        print("A) Implement this datatype (examples: sqlpp11/data_types)" )
        print("B) Extend/upgrade ddl2cpp (edit types map)" )
        print("C) Raise an issue on github" )
        sys.exit(10) #return non-zero error code, we might need it for automation

def get_include_guard_name(namespace, inputfile):
  val = re.sub("[^A-Za-z0-9]+", "_", namespace + '_' + os.path.basename(inputfile))
  return val.upper()

def identity_naming_func(s):
    return s

def repl_camel_case_func(m):
    if m.group(1) == '_':
        return m.group(2).upper()
    else:
        return m.group(1) + m.group(2).upper()

def class_name_naming_func(s):
    return re.sub("(^|\s|[_0-9])(\S)", repl_camel_case_func, s)

def member_name_naming_func(s):
    return re.sub("(\s|_|[0-9])(\S)", repl_camel_case_func, s)

def escape_if_reserved(name):
    reserved_names = [
        'GROUP',
        'ORDER',
    ]
    if name.upper() in reserved_names:
        return '!{}'.format(name)
    return name

if __name__ == '__main__':
    main()
    sys.exit(0)