#include "../lib/io/io.h"
#include "../lib/container/container.h"
#include "../lib/cstr/cstr.h"
#include "../lib/tctl/tctl.h"

#define BUFFER_SIZE 128
char buf[BUFFER_SIZE];

int main(int argc, const char **argv)
{
	u64 size;

	if (argc > 3) {
		wf(STDERR_FD, "write [-nobuf] <filename>\n");
		return -1;
	}

	if (argc == 3) {
		if (cstr_compare(argv[1], "-nobuf") == 0) {

			int fd = open(argv[2], OPEN_WRITE_ONLY | OPEN_CREATE, MODE_USER_READ | MODE_USER_WRITE | MODE_OTHER_READ | MODE_GROUP_READ);

			while ((size = read(STDIN_FD, buf, BUFFER_SIZE))) {
				write(fd, buf, BUFFER_SIZE);
			}

			close(fd);

		} else {
			wstd("write [-nobuf] <filename>\n");
			return -1;
		}
	} else {
		int fd = open(argv[1], OPEN_WRITE_ONLY | OPEN_CREATE, MODE_USER_READ | MODE_USER_WRITE | MODE_OTHER_READ | MODE_GROUP_READ);

		container_t *string = new_container(sizeof(char));

		while ((size = read(STDIN_FD, buf, BUFFER_SIZE))) {
			container_append(string, buf, size);
		}


		container_push(string, 0);

		write(fd, string->data, string->size);

		close(fd);
	}
	return 0;
}