/*
 * Copyright 2023, Vincent Douillet <vincent@vdouillet.fr>
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 * 1. Redistributions of source code must retain the above copyright notice,
 * this list of conditions and the following disclaimer.
 *
 * 2. 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.
 *
 * 3. Neither the name of the copyright holder nor the names of its contributors
 * may be used to endorse or promote products derived from this software without
 * specific prior written permission.
 *
 * 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.
 */

#include <stdarg.h>
#include <stdio.h>
#include <stdlib.h>

#include "http.h"
#include "mime.h"
#include "url.h"

bool
http_open_file(struct kreq * r, enum khttp code, const struct file * f)
{
	char   *filename;

	if (f == NULL || f->path == NULL)
		return false;

	filename = file_get_basename(f);
	if (filename == NULL)
		return false;

	/* file name needs to be url encoded for special chars */
	if ((filename = url_encode(filename)) == NULL)
		return false;

	khttp_head(r, kresps[KRESP_STATUS], "%s", khttps[code]);
	khttp_head(r, kresps[KRESP_CONTENT_DISPOSITION],
		   "attachment;filename=\"%s\"", filename);
	free(filename);
	khttp_head(r, kresps[KRESP_CONTENT_TYPE], "%s", mime_str(f->mime));
	khttp_head(r, kresps[KRESP_CONTENT_LENGTH], "%zu", f->size);
	khttp_body_compress(r, 0);	/* file is not compressed */

	return true;
}

void
http_open(struct kreq * r, enum khttp code, enum kmime mime)
{
	khttp_head(r, kresps[KRESP_STATUS], "%s", khttps[code]);
	khttp_head(r, kresps[KRESP_CONTENT_TYPE], "%s", kmimetypes[mime]);
	khttp_body(r);
}

void
http_exit(const struct kreq * r, enum khttp code, const char *content,...)
{
	va_list args;

	printf("Status: %s\n", khttps[code]);
	printf("Content-Type: text/plain\n");
	printf("\n");

	if (content == NULL)
		printf("%s", khttps[code]);
	else {
		va_start(args, content);
		vprintf(content, args);
		va_end(args);
	}

	va_start(args, content);
	kutil_verr(r, NULL, content, args);
	va_end(args);
}