1
0
mirror of https://github.com/vlang/v.git synced 2023-08-10 21:13:21 +03:00
v/vlib/os/os.v
2019-06-30 21:07:24 +02:00

567 lines
10 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>
#include <signal.h>
#include <unistd.h>
#include <dirent.h>
#include <errno.h>
//#include <execinfo.h> // for backtrace_symbols_fd
const (
args = []string
MAX_PATH = 4096
)
const (
FILE_ATTRIBUTE_DIRECTORY = 16 // Windows
)
struct FILE {
}
struct File {
cfile *FILE
}
struct FileInfo {
name string
size int
}
import const (
SEEK_SET
SEEK_END
SA_SIGINFO
SIGSEGV
S_IFMT
S_IFDIR
)
struct C.stat {
st_size int
st_mode int
}
struct C.DIR {
}
struct C.dirent {
d_name byteptr
}
struct C.sigaction {
mut:
sa_mask int
sa_sigaction int
sa_flags int
}
fn C.getline(voidptr, voidptr, voidptr) int
fn C.ftell(fp voidptr) int
fn C.getenv(byteptr) byteptr
fn C.sigaction(int, voidptr, int)
fn todo_remove(){}
fn init_os_args(argc int, argv *byteptr) []string {
mut args := []string
for i := 0; i < argc; i++ {
args << string(argv[i])
}
return args
}
fn parse_windows_cmd_line(cmd byteptr) []string {
s := string(cmd)
return s.split(' ')
}
// read_file reads the file in `path` and returns the contents.
pub fn read_file(path string) ?string {
mut res := ''
mut mode := 'r'
777 // TODO
// Need 'rb' on windows to avoid the \r\n mess.
$if windows {
mode = 'rb'
}
cpath := path.cstr()
fp := C.fopen(cpath, mode.cstr())
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
}
// file_size returns the size of the file located in `path`.
pub fn file_size(path string) int {
s := C.stat{}
C.stat(path.str, &s)
return s.st_size
}
pub fn mv(old, new string) {
C.rename(old.cstr(), new.cstr())
}
// 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_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
}
// TODO return `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_file2(path, '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, mode string) File {
res := File {
cfile: C.fopen(file.cstr(), mode.cstr())
}
if isnil(res.cfile) {
println('coudlnt create file "$file"')
}
return res
}
pub fn (f File) write(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"
pub fn (f File) write_bytes(data voidptr, size int) {
C.fwrite(data, 1, size, f.cfile)
}
pub fn (f File) write_bytes_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) writeln(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 isnil(fp) {
return
}
C.fclose(fp)
}
// system starts the specified command, waits for it to complete, and returns its code.
pub fn system(cmd string) int {
ret := C.system(cmd.cstr())
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')
}
}
// exec starts the specified command, waits for it to complete, and returns its output.
pub fn exec(cmd string) string {
cmd = '$cmd 2>&1'
f := popen(cmd)
if isnil(f) {
// TODO optional or error code
println('popen $cmd failed')
return ''
}
buf := [1000]byte
mut res := ''
for C.fgets(buf, 1000, f) != 0 {
res += tos(buf, strlen(buf))
}
return res.trim_space()
}
// `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 string(s)
}
pub fn setenv(name string, value string, overwrite bool) int {
$if windows {
}
$else {
return C.setenv(name.cstr(), value.cstr(), overwrite)
}
}
pub fn unsetenv(name string) int {
$if windows {
}
$else {
return C.unsetenv(name.cstr())
}
}
// `file_exists` returns true if `path` exists.
pub fn file_exists(path string) bool {
$if windows {
return C._access( path.str, 0 ) != -1
}
return C.access( path.str, 0 ) != -1
}
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, guard string) {
if !path.contains(guard) {
println('rmdir canceled because the path doesnt contain $guard')
return
}
$if !windows {
}
$else {
}
}
*/
fn print_c_errno() {
//C.printf('errno=%d err="%s"\n', errno, C.strerror(errno))
}
pub fn ext(path string) string {
pos := path.last_index('.')
if pos == -1 {
return ''
}
return path.right(pos)
}
fn path_sans_ext(path string) string {
pos := path.last_index('.')
if pos == -1 {
return path
}
return path.left(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('/')
}
// get_line returns a one-line string from stdin
//u64 is used because C.getline needs a size_t as second argument
//Otherwise, it would cause a valgrind warning and may be dangerous
//Malloc takes an int as argument so a cast has to be made
pub fn get_line() string {
$if windows {
panic('get_line() not implemented on Windows yet, sorry!')
}
$else {
max := u64(256)
buf := malloc(int(max))
nr_chars := C.getline(&buf, &max, stdin)
if nr_chars == 0 {
return ''
}
if buf[nr_chars - 1] == `\n` { // newline
return tos(buf, nr_chars - 1)
}
// To prevent cutting end of line if no newline
return tos(buf, nr_chars)
}
}
// get_raw_line returns a one-line string from stdin along with '\n' if there is any
pub fn get_raw_line() string {
$if windows {
panic('get_raw_line() not implemented on Windows yet, sorry!')
}
$else {
max := u64(256)
buf := malloc(int(max))
nr_chars := C.getline(&buf, &max, stdin)
if nr_chars == 0 {
return ''
}
return tos(buf, nr_chars)
}
}
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
}
// write_file writes text data to a file in `path`.
pub fn write_file(path, text string) {
f := os.create(path)
f.writeln(text)
f.close()
}
pub fn clear() {
C.printf('\x1b[2J')
C.printf('\x1b[H')
}
fn on_segfault(f voidptr) {
$if windows {
return
}
$if mac {
mut sa := C.sigaction{}
C.memset(&sa, 0, sizeof(sigaction))
C.sigemptyset(&sa.sa_mask)
sa.sa_sigaction = f
sa.sa_flags = SA_SIGINFO
C.sigaction(SIGSEGV, &sa, 0)
}
}
pub fn getexepath() string {
mut result := [4096]byte // [MAX_PATH]byte --> error byte undefined
$if linux {
count := int(C.readlink('/proc/self/exe', result, MAX_PATH ))
if(count < 0) {
panic('error reading /proc/self/exe to get exe path')
}
return tos(result, count)
}
$if windows {
ret := int(C.GetModuleFileName( 0, result, MAX_PATH ))
return tos( result, ret)
}
$if mac {
//panic('getexepath() not impl')
return ''
}
}
pub fn is_dir(path string) bool {
$if windows {
val := int(C.GetFileAttributes(path.cstr()))
return val &FILE_ATTRIBUTE_DIRECTORY > 0
}
$else {
statbuf := C.stat{}
cstr := path.cstr()
if C.stat(cstr, &statbuf) != 0 {
return false
}
return statbuf.st_mode & S_IFMT == S_IFDIR
}
}
pub fn chdir(path string) {
$if windows {
C._chdir(path.cstr())
}
$else {
C.chdir(path.cstr())
}
}
pub fn getwd() string {
buf := malloc(512)
$if windows {
if C._getcwd(buf, 512) == 0 {
return ''
}
}
$else {
if C.getcwd(buf, 512) == 0 {
return ''
}
}
return string(buf)
}
pub fn ls(path string) []string {
mut res := []string
dir := C.opendir(path.str)
if isnil(dir) {
println('ls() couldnt open dir "$path"')
print_c_errno()
return res
}
mut ent := &C.dirent{!}
for {
ent = C.readdir(dir)
if isnil(ent) {
break
}
name := tos_clone(ent.d_name)
if name != '.' && name != '..' && name != '' {
res << name
}
}
C.closedir(dir)
return res
}
fn log(s string) {
}
pub fn print_backtrace() {
/*
# void *buffer[100];
nptrs := 0
# nptrs = backtrace(buffer, 100);
# printf("%d!!\n", nptrs);
# backtrace_symbols_fd(buffer, nptrs, STDOUT_FILENO) ;
*/
}