/* Copyright (c) Pigeonhole authors, see top-level COPYING file */

#include "lib.h"

#include "sieve.h"
#include "sieve-code.h"
#include "sieve-commands.h"
#include "sieve-binary.h"
#include "sieve-generator.h"
#include "sieve-interpreter.h"
#include "sieve-dump.h"

#include "testsuite-common.h"
#include "testsuite-substitutions.h"

/*
 * Forward declarations
 */

void testsuite_opr_substitution_emit(struct sieve_binary_block *sblock,
				     const struct testsuite_substitution *tsub,
				     const char *param);

/*
 * Testsuite substitutions
 */

/* FIXME: make this extendible */

enum {
	TESTSUITE_SUBSTITUTION_FILE,
};

static const struct testsuite_substitution_def testsuite_file_substitution;

static const struct testsuite_substitution_def *substitutions[] = {
	&testsuite_file_substitution,
};

static const unsigned int substitutions_count = N_ELEMENTS(substitutions);

static inline const struct testsuite_substitution_def *
testsuite_substitution_get(unsigned int code)
{
	if (code >= substitutions_count)
		return NULL;

	return substitutions[code];
}

static const struct testsuite_substitution *
testsuite_substitution_create(struct sieve_ast *ast, const char *identifier)
{
	unsigned int i;

	for (i = 0; i < substitutions_count; i++) {
		if (strcasecmp(substitutions[i]->obj_def.identifier,
			       identifier) == 0) {
			const struct testsuite_substitution_def *tsub_def =
				substitutions[i];
			struct testsuite_substitution *tsub;

			tsub = p_new(sieve_ast_pool(ast),
				     struct testsuite_substitution, 1);
			tsub->object.def = &tsub_def->obj_def;
			tsub->object.ext = testsuite_ext;
			tsub->def = tsub_def;

			return tsub;
		}
	}

	return NULL;
}

/*
 * Substitution argument
 */

static bool
arg_testsuite_substitution_generate(const struct sieve_codegen_env *cgenv,
				    struct sieve_ast_argument *arg,
				    struct sieve_command *context);

struct _testsuite_substitution_context {
	const struct testsuite_substitution *tsub;
	const char *param;
};

const struct sieve_argument_def testsuite_substitution_argument = {
	.identifier = "@testsuite-substitution",
	.generate = arg_testsuite_substitution_generate,
};

struct sieve_ast_argument *
testsuite_substitution_argument_create(
	struct sieve_validator *valdtr ATTR_UNUSED, struct sieve_ast *ast,
	unsigned int source_line, const char *substitution, const char *param)
{
	const struct testsuite_substitution *tsub;
	struct _testsuite_substitution_context *tsctx;
	struct sieve_ast_argument *arg;
	pool_t pool;

	tsub = testsuite_substitution_create(ast, substitution);
	if (tsub == NULL)
		return NULL;

	arg = sieve_ast_argument_create(ast, source_line);
	arg->type = SAAT_STRING;

	pool = sieve_ast_pool(ast);
	tsctx = p_new(pool, struct _testsuite_substitution_context, 1);
	tsctx->tsub = tsub;
	tsctx->param = p_strdup(pool, param);

	arg->argument = sieve_argument_create
		(ast, &testsuite_substitution_argument, testsuite_ext, 0);
	arg->argument->data = tsctx;
	return arg;
}

static bool
arg_testsuite_substitution_generate(const struct sieve_codegen_env *cgenv,
				    struct sieve_ast_argument *arg,
				    struct sieve_command *context ATTR_UNUSED)
{
	struct _testsuite_substitution_context *tsctx =
		(struct _testsuite_substitution_context *)arg->argument->data;

	testsuite_opr_substitution_emit(cgenv->sblock, tsctx->tsub, tsctx->param);
	return TRUE;
}

/*
 * Substitution operand
 */

static bool
opr_substitution_dump(const struct sieve_dumptime_env *denv,
		      const struct sieve_operand *oprnd, sieve_size_t *address);
static int
opr_substitution_read_value(const struct sieve_runtime_env *renv,
			    const struct sieve_operand *oprnd,
			    sieve_size_t *address, string_t **str);

const struct sieve_opr_string_interface testsuite_substitution_interface = {
	opr_substitution_dump,
	opr_substitution_read_value
};

const struct sieve_operand_def testsuite_substitution_operand = {
	.name = "test-substitution",
	.ext_def = &testsuite_extension,
	.code = TESTSUITE_OPERAND_SUBSTITUTION,
	.class = &string_class,
	.interface = &testsuite_substitution_interface
};

void testsuite_opr_substitution_emit(struct sieve_binary_block *sblock,
				     const struct testsuite_substitution *tsub,
				     const char *param)
{
	/* Default variable storage */
	(void)sieve_operand_emit(sblock, testsuite_ext,
				 &testsuite_substitution_operand);
	(void)sieve_binary_emit_unsigned(sblock, tsub->object.def->code);
	(void)sieve_binary_emit_cstring(sblock, param);
}

static bool
opr_substitution_dump(const struct sieve_dumptime_env *denv,
		      const struct sieve_operand *oprnd, sieve_size_t *address)
{
	unsigned int code = 0;
	const struct testsuite_substitution_def *tsub;
	string_t *param;

	if (!sieve_binary_read_unsigned(denv->sblock, address, &code))
		return FALSE;

	tsub = testsuite_substitution_get(code);
	if (tsub == NULL)
		return FALSE;

	if (!sieve_binary_read_string(denv->sblock, address, &param))
		return FALSE;

	if (oprnd->field_name != NULL) {
		sieve_code_dumpf(denv, "%s: TEST_SUBS %%{%s:%s}",
				 oprnd->field_name, tsub->obj_def.identifier,
				 str_c(param));
	} else {
		sieve_code_dumpf(denv, "TEST_SUBS %%{%s:%s}",
				 tsub->obj_def.identifier, str_c(param));
	}
	return TRUE;
}

static int
opr_substitution_read_value(const struct sieve_runtime_env *renv,
			    const struct sieve_operand *oprnd ATTR_UNUSED,
			    sieve_size_t *address, string_t **str_r)
{
	const struct testsuite_substitution_def *tsub;
	unsigned int code = 0;
	string_t *param;

	if (!sieve_binary_read_unsigned(renv->sblock, address, &code))
		return SIEVE_EXEC_BIN_CORRUPT;

	tsub = testsuite_substitution_get(code);
	if (tsub == NULL)
		return SIEVE_EXEC_FAILURE;

	/* Parameter str can be NULL if we are requested to only skip and not
	   actually read the argument.
	 */
	if (str_r == NULL) {
		if (!sieve_binary_read_string(renv->sblock, address, NULL))
			return SIEVE_EXEC_BIN_CORRUPT;
		return SIEVE_EXEC_OK;
	}

	if (!sieve_binary_read_string(renv->sblock, address, &param))
		return SIEVE_EXEC_BIN_CORRUPT;
	if (!tsub->get_value(str_c(param), str_r))
		return SIEVE_EXEC_FAILURE;

	return SIEVE_EXEC_OK;
}

/*
 * Testsuite substitution definitions
 */

static bool
testsuite_file_substitution_get_value(const char *param, string_t **result);

static const struct testsuite_substitution_def
testsuite_file_substitution = {
	SIEVE_OBJECT("file", &testsuite_substitution_operand,
		     TESTSUITE_SUBSTITUTION_FILE),
	.get_value = testsuite_file_substitution_get_value,
};

static bool
testsuite_file_substitution_get_value(const char *param, string_t **result)
{
	*result = t_str_new(256);

	str_printfa(*result, "[FILE: %s]", param);
	return TRUE;
}
