#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <error.h>
#include <signal.h>
#include <cjson/cJSON.h>
#include <bencoding.c>
#define S0(x) (x ? x : "")
int main (int argc, char ** argv) {
	if (argc < 1+1)
		error_at_line(1, 0, __FILE__, __LINE__, "%s encode < json || %s decode < bencoding || %s path path/to/obj < bencoding || %s foreach < bencoding", S0(argv[0]), S0(argv[0]), S0(argv[0]), S0(argv[0]));
	if (argv[1][0] == 'p' && argc != 1+2)
		error_at_line(1, 0, __FILE__, __LINE__, "set path!");
	int size = 2048;
	int len = 0;
	char * in = malloc(size);
	while (!feof(stdin) && !ferror(stdin)) {
		if (!in)
			error_at_line(2, 0, __FILE__, __LINE__, "heap alloc failed");
		len += fread(in+len, 1, size-len-1, stdin);
		if ((size - len) < 1024)
			in = realloc(in, size *= 2);
	}
	if (argv[1][0] == 'e')
		error_at_line(3, 0, __FILE__, __LINE__, "N/I");
	struct bencoding * bencoding = bdecode(in, size, terminate);
	if (argv[1][0] == 'd') {
		len = b2json_length(bencoding);
		char out[len+1];
		char * end = b2json(out, bencoding);
		*end = '\0';
		puts(out);
		if (end - out != len)
			error_at_line(4, 0, __FILE__, __LINE__, "b2json wrote %ld instead of %d bytes.", end-out, len);
		fprintf(stderr, "len: %d\n", len);
	}
	if (argv[1][0] == 'p') {
		len = b2json_length(bpath(bencoding, argv[2]));
		char out[len+1];
		char * end = b2json(out, bpath(bencoding, argv[2]));
		*end = '\0';
		puts(out);
		if (end - out != len)
			error_at_line(4, 0, __FILE__, __LINE__, "b2json wrote %ld instead of %d bytes.", end-out, len);
		fprintf(stderr, "len: %d\n", len);
	}
	if (argv[1][0] == 'f') {
		bforeach (value, bencoding) {
			len = b2json_length(value->key);
			char out[len+1];
			char * end = b2json(out, value->key);
			*end = '\0';
			if (end - out != len)
				error_at_line(4, 0, __FILE__, __LINE__, "b2json wrote %ld instead of %d bytes.", end-out, len);
			printf("key(%d): %s\n", len, out);
			len = b2json_length(value);
			char out2[len+1];
			end = b2json(out2, value);
			*end = '\0';
			if (end - out2 != len)
				error_at_line(4, 0, __FILE__, __LINE__, "b2json wrote %ld instead of %d bytes.", end-out2, len);
			printf("value(%d): %s\n", len, out2);
		}
	}
	free_bencoding(bencoding);
	free(in);
	return 0;
}