v/os/os.v

// Copyright (c) 2019 Alexander Medvednikov. All rights reserved.
// Use of this source code is governed by an MIT license
// that can be found in the LICENSE file.

module os

#include <sys/stat.h>
const (
	args = []string
)

struct FILE {
}

struct File {
	cfile *FILE
}

import const (
	SEEK_SET
	SEEK_END
)

fn init_os_args(argc int, c voidptr) []string {
	mut args := []string
	# char** argv = (char**) c;
	for i := 0; i < argc; i++ {
		// # printf("ARG %d = '%s'\n", i, argv[i]);
		arg := ''
		# arg = tos(argv[i], strlen(argv[i]));
		args << arg
	}
	# os__args = args;
	return args
}

fn parse_windows_cmd_line(cmd byteptr) {
	s := tos2(cmd)
	vals := s.split(' ')
	println(vals)
	# os__args = vals;
}

// read_file reads the file in `path` and returns the contents.
// TODO return `?string`
pub fn read_file(path string) string {
	res := ''
	# FILE *f = fopen(path.str, "r");
	# if (!f) return tos("", 0);
	# fseek(f, 0, SEEK_END);
	# long fsize = ftell(f);
	// # fseek(f, 0, SEEK_SET);  //same as rewind(f);
	# rewind(f);
	# char *string = malloc(fsize + 1);
	# fread(string, fsize, 1, f);
	# fclose(f);
	# string[fsize] = 0;
	// # printf("RFILE= %s\n", string);
	# res = tos(string, fsize);
	return res
}

fn (f File) read_rune() string {
	# if (!f.cfile) return tos("", 0);
	c := malloc(1)
	C.fread(c, 1, 1, f.cfile)
	return tos(c, 1)
}

// `file_size` returns the size of the file located in `path`.
pub fn file_size(path string) int {
	# struct stat s;
	# stat(path.str, &s);
	// # if (S_ISLNK(s.st_mode)) return -1;
	# return s.st_size;
	// //////////////////////
	# FILE *f = fopen(path.str, "r");
	# if (!f) return 0;
	# fseek(f, 0, SEEK_END);
	# long fsize = ftell(f);
	// # fseek(f, 0, SEEK_SET);  //same as rewind(f);
	# rewind(f);
	# return fsize;
	return 0
}

pub fn file_last_mod_unix(path string) int {
	# struct stat attr;
	# stat(path.str, &attr);
	# return attr.st_mtime ;
	return 0
}

/* 
pub fn file_last_mod_time(path string) time.Time {
	return time.now()
	q := C.tm{}
	# struct stat attr;
	# stat(path.str, &attr);
	// # q = attr.st_mtime;
	# struct tm * now = localtime(&attr.st_mtime);
	# q = *now;
	# printf("Last modified time: %s", ctime(&attr.st_mtime));
	return time.convert_ctime(q)
}
*/
// read_lines reads the file in `path` into an array of lines.
// TODO return `?[]string` TODO implement `?[]` support
pub fn read_lines(path string) []string {
	mut res := []string
	mut buf := [1000]byte
	cpath := path.cstr()
	fp := C.fopen(cpath, 'rb')
	if isnil(fp) {
		// TODO
		// return error('failed to open file "$path"')
		return res
	}
	for C.fgets(buf, 1000, fp) != 0 {
		mut val := ''
		buf[C.strlen(buf) - 1] = `\0` // eat the newline fgets() stores
		$if windows {
			if buf[strlen(buf)-2] == 13 {
				buf[strlen(buf) - 2] = `\0`
			}
		}
		res << tos_clone(buf)
	}
	C.fclose(fp)
	return res
}

fn read_file_into_ulines(path string) []ustring {
	lines := read_lines(path)
	// mut ulines := new_array(0, lines.len, sizeof(ustring))
	mut ulines := []ustring
	for myline in lines {
		// ulines[i] = ustr
		ulines << myline.ustring()
	}
	return ulines
}

const (
	BUF_SIZE = 5000
)

fn append_to_file(file, s string) {
	# FILE* fp = fopen(file.str, "a");
	# fputs(s.str, fp);
	# fputs("\n", fp);
	# fclose(fp);
}

struct Reader {
	fp *FILE
}

struct FileInfo {
	name string
	size int
}

// fn open(file string) File? {
// return open_file(file)
// }
pub fn open(path string) File {
	return open_file(path)
}

fn open_file(file string) File {
	return create_file2(file, 'r')
}

// `create` creates a file at a specified location and returns a writable `File` object.
pub fn create(path string) File {
	return create_file(path)
}

pub fn open_append(path string) File {
	return create_file(path)
}

fn create_file(file string) File {
	return create_file2(file, 'w')
}

fn create_file_a(file string) File {
	return create_file2(file, 'a')
}

fn open_file_a(file string) File {
	return create_file2(file, 'a')
}

fn create_file2(file string, mode string) File {
	res := File {
		cfile: C.fopen(file.cstr(), mode.cstr())
	}
	if isnil(res.cfile) {
		println('coudlnt create file "$file"')
	}
	return res
}

fn (f File) append(s string) {
	ss := s.clone()
	C.fputs(ss.cstr(), f.cfile)
	// ss.free()
	// C.fwrite(s.str, 1, s.len, f.cfile)
}

// convert any value to []byte (LittleEndian) and write it
// for example if we have write(7, 4), "07 00 00 00" gets written
// write(0x1234, 2) => "34 12"
fn (f File) write(data voidptr, size int) {
	C.fwrite(data, 1, size, f.cfile)
}

fn (f File) write_at(data voidptr, size, pos int) {
	C.fseek(f.cfile, pos, SEEK_SET)
	C.fwrite(data, 1, size, f.cfile)
	C.fseek(f.cfile, 0, SEEK_END)
}

fn (f File) appendln(s string) {
	// C.fwrite(s.str, 1, s.len, f.cfile)
	// ss := s.clone()
	// TODO perf
	C.fputs(s.cstr(), f.cfile)
	// ss.free()
	C.fputs('\n', f.cfile)
}

fn (f File) close() {
	C.fclose(f.cfile)
}

fn close_file(fp *FILE) {
	$if windows {
	}
	# if (fp)
	C.fclose(fp)
}

// `system2` starts the specified command, waits for it to complete, and returns its code.
pub fn system2(cmd string) int {
	cstr := cmd.clone()
	ret := int(C.system(cstr.cstr()))
	// println(' system2 ret=$ret cmd="$s"')
	if ret == -1 {
		os.print_c_errno()
	}
	return ret
}

fn popen(path string) *FILE {
	cpath := path.cstr()
	$if windows {
		return C._popen(cpath, 'r')
	}
	$else {
		return C.popen(cpath, 'r')
	}
}

// TODO rename to run or exec (system doesnt return  a string)
// `system` starts the specified command, waits for it to complete, and returns its output.
// TODO merge the two functions.
pub fn system(cmd string) string {
	// println('OS SYSTEM($s)')
	res := ''
	ss := '$cmd 2>&1'
	_ := 0// TODO DOLLAR TOKEN
	f := popen(ss)// cmd)
	// # if (!f)
	if isnil(f) {
		println('popen $cmd failed')
	}
	max := 1000 
	# char buf[max];
	# while (fgets(buf, max, f) != NULL)  {
	# res = string_add(res, tos(buf, strlen(buf)));
	# }
	return res.trim_space()
}

fn system_into_lines(s string) []string {
	mut res := []string
	cmd := '$s 2>&1'
#ifdef windows
	# FILE* f = _popen(cmd.str, "r");
#else
	# FILE* f = popen(cmd.str, "r");
#endif
	#define MAX 5000
	// # char buf[MAX];
	# char * buf = malloc(sizeof(char) * MAX);
	# while (fgets(buf, MAX, f) != NULL)
	{
		val := ''
		# buf[strlen(buf) - 1] = '\0'; // eat the newline fgets() stores
		# val=tos_clone(buf);
		res << val
	}
	return res
}

// `getenv` returns the value of the environment variable named by the key.
pub fn getenv(key string) string {
	s := C.getenv(key.cstr())
	if isnil(s) {
		return ''
	}
	return tos2(s)
}

fn exit(reason string) {
	println2('exit(): $reason')
	log(reason)
	C.exit(0)
}

fn exit1(reason string) {
	println2('exit(): $reason')
	C.exit(1)
}

// `file_exists` returns true if `path` exists.
pub fn file_exists(path string) bool {
	// # return access( path.str, F_OK ) != -1 ;
	res := false
#ifdef windows
	# res = _access( path.str, 0 ) != -1 ;
#else
	# res = access( path.str, 0 ) != -1 ;
#endif
	return res
}

// `mkdir` creates a new directory with the specified path.
pub fn mkdir(path string) {
	$if windows {
		path = path.replace('/', '\\')
		C.CreateDirectory(path.cstr(), 0)
	}
	$else {
		C.mkdir(path.cstr(), 511)// S_IRWXU | S_IRWXG | S_IRWXO
	}
}

// `rm` removes file in `path`.
pub fn rm(path string) {
	$if windows {
		// os.system2('del /f $path')
	}
	$else {
		C.remove(path.cstr())
	}
	// C.unlink(path.cstr())
}

/* 
// TODO 
fn rmdir(path string, guard string) {
	if !path.contains(guard) {
		println('rmdir canceled because the path doesnt contain $guard')
		return
	}
	$if !windows { 
	} 
	$else { 
	} 
}
*/ 

pub fn unzip(path, out string) {
	$if windows {
		// TODO native string
		// TODO handle older Windows
		// The only way to unzip a file without installing dependencies is to use PowerShell + .NET
		# char *s="powershell.exe -nologo -noprofile -command \"& { Add-Type -A 'System.IO.Compression.FileSystem'; [IO.Compression.ZipFile]::ExtractToDirectory('PATH', 'OUT'); }\" ";
		mut cmd := ''
		# cmd = tos(s, strlen(s));
		cmd = cmd.replace('PATH', path)
		cmd = cmd.replace('OUT', out)
		os.system(cmd)
	}
	$else {
		os.system('unzip -o -d "$out" "$path"')
	}
}

fn print_c_errno() {
	# printf("errno=%d err='%s'\n", errno, strerror(errno));
}

pub fn basedir(path string) string {
	pos := path.last_index('/')
	if pos == -1 {
		return path
	}
	return path.left(pos + 1)
}

pub fn filename(path string) string {
	return path.all_after('/')
}

fn C.getline(voidptr, voidptr, voidptr) int

pub fn get_line() string {
	max := 256
	buf := malloc(max)
	nr_chars := C.getline(&buf, &max, stdin)
	if nr_chars == 0 {
		return ''
	}
	return tos(buf, nr_chars - 1)
}

pub fn user_os() string {
	$if linux {
		return 'linux'
	}
	$if mac {
		return 'mac'
	}
	$if windows {
		return 'windows'
	}
	return 'unknown'
}

// `home_dir` returns path to user's home directory.
pub fn home_dir() string {
	mut home := os.getenv('HOME')
	$if windows {
		home = os.getenv('HOMEDRIVE')
		home += os.getenv('HOMEPATH')
	}
	home += '/'
	return home
}

pub fn write_file(path, text string) {
	f := os.create(path)
	f.appendln(text)
	f.close()
}

fn on_segfault(f voidptr) {
#ifdef windows
	return
#endif
#ifdef mac
	# struct sigaction sa;
	# memset(&sa, 0, sizeof(struct sigaction));
	# sigemptyset(&sa.sa_mask);
	# sa.sa_sigaction = f;
	# sa.sa_flags   = SA_SIGINFO;
	# sigaction(SIGSEGV, &sa, 0);
#endif
}