path: root/src/mime.c

// jurl_mime implements curl_mime_*
#include "jurl.h"

// static mime
static JanetMethod jurl_mime_methods[] = {
	{"addpart", jurl_mime_addpart},
	{"attach",  jurl_mime_subparts},
	{NULL, NULL},
};

static int jurl_mime_gc(void *p, size_t s) {
	(void) s;
	jurl_mime *mime = (jurl_mime*)p;
	if (mime->clean) curl_mime_free(mime->mime);
	return 0;
}

static int jurl_mime_get(void *p, Janet key, Janet *out) {
	(void) p;
	if (!janet_checktype(key, JANET_KEYWORD)) {
		return 0;
	}
	return janet_getmethod(janet_unwrap_keyword(key), jurl_mime_methods, out);
}

static const JanetAbstractType jurl_mimetype = {
	"jurl-mime",   // name
	jurl_mime_gc,  // gc
	NULL,          // gcmark
	jurl_mime_get, //get
	JANET_ATEND_GET
};

// static mimepart
static JanetMethod jurl_mimepart_methods[] = {
	{"data",     jurl_mime_data},
	{"data-cb",  jurl_mime_data_cb},
	{"encoder",  jurl_mime_encoder},
	{"filedata", jurl_mime_filedata},
	{"filename", jurl_mime_filename},
	{"headers",  jurl_mime_headers},
	{"name",     jurl_mime_name},
	{"subparts", jurl_mime_subparts},
	{"type",     jurl_mime_type},
	{NULL, NULL},
};

static int jurl_mimepart_gc(void *p, size_t s) {
	(void) s;
	jurl_mimepart *part = (jurl_mimepart*)p;
	jurl_do_cleanup(&part->cleanup);
	return 0;
}

static int jurl_mimepart_get(void *p, Janet key, Janet *out) {
	(void) p;
	if (!janet_checktype(key, JANET_KEYWORD)) {
		return 0;
	}
	return janet_getmethod(janet_unwrap_keyword(key), jurl_mimepart_methods, out);
}

static const JanetAbstractType jurl_mimeparttype = {
	"jurl-mimepart",   // name
	jurl_mimepart_gc,  // gc
	NULL,              // gcmark
	jurl_mimepart_get, //get
	JANET_ATEND_GET
};

// mime
jurl_mime *janet_getjurlmime(Janet *argv, int32_t n) {
	return (jurl_mime*)janet_getabstract(argv, n, &jurl_mimetype);
}

JANET_CFUN(jurl_mime_addpart) {
	janet_fixarity(argc, 1);
	jurl_mime     *mime = janet_getjurlmime(argv, 0);
	jurl_mimepart *part = (jurl_mimepart*)janet_abstract(&jurl_mimeparttype, sizeof(jurl_mimepart));
	part->cleanup  = NULL;
	part->mimepart = curl_mime_addpart(mime->mime);
	part->parent   = mime;
	return janet_wrap_abstract(part);
}

// XXX: this could break in GC, I don't know exactly how it works
// if it does break for you, please tell me!
JANET_CFUN(jurl_mime_new) {
	janet_fixarity(argc, 1);
	jurl_mime   *mime = (jurl_mime*)janet_abstract(&jurl_mimetype, sizeof(jurl_mime));
	jurl_handle *jurl = (jurl_handle*)janet_getjurl(argv, 0);
	mime->clean = 1; // clean by default
	mime->mime  = curl_mime_init(jurl->handle);
	return janet_wrap_abstract(mime);
}

// mimepart
jurl_mimepart *janet_getjurlmimepart(Janet *argv, int32_t n) {
	return (jurl_mimepart*)janet_getabstract(argv, n, &jurl_mimeparttype);
}

JANET_CFUN(jurl_mime_name) {
	janet_fixarity(argc, 2);
	jurl_mimepart *part = janet_getjurlmimepart(argv, 0);
	CURLcode ret;
	if (janet_checktype(argv[1], JANET_NIL)) {
		ret = curl_mime_name(part->mimepart, NULL);
	} else {
		const char *s = (const char*)janet_getcstring(argv, 1);
		ret = curl_mime_name(part->mimepart, s);
	}
	return jurl_geterror(ret);
}

JANET_CFUN(jurl_mime_data) {
	janet_fixarity(argc, 2);
	jurl_mimepart *part = janet_getjurlmimepart(argv, 0);
	CURLcode ret;
	if (janet_checktype(argv[1], JANET_NIL)) {
		ret = curl_mime_data(part->mimepart, NULL, 0);
	} else {
		JanetByteView bytes = janet_getbytes(argv, 1);
		ret = curl_mime_data(part->mimepart, (const char*)bytes.bytes, bytes.len);
	}
	return jurl_geterror(ret);
}

static size_t readfunc(char *buffer, size_t size, size_t nitems, void *arg) {
	JanetFunction *fun = (JanetFunction*)arg;
	size_t realsize = size * nitems;

	Janet argv[2] = { janet_ckeywordv("read"),
					  janet_wrap_integer(realsize), };
	Janet res = janet_call(fun, 2, argv);
	if (janet_checktype(res, JANET_NIL)) return 0;
	JanetByteView bytes;
	if (!janet_bytes_view(res, &bytes.bytes, &bytes.len)) {
		janet_panic("could not open bytesview in readfunc");
	}

	size_t actualsize = realsize > bytes.len ? bytes.len : realsize;
	memcpy(buffer, bytes.bytes, actualsize);

	return realsize;
}

static int seekfunc(void *arg, curl_off_t offset, int origin) {
	JanetFunction *fun = (JanetFunction*)arg;
	char *pos;
	switch (origin) {
	case SEEK_SET:
		pos = "set";
		break;
	case SEEK_CUR:
		pos = "cur";
		break;
	case SEEK_END:
		pos = "end";
		break;
	default:
		janet_panicf("unrecognized origin in seekfunc: %d", origin);
	}

	Janet argv[3] = { janet_ckeywordv("seek"), 
					  janet_wrap_integer(offset),
					  janet_ckeywordv(pos), };
	Janet res = janet_call(fun, 3, argv);

	if        (janet_keyeq(res, "ok")) {
		return CURL_SEEKFUNC_OK;
	} else if (janet_keyeq(res, "fail")) {
		return CURL_SEEKFUNC_FAIL;
	} else if (janet_keyeq(res, "cantseek")) {
		return CURL_SEEKFUNC_CANTSEEK;
	} else {
		janet_panicf("unrecognized return in seekfunc: %v", res);
	}

	return CURL_SEEKFUNC_CANTSEEK; // unreachable
}

// (callback :read size)
// (callback :seek offset :set | :cur | :end)
JANET_CFUN(jurl_mime_data_cb) {
	janet_fixarity(argc, 3);
	jurl_mimepart *part = janet_getjurlmimepart(argv, 0);
	curl_off_t     size = janet_getinteger64(argv, 1);
	JanetFunction  *fun = janet_getfunction(argv, 2);

	CURLcode res = curl_mime_data_cb(part->mimepart, size, readfunc, seekfunc, NULL, fun);
	return jurl_geterror(res);
}

JANET_CFUN(jurl_mime_filedata) {
	janet_fixarity(argc, 2);
	jurl_mimepart *part = janet_getjurlmimepart(argv, 0);
	CURLcode ret;
	if (janet_checktype(argv[1], JANET_NIL)) {
		ret = curl_mime_filedata(part->mimepart, NULL);
	} else {
		const char *s = (const char*)janet_getcstring(argv, 1);
		ret = curl_mime_filedata(part->mimepart, s);
	}
	return jurl_geterror(ret);
}

JANET_CFUN(jurl_mime_filename) {
	janet_fixarity(argc, 2);
	jurl_mimepart *part = janet_getjurlmimepart(argv, 0);
	CURLcode ret;
	if (janet_checktype(argv[1], JANET_NIL)) {
		ret = curl_mime_filename(part->mimepart, NULL);
	} else {
		const char *s = (const char*)janet_getcstring(argv, 1);
		ret = curl_mime_filename(part->mimepart, s);
	}
	return jurl_geterror(ret);
}

JANET_CFUN(jurl_mime_type) {
	janet_fixarity(argc, 2);
	jurl_mimepart *part = janet_getjurlmimepart(argv, 0);
	CURLcode ret;
	if (janet_checktype(argv[1], JANET_NIL)) {
		ret = curl_mime_type(part->mimepart, NULL);
	} else {
		const char *s = (const char*)janet_getcstring(argv, 1);
		ret = curl_mime_type(part->mimepart, s);
	}
	return jurl_geterror(ret);
}

JANET_CFUN(jurl_mime_headers) {
	janet_fixarity(argc, 2);
	jurl_mimepart *part = janet_getjurlmimepart(argv, 0);
	struct jurl_cleanup *clean = register_cleanup(&part->cleanup, JURL_CLEANUP_TYPE_SLIST);
	if (!janet_getslist(&clean->slist, argv, 1)) {
		janet_panic("failed to append to slist in jurl_mime_headers");
	}
	return jurl_geterror(
			curl_mime_headers(part->mimepart, clean->slist, 0)
			);
}

JANET_CFUN(jurl_mime_encoder) {
	janet_fixarity(argc, 2);
	jurl_mimepart *part = janet_getjurlmimepart(argv, 0);
	CURLcode ret;
	if (janet_checktype(argv[1], JANET_NIL)) {
		ret = curl_mime_encoder(part->mimepart, NULL);
	} else {
		if        (janet_keyeq(argv[1], "binary")) {
			ret = curl_mime_encoder(part->mimepart, "binary");
		} else if (janet_keyeq(argv[1], "8bit")) {
			ret = curl_mime_encoder(part->mimepart, "8bit");
		} else if (janet_keyeq(argv[1], "7bit")) {
			ret = curl_mime_encoder(part->mimepart, "7bit");
		} else if (janet_keyeq(argv[1], "base64")) {
			ret = curl_mime_encoder(part->mimepart, "base64");
		} else if (janet_keyeq(argv[1], "quoted-printable")) {
			ret = curl_mime_encoder(part->mimepart, "quoted-printable");
		} else {
			janet_panic("jurl_mime_encoder: invalid encoding type");
		}
	}
	return jurl_geterror(ret);
}

JANET_CFUN(jurl_mime_subparts) {
	janet_fixarity(argc, 2);
	jurl_mimepart *part = janet_getjurlmimepart(argv, 0);
	jurl_mime     *mime = janet_getjurlmime(argv, 1);
	CURLcode ret = curl_mime_subparts(part->mimepart, mime->mime);
	if (ret == CURLE_OK) mime->clean = 0; // no longer needs cleaning
	return jurl_geterror(ret);
}