#!/usr/bin/python
# -----------------------------------------------------------------------------
#
# This file is the copyrighted property of Tableau Software and is protected
# by registered patents and other applicable U.S. and international laws and
# regulations.
#
# Unlicensed use of the contents of this file is prohibited. Please refer to
# the NOTICES.txt file for further details.
#
# -----------------------------------------------------------------------------


import os
import sys


def processFile(input_file, output_file):
    f = open(input_file, "r")
    targetData = []
    customHandler = ""
    errorTrigger = "error (yyla.location, yysyntax_error_ (yystack_[0].state,"
    replacement = ""

    for line in f:
        if customHandler == "":
            if line in ["// A Bison parser, made by GNU Bison 3.0.4.\n", "// A Bison parser, made by GNU Bison 3.3.2.\n", "// A Bison parser, made by GNU Bison 3.4.1.\n"]:
                customHandler = """
#define customErrorReporting(location_ARG,state_ARG,yyla_ARG) \\
{\\
   hyper::SQLLexer::Token token;\\
   auto yystate=state_ARG;\\
   auto yylaTmp=yyla_ARG;auto yyla=yylaTmp;\\
   const char* got=nullptr;\\
   std::vector<const char*> possibilities;\\
   if (!yyla.empty()) {\\
      auto& astToken=hyper::astql::Token::ref(*yyla.value);\\
      token={astToken.begin(),astToken.end(),static_cast<hyper::SQLLexer::Decoding>(astToken.decoding)};\\
      int yytoken=yyla.type_get();\\
      int ident=SQLParser::yytranslate_(token::IDENT);\\
      int abort_p=SQLParser::yytranslate_(token::ABORT_P);\\
      got = yytname_[yytoken]; if ((yytoken<abort_p)&&(got[0]=='"')) ++got;\\
      bool expectedIdent=false;\\
      int yyn = yypact_[yystate];\\
      if (!yy_pact_value_is_default_ (yyn)) {\\
         int yyxbegin = yyn < 0 ? -yyn : 0;\\
         int yychecklim = yylast_ - yyn + 1;\\
         int yyxend = yychecklim < yyntokens_ ? yychecklim : yyntokens_;\\
         for (int yyx = yyxbegin; yyx < yyxend; ++yyx) {\\
            if (yycheck_[yyx + yyn] == yyx && yyx != yyterror_ && !yy_table_value_is_error_ (yytable_[yyx + yyn]))\\
               if (yyx==ident) { expectedIdent=true; break; }\\
         }\\
      }\\
      if (!yy_pact_value_is_default_ (yyn)) {\\
         int yyxbegin = yyn < 0 ? -yyn : 0;\\
         int yychecklim = yylast_ - yyn + 1;\\
         int yyxend = yychecklim < yyntokens_ ? yychecklim : yyntokens_;\\
         for (int yyx = yyxbegin; yyx < yyxend; ++yyx) {\\
            if (yycheck_[yyx + yyn] == yyx && yyx != yyterror_ && !yy_table_value_is_error_ (yytable_[yyx + yyn])) {\\
               if ((expectedIdent)&&(yyx>=abort_p)) continue;\\
               const char* p=yytname_[yyx]; if ((yyx<abort_p)&&(p[0]=='\"')) ++p; \\
               possibilities.push_back(p); \\
            }\\
         }\\
      }\\
   } else {\\
      /* Construct a pseudo token placed at the right offset*/\\
      const char* loc=location_ARG.begin.column+scanner.getQueryStart()-1;\\
      token={loc,loc,hyper::SQLLexer::Decoding::Plain};\\
   }\\
   scanner.reportSyntaxError(token,got,possibilities);\\
}
"""
                replacement = "customErrorReporting(yyla.location,yystack_[0].state,yyla); /* override error reporting */"
            else:
                print("Error: unrecognized Bison version")
                sys.exit(1)
            targetData.append(line)
            targetData.append(customHandler)
        else:
            pos = line.find(errorTrigger)
            if pos > 0:
                line = line[0:pos] + replacement + line[pos:]
            targetData.append(line)
    f.close()

    f = open(output_file, "w")
    f.write("".join(targetData))


# Process the arguments
if __name__ == "__main__":
    import sys
    processFile(sys.argv[1], sys.argv[2])