1
0
mirror of https://github.com/vlang/v.git synced 2023-08-10 21:13:21 +03:00
v/os/os.v
2019-06-26 14:45:03 +02:00

465 lines
8.8 KiB
Go

// 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 C.getline(voidptr, voidptr, voidptr) int
fn C.ftell(fp voidptr) int
fn todo_remove(){}
fn init_os_args(argc int, _argv *byteptr) []string {
mut args := []string
# char** argv = (char**) _argv;
for i := 0; i < argc; i++ {
arg := ''
//arg := tos(argv[i], strlen(argv[i]))
# arg = tos((char**)(argv[i]), strlen((char**)(argv[i])));
args << arg
}
return args
}
fn parse_windows_cmd_line(cmd byteptr) []string {
s := tos2(cmd)
return s.split(' ')
}
// read_file reads the file in `path` and returns the contents.
pub fn read_file(path string) ?string {
mut res := ''
cpath := path.cstr()
fp := C.fopen(cpath, 'r')
if isnil(fp) {
return error('failed to open file "$path"')
}
C.fseek(fp, 0, SEEK_END)
fsize := C.ftell(fp)
// C.fseek(fp, 0, SEEK_SET) // same as C.rewind(fp) below
C.rewind(fp)
mut str := malloc(fsize + 1)
C.fread(str, fsize, 1, fp)
C.fclose(fp)
str[fsize] = 0
res = tos(str, 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 mv(old, new string) {
C.rename(old.cstr(), new.cstr())
}
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)
}
// TODO remove
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)
}
pub 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)
}
pub 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'
max := 5000
$if windows {
# FILE* f = _popen(cmd.str, "r");
}
$else {
# FILE* f = popen(cmd.str, "r");
}
# 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)
}
// `file_exists` returns true if `path` exists.
pub fn file_exists(path string) bool {
// # return access( path.str, F_OK ) != -1 ;
res := false
$if windows {
# res = _access( path.str, 0 ) != -1 ;
}
$else {
# res = access( path.str, 0 ) != -1 ;
}
return res
}
pub fn dir_exists(path string) bool {
dir := C.opendir(path.cstr())
res := !isnil(dir)
if res {
C.closedir(dir)
}
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 {
}
}
*/
fn print_c_errno() {
# printf("errno=%d err='%s'\n", errno, strerror(errno));
}
pub fn get_extension(path string) string {
pos := path.last_index('.')
if pos == -1 {
return ''
}
return path.right(pos)
}
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('/')
}
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) {
$if windows {
return
}
$if 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);
}
}