#include <vector>
#include <string>
#include <cstring>
#include <algorithm>
#include <iostream>

#include "proxysql.h"

#include "tap.h"
#include "command_line.h"

#include "utils.h"

#include <ctype.h>

__thread int mysql_thread___query_digests_max_query_length = 65000;
__thread bool mysql_thread___query_digests_lowercase = false;
__thread bool mysql_thread___query_digests_replace_null = false;
__thread bool mysql_thread___query_digests_no_digits = false;
__thread int mysql_thread___query_digests_grouping_limit = 3;

const std::vector<std::string> queries {
	// floats
	"select 1.1",
	"select 1192.1102",
	"select 99.1929",
	// exponentials
	"select 1.1e9",
	"select 1.1e+9",
	"select 1.1e-9",
	// operators
	"select 1 +1",
	"select 1+ 1",
	"select 1- 1",
	"select 1 -1",
	"select 1* 1",
	"select 1 *1",
	"select 1/ 1",
	"select 1 /1",
	"select 1% 1",
	"select 1 %1",
	// operators and commas
	"select 1+ 1, 1 -1, 1 * 1 , 1/1 , 100 % 3",
	"SELECT * FROM t t1, t t2 ,t t3,t t4 LIMIT 0",
	// strings
	"select * from t where t = \"foo\"",
	"select \"1+ 1, 1 -1, 1 * 1 , 1/1 , 100 % 3\"",
	// not modified
	"select * fromt t",
	// test query_digest reduction
	"SELECT * FROM tablename WHERE id IN (1,2,3,4,5,6,7,8,9,10)",
	"SELECT * FROM tablename WHERE id IN (1,2,3,4)",
	// invalid request grouping
	"SELECT * tablename where id IN (1,2,3,4,5,6,7,8,  AND j in (1,2,3,4,5,6  and k=1",
	// random insert queries
	"INSERT INTO db.table(col1) VALUES ('val')",
	"INSERT INTO db.table (col1) VALUES ('val')",
	"INSERT INTO db.table( col1) VALUES ( 'val' )",
	"INSERT INTO db.table( col1) VALUES ( 'val' )",
	"INSERT INTO db.table  ( col1 )  VALUES ( 'val' )",
	"INSERT INTO db.table (col1, col2,col3,col4) VALUES ('val',2,3,'foo')",
	"INSERT INTO db.table ( col1, col2,col3,col4 ) VALUES ('val',2,3,'foo')"
};

const std::vector<std::string> exp_results {
	// floats
	"select ?",
	"select ?",
	"select ?",
	// exponentials
	"select ?",
	"select ?",
	"select ?",
	// operators
	"select ?+?",
	"select ?+?",
	"select ?-?",
	"select ?-?",
	"select ?*?",
	"select ?*?",
	"select ?/?",
	"select ?/?",
	"select ?%?",
	"select ?%?",
	// operators and commas
	"select ?+?,?-?,?*?,?/?,?%?",
	"SELECT * FROM t t1,t t2,t t3,t t4 LIMIT ?",
	// strings
	"select * from t where t = ?",
	"select ?",
	// not modified
	"select * fromt t",
	// test query_digest reduction
	"SELECT * FROM tablename WHERE id IN (?,?,?,...)",
	"SELECT * FROM tablename WHERE id IN (?,?,?,...)",
	// invalid request grouping
	"SELECT * tablename where id IN (?,?,?,... AND j in (?,?,?,... and k=?",
	// random insert queries
	"INSERT INTO db.table(col1) VALUES (?)",
	"INSERT INTO db.table (col1) VALUES (?)",
	"INSERT INTO db.table( col1) VALUES ( ? )",
	"INSERT INTO db.table( col1) VALUES ( ? )",
	"INSERT INTO db.table ( col1 ) VALUES ( ? )",
	"INSERT INTO db.table (col1,col2,col3,col4) VALUES (?,?,?,?)",
	"INSERT INTO db.table ( col1,col2,col3,col4 ) VALUES (?,?,?,?)"
};

const std::vector<std::string> queries_grouping {
	// test query_digest reduction
	"SELECT * FROM tablename WHERE id IN (1,2, 3,4 ,5 ,6,7,8,9,10)",
	// invalid request grouping
	"SELECT * tablename where id IN (1,2,3,4,5 , 6,7,8,  AND j in (1, 2,3, 4 ,5,6,7,8,9  and k=1",
	"SELECT (1.1, 1, 2, 13, 4.81, 12) FROM db.table",
	"SELECT (1.1, 1.12934 , 21.32 , 91, 91, 12.93 ) FROM db.table2",
	"SELECT (1.1, 1.12934 , 21.32 , 91.2 , 91, 12 ) FROM db.table7",
	"SELECT (1.1, 1.12934, 21.32, 391,2381,28.493,1283 ) FROM db.table2",
	"SELECT (1.1, 1.12934, 21.32 , 91, 91, 12.1 ) FROM db.table3"
};

const std::vector<std::string> exp_queries_grouping {
	// test query_digest reduction
	"SELECT * FROM tablename WHERE id IN (?,...)",
	// invalid request grouping
	"SELECT * tablename where id IN (?,... AND j in (?,... and k=?",
	"SELECT (?,...) FROM db.table",
	"SELECT (?,...) FROM db.table2",
	"SELECT (?,...) FROM db.table7",
	"SELECT (?,...) FROM db.table2",
	"SELECT (?,...) FROM db.table3"
};

std::string replace_str(const std::string& str, const std::string& match, const std::string& repl) {
	if(match.empty()) {
		return str;
	}

	std::string result = str;
	size_t start_pos = 0;

	while((start_pos = result.find(match, start_pos)) != std::string::npos) {
		result.replace(start_pos, match.length(), repl);
		start_pos += repl.length();
	}

	return result;
}

std::string increase_mark_num(const std::string query, uint32_t num) {
	std::string result = query;
	std::string marks = "";

	for (uint32_t i = 0; i < num - 1; i++) {
		marks += "?,";
	}
	marks += "?,...";

	result = replace_str(result, "?,...", marks);

	return result;
}

int main(int argc, char** argv) {
	if (queries.size() != exp_results.size()) {
		ok(0, "queries and exp_results sizes doesn't match");
		return exit_status();
	}

	char buf[QUERY_DIGEST_BUF];

	for (size_t i = 0; i < queries.size(); i++) {
		const auto& query = queries[i];
		const auto& exp_res = exp_results[i];

		char* c_res = mysql_query_digest_and_first_comment(const_cast<char*>(query.c_str()), query.length(), NULL, buf);
		std::string result(c_res);

		ok(result == exp_res, "Digest should be equal to exp result: '%s' == '%s'", result.c_str(), exp_res.c_str());
	}

	for (size_t i = 0; i < queries_grouping.size(); i++) {
		for (int j = 1; j <= 5; j++) {
			mysql_thread___query_digests_grouping_limit = j;
	
			const auto& query = queries_grouping[i];
			const auto& exp_res = increase_mark_num(exp_queries_grouping[i], j);

			char* c_res = mysql_query_digest_and_first_comment(const_cast<char*>(query.c_str()), query.length(), NULL, buf);
			std::string result(c_res);

			ok(result == exp_res, "Grouping digest should be equal to exp result: '%s' == '%s'", result.c_str(), exp_res.c_str());
		}
	}

	return exit_status();
}