#include "str.h"


__attribute__ ((optimize(3))) bool charisalpha(char c){
	return (c >= 'a' && c <= 'z') || (c >= 'A' && c <= 'Z');
}

__attribute__ ((optimize(3))) bool charisnum(char c){
	return c >= '0' && c <= '9';
}

__attribute__ ((optimize(3))) bool charisblank(char c){
	return c == ' ' || c == '\t';
}

__attribute__ ((optimize(3))) bool charislinebreak(char c){
	return c == '\n' || c == '\f' || c == '\v';
}

__attribute__ ((optimize(3))) bool charisspace(char c){
	switch(c){
		case ' ': case '\t': case '\n': case '\r': case '\f': case '\v':
			return true;
		default:
			return false;
	}
}

char lowerchar(char c){
	if(c >= 'A' && c <= 'Z') c += 32;
	return c;
}

u32 lowers(char *s){
	u32 l = -1;
	while(s[++l]) s[l] = lowerchar(s[l]);
	return l;
}

__attribute__ ((optimize(1))) u32 len(const char *s){
	if(s == NULL) return 0;
	u32 l = -1;
	while(s[++l]);
	return l;
}

str dstr(char *s){
	str r = {
		.cap = len(s),
		.len = r.cap,
		.ptr = calloc(r.len+1, sizeof(char))
	};
	if(r.ptr == NULL) r = (str){0};
	else memcpy(r.ptr, s, r.len);
	return r;
}

str dnstr(u32 cap){
	str s = {
		.cap = cap,
		.len = 0,
		.ptr = calloc(cap+1, sizeof(char))
	};
	return s;
}

str dsstr(char *s){
	str r = {
		.cap = 0,
		.len = len(s),
		.ptr = s
	};
	return r;
}

str dsnstr(char *s, u32 len){
	str r = {
		.cap = 0,
		.len = len,
		.ptr = s
	};
	return r;
}

int resize_str(str *s, u32 nsize){
	if(nsize == 0) return 1;
	char *nptr = realloc(s->ptr, nsize);
	if(nptr == NULL) return 1;
	s->ptr = nptr;
	s->cap = nsize;
	return 0;
}

str dup_str(str s){
	// we use len here to account for static strings that have no cap
	str r = {.cap = s.len, .len = s.len, .ptr = calloc(s.len+1, sizeof(char))};
	if(r.ptr == NULL) return (str){0};
	memcpy(r.ptr, s.ptr, s.len);
	return r;
}

u32 lowerstr(str s){
	return lowers(s.ptr);
}

str utostr(u64 n, int b){
	str s = {
		.ptr = calloc(sizeof(n)*8 + 2, sizeof(char))
	};
	if(s.ptr == NULL) return (str){0};
	u64 dn = n, p = 1;
	while(dn /= b) p *= b;
	while(p){
		dn = n/p;
		s.ptr[s.len++] = dn > 9 ? 'A'-10+dn : '0'+dn;
		n %= p, p /= b;
	}
	char *nptr = realloc(s.ptr, s.len+1);
	if(nptr == NULL){
		free(s.ptr);
		return (str){0};
	}
	s.ptr = nptr;
	s.cap = s.len;
	return s;
}

// TODO add support for different bases
// TODO check length
u64 strtou(str s){
	u64 i = 0, r = 0;
	while(charisnum(s.ptr[i])){
		r *= 10;
		r += s.ptr[i++] - '0';
	}
	return r;
}

int streq(str s1, str s2){
	if(s1.len != s2.len){
		return 0;
	}
	u32 i = 0;
	while(i < s1.len){
		if(s1.ptr[i] != s2.ptr[i]){
			return 0;
		}
		i++;
	}
	return 1;
}

int streqn(str s1, str s2, u32 n){
	if(s1.len < n || s2.len < n){
		return 0;
	}
	u32 i = 0;
	while(i < n){
		if(s1.ptr[i] != s2.ptr[i]){
			return 0;
		}
		i++;
	}
	return 1;
}

u64 len_nstrs(u64 n, ...){
	u64 s = 0;
	va_list vl;
	va_start(vl, n);
	for(u64 i = 0; i < n; i++) {
		str t = va_arg(vl, str);
		s += t.len;
	}
	va_end(vl);
	return s;
}

u64 vlen_nstrs(u64 n, va_list args){
	u64 s = 0;
	for(u64 i = 0; i < n; i++) {
		str t = va_arg(args, str);
		s += t.len;
	}
	return s;
}

void copy_nstrs(str dst, u64 n, ...){
	va_list vl;
	va_start(vl, n);
	for(u64 i = 0; i < n; i++) {
		str t = va_arg(vl, str);
		copy_str(dst, t);
	}
	va_end(vl);
}

void vcopy_nstrs(str dst, u64 n, va_list args){
	for(u64 i = 0; i < n; i++) {
		str t = va_arg(args, str);
		copy_str(dst, t);
	}
}

void move_nstrs(str dst, u64 n, ...){
	va_list vl;
	va_start(vl, n);
	for(u64 i = 0; i < n; i++) {
		str t = va_arg(vl, str);
		move_str(dst, t);
	}
	va_end(vl);
}

void vmove_nstrs(str dst, u64 n, va_list args){
	for(u64 i = 0; i < n; i++) {
		str t = va_arg(args, str);
		move_str(dst, t);
	}
}

str dup_nstrs(u64 n, ...){
	va_list vl, vlc;
	va_start(vl, n);
	va_copy(vlc, vl);
	str r = dnstr(vlen_nstrs(n, vlc));
	for(u64 i = 0; i < n; i++) {
		str t = va_arg(vl, str);
		copy_str(r, t);
	}
	va_end(vlc);
	va_end(vl);
	return r;
}

str vdup_nstrs(u64 n, va_list args){
	va_list vlc;
	va_copy(vlc, args);
	str r = dnstr(vlen_nstrs(n, vlc));
	for(u64 i = 0; i < n; i++) {
		str t = va_arg(args, str);
		copy_str(r, t);
	}
	va_end(vlc);
	return r;
}

str read_delim(char *buf, char d){
	u32 i = 0;
	while(buf[i] && buf[i] != d) i++;
	str s = {.cap = i, .len = i, .ptr = calloc(s.cap+1, sizeof(char))};
	if(s.ptr == NULL) return (str){0};
	memcpy(s.ptr, buf, i);
	return s;
}

str sread_delim(char *buf, char d){
	u32 i = 0;
	while(buf[i] && buf[i] != d) i++;
	str s = {.cap = 0, .len = i, .ptr = buf};
	return s;
}

str read_delim_f(char *buf, bool (*func)(char), bool func_cond){
	u32 i = 0;
	while(buf[i] && (func(buf[i]) != func_cond)) i++;
	str s = {.cap = i, .len = i, .ptr = calloc(s.cap+1, sizeof(char))};
	if(s.ptr == NULL) return (str){0};
	memcpy(s.ptr, buf, i);
	return s;
}

str sread_delim_f(char *buf, bool (*func)(char), bool func_cond){
	u32 i = 0;
	while(buf[i] && (func(buf[i]) != func_cond)) i++;
	str s = {.cap = 0, .len = i, .ptr = buf};
	return s;
}

u32 get_line_len(char *buf){
	u32 l = 0;
	while(buf[l] != '\0' && buf[l] != '\n') l++;
	return l;
}

str fd_to_str(int fd){
	u64 len = get_fd_size(fd);
	str s = {
		.cap = len,
		.len = len,
		.ptr = calloc(len+1, sizeof(char))
	};
	if(s.ptr == NULL) return (str){0};
	for(u32 l = len; l != 0; l -= read(fd, s.ptr+(len-l), l));
	return s;
}

str fd_to_nstr(int fd, u32 len){
	str s = {
		.cap = len,
		.len = len,
		.ptr = calloc(len+1, sizeof(char))
	};
	if(s.ptr == NULL) return (str){0};
	for(u32 l = len; l != 0; l -= read(fd, s.ptr+(len-l), l));
	return s;
}

str fp_to_str(FILE *fp){
	u64 len = get_fp_size(fp);
	str s = {
		.cap = len,
		.len = len,
		.ptr = calloc(len+1, sizeof(char))
	};
	if(s.ptr == NULL) return (str){0};
	for(u32 l = len; l != 0; l -= fread(s.ptr, sizeof(char), len, fp));
	return s;
}

str fp_to_nstr(FILE *fp, u32 len){
	str s = {
		.cap = len,
		.len = len,
		.ptr = calloc(len+1, sizeof(char))
	};
	if(s.ptr == NULL) return (str){0};
	for(u32 l = len; l != 0; l -= fread(s.ptr, sizeof(char), len, fp));
	return s;
}

str file_to_str(char *filename){
	u64 len = get_file_size(filename);
	FILE *fp = fopen(filename, "r");
	str s = {
		.cap = len,
		.len = len,
		.ptr = calloc(len+1, sizeof(char))
	};
	if(fp == NULL || s.ptr == NULL) return (str){0};
	for(u32 l = len; l != 0; l -= fread(s.ptr, sizeof(char), len, fp));
	fclose(fp);
	return s;
}

str file_to_nstr(char *filename, u32 len){
	FILE *fp = fopen(filename, "r");
	str s = {
		.cap = len,
		.len = len,
		.ptr = calloc(len+1, sizeof(char))
	};
	if(fp == NULL || s.ptr == NULL) return (str){0};
	for(u32 l = len; l != 0; l -= fread(s.ptr, sizeof(char), len, fp));
	fclose(fp);
	return s;
}

str map_file(char *filename){
	int fd = open(filename, 0);
	if(fd == -1){
		return (str){0};
	}
	str s = {
		.cap = get_fd_size(fd),
		.len = s.cap,
		.ptr = mmap(NULL, s.len, PROT_READ, MAP_SHARED, fd, 0)
	};
	if(s.ptr == MAP_FAILED){
		unmap_file(&s);
	}
	close(fd);
	return s;
}

str map_file_at(char *filename, int len, int at){
	int fd = open(filename, 0);
	if(fd == -1){
		return (str){0};
	}
	str s = {
		.cap = len,
		.len = s.cap,
		.ptr = mmap(NULL, s.len, PROT_READ, MAP_SHARED, fd, at)
	};
	if(s.ptr == MAP_FAILED){
		unmap_file(&s);
	}
	close(fd);
	return s;
}

void unmap_file(str *s){
	munmap(s->ptr, s->len);
	s->ptr = NULL;
	s->len = 0;
	s->cap = 0;
}

str fread_str(int fd, int max){
	str s = dnstr(max);
	s.len = read(fd, s.ptr, s.cap);
	if(s.len == 0){
		free_str(&s);
		return (str){0};
	}
	if(s.ptr[s.len-1] == '\n'){
		s.ptr[--s.len] = '\0';
	}
	if(s.len != s.cap){
		char *p = realloc(s.ptr, s.len+1);
		if(p == NULL){
			free_str(&s);
			return (str){0};
		}
		s.ptr = p;
		s.cap = s.len;
	}
	return s;
}

void print_str(str s){
	for(u32 i = 0; i < s.len; i++){
		putchar(s.ptr[i]);
	}
}

void free_str(str *s){
	if(s->ptr != NULL && s->cap > 0) free(s->ptr);
	s->ptr = NULL;
	s->cap = 0;
	s->len = 0;
}

char *charinstr(char c, str s){
	for(u32 i = 0; i < s.len; i++){
		if(s.ptr[i] == c){
			return s.ptr + i;
		}
	}
	return NULL;
}