Cory McWilliams
7091b6e6a5
git-svn-id: https://www.unprompted.com/svn/projects/tildefriends/trunk@4154 ed5197a5-7fde-0310-b194-c3ffbd925b24
398 lines
12 KiB
C
398 lines
12 KiB
C
#include "file.js.h"
|
|
|
|
#include "mem.h"
|
|
#include "task.h"
|
|
#include "util.js.h"
|
|
|
|
#include <stdbool.h>
|
|
#include <string.h>
|
|
#include <uv.h>
|
|
|
|
#ifdef _WIN32
|
|
#include <windows.h>
|
|
#else
|
|
#include <dirent.h>
|
|
#include <unistd.h>
|
|
#endif
|
|
|
|
static JSValue _file_make_directory(JSContext* context, JSValueConst this_val, int argc, JSValueConst* argv);
|
|
static JSValue _file_remove_directory(JSContext* context, JSValueConst this_val, int argc, JSValueConst* argv);
|
|
static JSValue _file_read_file(JSContext* context, JSValueConst this_val, int argc, JSValueConst* argv);
|
|
static JSValue _file_rename_file(JSContext* context, JSValueConst this_val, int argc, JSValueConst* argv);
|
|
static JSValue _file_stat(JSContext* context, JSValueConst this_val, int argc, JSValueConst* argv);
|
|
static JSValue _file_unlink_file(JSContext* context, JSValueConst this_val, int argc, JSValueConst* argv);
|
|
static JSValue _file_write_file(JSContext* context, JSValueConst this_val, int argc, JSValueConst* argv);
|
|
|
|
static double _time_spec_to_double(const uv_timespec_t* time_spec);
|
|
static void _file_on_stat_complete(uv_fs_t* request);
|
|
|
|
typedef struct file_stat_t {
|
|
void* _task;
|
|
JSContext* _context;
|
|
promiseid_t _promise;
|
|
uv_fs_t _request;
|
|
} file_stat_t;
|
|
|
|
typedef struct fs_req_t {
|
|
uv_fs_t fs;
|
|
uv_file file;
|
|
size_t size;
|
|
char buffer[];
|
|
} fs_req_t;
|
|
|
|
void tf_file_register(JSContext* context)
|
|
{
|
|
JSValue global = JS_GetGlobalObject(context);
|
|
JSValue file = JS_NewObject(context);
|
|
JS_SetPropertyStr(context, global, "File", file);
|
|
JS_SetPropertyStr(context, file, "readFile", JS_NewCFunction(context, _file_read_file, "readFile", 1));
|
|
JS_SetPropertyStr(context, file, "writeFile", JS_NewCFunction(context, _file_write_file, "writeFile", 2));
|
|
JS_SetPropertyStr(context, file, "makeDirectory", JS_NewCFunction(context, _file_make_directory, "makeDirectory", 1));
|
|
JS_SetPropertyStr(context, file, "removeDirectory", JS_NewCFunction(context, _file_remove_directory, "removeDirectory", 1));
|
|
JS_SetPropertyStr(context, file, "unlinkFile", JS_NewCFunction(context, _file_unlink_file, "unlinkFile", 1));
|
|
JS_SetPropertyStr(context, file, "renameFile", JS_NewCFunction(context, _file_rename_file, "renameFile", 2));
|
|
JS_SetPropertyStr(context, file, "stat", JS_NewCFunction(context, _file_stat, "stat", 1));
|
|
JS_FreeValue(context, global);
|
|
}
|
|
|
|
static const int k_file_read_max = 8 * 1024 * 1024;
|
|
|
|
static void _file_async_close_callback(uv_fs_t* req)
|
|
{
|
|
uv_fs_req_cleanup(req);
|
|
tf_free(req);
|
|
}
|
|
|
|
static void _file_read_read_callback(uv_fs_t* req)
|
|
{
|
|
uv_fs_req_cleanup(req);
|
|
fs_req_t* fsreq = (fs_req_t*)req;
|
|
tf_task_t* task = req->loop->data;
|
|
JSContext* context = tf_task_get_context(task);
|
|
promiseid_t promise = (promiseid_t)(intptr_t)req->data;
|
|
if (req->result >= 0)
|
|
{
|
|
JSValue array = tf_util_new_uint8_array(context, (const uint8_t*)fsreq->buffer, req->result);
|
|
tf_task_resolve_promise(task, promise, array);
|
|
JS_FreeValue(context, array);
|
|
}
|
|
else
|
|
{
|
|
tf_task_reject_promise(task, promise, JS_ThrowInternalError(context, "%s", uv_strerror(req->result)));
|
|
}
|
|
int result = uv_fs_close(req->loop, req, fsreq->file, _file_async_close_callback);
|
|
if (result < 0)
|
|
{
|
|
uv_fs_req_cleanup(req);
|
|
tf_free(fsreq);
|
|
}
|
|
}
|
|
|
|
static void _file_read_open_callback(uv_fs_t* req)
|
|
{
|
|
uv_fs_req_cleanup(req);
|
|
fs_req_t* fsreq = (fs_req_t*)req;
|
|
tf_task_t* task = req->loop->data;
|
|
JSContext* context = tf_task_get_context(task);
|
|
promiseid_t promise = (promiseid_t)(intptr_t)req->data;
|
|
if (req->result >= 0)
|
|
{
|
|
uv_buf_t buf = { .base = fsreq->buffer, .len = fsreq->size };
|
|
fsreq->file = req->result;
|
|
int result = uv_fs_read(req->loop, req, fsreq->file, &buf, 1, 0, _file_read_read_callback);
|
|
if (result < 0)
|
|
{
|
|
tf_task_reject_promise(task, promise, JS_ThrowInternalError(context, "%s", uv_strerror(result)));
|
|
result = uv_fs_close(req->loop, req, fsreq->file, _file_async_close_callback);
|
|
if (result < 0)
|
|
{
|
|
uv_fs_req_cleanup(req);
|
|
tf_free(fsreq);
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
tf_task_reject_promise(task, promise, JS_ThrowInternalError(context, "%s", uv_strerror(req->result)));
|
|
uv_fs_req_cleanup(req);
|
|
tf_free(req);
|
|
}
|
|
}
|
|
|
|
static JSValue _file_read_file(JSContext* context, JSValueConst this_val, int argc, JSValueConst* argv)
|
|
{
|
|
void* task = JS_GetContextOpaque(context);
|
|
const char* file_name = JS_ToCString(context, argv[0]);
|
|
|
|
promiseid_t promise = -1;
|
|
JSValue promise_value = tf_task_allocate_promise(task, &promise);
|
|
fs_req_t* req = tf_malloc(sizeof(fs_req_t) + k_file_read_max);
|
|
*req = (fs_req_t)
|
|
{
|
|
.fs =
|
|
{
|
|
.data = (void*)(intptr_t)promise,
|
|
},
|
|
.size = k_file_read_max,
|
|
};
|
|
int result = uv_fs_open(tf_task_get_loop(task), &req->fs, file_name, UV_FS_O_RDONLY, 0, _file_read_open_callback);
|
|
if (result < 0)
|
|
{
|
|
tf_task_reject_promise(task, promise, JS_ThrowInternalError(context, "%s", uv_strerror(result)));
|
|
uv_fs_req_cleanup(&req->fs);
|
|
tf_free(req);
|
|
}
|
|
JS_FreeCString(context, file_name);
|
|
return promise_value;
|
|
}
|
|
|
|
static void _file_write_write_callback(uv_fs_t* req)
|
|
{
|
|
uv_fs_req_cleanup(req);
|
|
fs_req_t* fsreq = (fs_req_t*)req;
|
|
tf_task_t* task = req->loop->data;
|
|
JSContext* context = tf_task_get_context(task);
|
|
promiseid_t promise = (promiseid_t)(intptr_t)req->data;
|
|
if (req->result >= 0)
|
|
{
|
|
tf_task_resolve_promise(task, promise, JS_NewInt64(context, req->result));
|
|
}
|
|
else
|
|
{
|
|
tf_task_reject_promise(task, promise, JS_ThrowInternalError(context, "%s", uv_strerror(req->result)));
|
|
}
|
|
int result = uv_fs_close(req->loop, req, fsreq->file, _file_async_close_callback);
|
|
if (result < 0)
|
|
{
|
|
uv_fs_req_cleanup(req);
|
|
tf_free(fsreq);
|
|
}
|
|
}
|
|
|
|
static void _file_write_open_callback(uv_fs_t* req)
|
|
{
|
|
fs_req_t* fsreq = (fs_req_t*)req;
|
|
uv_fs_req_cleanup(req);
|
|
tf_task_t* task = req->loop->data;
|
|
JSContext* context = tf_task_get_context(task);
|
|
promiseid_t promise = (promiseid_t)(intptr_t)req->data;
|
|
if (req->result >= 0)
|
|
{
|
|
uv_buf_t buf = { .base = fsreq->buffer, .len = fsreq->size };
|
|
fsreq->file = req->result;
|
|
int result = uv_fs_write(req->loop, req, fsreq->file, &buf, 1, 0, _file_write_write_callback);
|
|
if (result < 0)
|
|
{
|
|
tf_task_reject_promise(task, promise, JS_ThrowInternalError(context, "%s", uv_strerror(result)));
|
|
result = uv_fs_close(req->loop, req, fsreq->file, _file_async_close_callback);
|
|
if (result < 0)
|
|
{
|
|
uv_fs_req_cleanup(req);
|
|
tf_free(fsreq);
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
tf_task_reject_promise(task, promise, JS_ThrowInternalError(context, "%s", uv_strerror(req->result)));
|
|
uv_fs_req_cleanup(req);
|
|
tf_free(req);
|
|
}
|
|
}
|
|
|
|
static JSValue _file_write_file(JSContext* context, JSValueConst this_val, int argc, JSValueConst* argv)
|
|
{
|
|
void* task = JS_GetContextOpaque(context);
|
|
const char* file_name = JS_ToCString(context, argv[0]);
|
|
|
|
size_t size;
|
|
uint8_t* buffer = tf_util_try_get_array_buffer(context, &size, argv[1]);
|
|
bool is_array_buffer = false;
|
|
if (buffer)
|
|
{
|
|
is_array_buffer = true;
|
|
}
|
|
else
|
|
{
|
|
buffer = (uint8_t*)JS_ToCStringLen(context, &size, argv[1]);
|
|
}
|
|
|
|
promiseid_t promise = -1;
|
|
JSValue promise_value = tf_task_allocate_promise(task, &promise);
|
|
fs_req_t* req = tf_malloc(sizeof(fs_req_t) + size);
|
|
*req = (fs_req_t)
|
|
{
|
|
.fs =
|
|
{
|
|
.data = (void*)(intptr_t)promise,
|
|
},
|
|
.size = size,
|
|
};
|
|
memcpy(req->buffer, buffer, size);
|
|
if (!is_array_buffer)
|
|
{
|
|
JS_FreeCString(context, (const char*)buffer);
|
|
}
|
|
|
|
int result = uv_fs_open(tf_task_get_loop(task), &req->fs, file_name, UV_FS_O_CREAT | UV_FS_O_WRONLY, 0644, _file_write_open_callback);
|
|
if (result < 0)
|
|
{
|
|
tf_task_reject_promise(task, promise, JS_ThrowInternalError(context, "%s", uv_strerror(result)));
|
|
}
|
|
JS_FreeCString(context, file_name);
|
|
return promise_value;
|
|
}
|
|
|
|
static void _file_async_callback(uv_fs_t* req)
|
|
{
|
|
uv_fs_req_cleanup(req);
|
|
tf_task_t* task = req->loop->data;
|
|
JSContext* context = tf_task_get_context(task);
|
|
promiseid_t promise = (promiseid_t)(intptr_t)req->data;
|
|
if (req->result == 0)
|
|
{
|
|
tf_task_resolve_promise(task, promise, JS_NewInt32(context, req->result));
|
|
}
|
|
else
|
|
{
|
|
tf_task_reject_promise(task, promise, JS_NewInt32(context, req->result));
|
|
}
|
|
tf_free(req);
|
|
}
|
|
|
|
static JSValue _file_rename_file(JSContext* context, JSValueConst this_val, int argc, JSValueConst* argv)
|
|
{
|
|
void* task = JS_GetContextOpaque(context);
|
|
const char* old_name = JS_ToCString(context, argv[0]);
|
|
const char* new_name = JS_ToCString(context, argv[1]);
|
|
promiseid_t promise = -1;
|
|
JSValue promise_value = tf_task_allocate_promise(task, &promise);
|
|
uv_fs_t* req = tf_malloc(sizeof(uv_fs_t));
|
|
*req = (uv_fs_t)
|
|
{
|
|
.data = (void*)(intptr_t)promise,
|
|
};
|
|
int result = uv_fs_rename(tf_task_get_loop(task), req, old_name, new_name,_file_async_callback);
|
|
JS_FreeCString(context, old_name);
|
|
JS_FreeCString(context, new_name);
|
|
if (result < 0)
|
|
{
|
|
tf_task_reject_promise(task, promise, JS_NewInt32(context, result));
|
|
}
|
|
return promise_value;
|
|
}
|
|
|
|
static JSValue _file_unlink_file(JSContext* context, JSValueConst this_val, int argc, JSValueConst* argv)
|
|
{
|
|
void* task = JS_GetContextOpaque(context);
|
|
const char* file_name = JS_ToCString(context, argv[0]);
|
|
promiseid_t promise = -1;
|
|
JSValue promise_value = tf_task_allocate_promise(task, &promise);
|
|
uv_fs_t* req = tf_malloc(sizeof(uv_fs_t));
|
|
*req = (uv_fs_t)
|
|
{
|
|
.data = (void*)(intptr_t)promise,
|
|
};
|
|
int result = uv_fs_unlink(tf_task_get_loop(task), req, file_name, _file_async_callback);
|
|
JS_FreeCString(context, file_name);
|
|
if (result < 0)
|
|
{
|
|
tf_task_reject_promise(task, promise, JS_NewInt32(context, result));
|
|
}
|
|
return promise_value;
|
|
}
|
|
|
|
JSValue _file_make_directory(JSContext* context, JSValueConst this_val, int argc, JSValueConst* argv)
|
|
{
|
|
void* task = JS_GetContextOpaque(context);
|
|
const char* directory = JS_ToCString(context, argv[0]);
|
|
|
|
promiseid_t promise = -1;
|
|
JSValue promise_value = tf_task_allocate_promise(task, &promise);
|
|
uv_fs_t* req = tf_malloc(sizeof(uv_fs_t));
|
|
*req = (uv_fs_t)
|
|
{
|
|
.data = (void*)(intptr_t)promise,
|
|
};
|
|
int result = uv_fs_mkdir(tf_task_get_loop(task), req, directory, 0755, _file_async_callback);
|
|
JS_FreeCString(context, directory);
|
|
if (result < 0)
|
|
{
|
|
tf_task_reject_promise(task, promise, JS_NewInt32(context, result));
|
|
}
|
|
return promise_value;
|
|
}
|
|
|
|
JSValue _file_remove_directory(JSContext* context, JSValueConst this_val, int argc, JSValueConst* argv)
|
|
{
|
|
void* task = JS_GetContextOpaque(context);
|
|
const char* directory = JS_ToCString(context, argv[0]);
|
|
|
|
promiseid_t promise = -1;
|
|
JSValue promise_value = tf_task_allocate_promise(task, &promise);
|
|
uv_fs_t* req = tf_malloc(sizeof(uv_fs_t));
|
|
*req = (uv_fs_t)
|
|
{
|
|
.data = (void*)(intptr_t)promise,
|
|
};
|
|
int result = uv_fs_rmdir(tf_task_get_loop(task), req, directory, _file_async_callback);
|
|
JS_FreeCString(context, directory);
|
|
if (result < 0)
|
|
{
|
|
tf_task_reject_promise(task, promise, JS_NewInt32(context, result));
|
|
}
|
|
return promise_value;
|
|
}
|
|
|
|
JSValue _file_stat(JSContext* context, JSValueConst this_val, int argc, JSValueConst* argv)
|
|
{
|
|
void* task = JS_GetContextOpaque(context);
|
|
const char* path = JS_ToCString(context, argv[0]);
|
|
promiseid_t promise = -1;
|
|
JSValue promise_value = tf_task_allocate_promise(task, &promise);
|
|
|
|
file_stat_t* data = tf_malloc(sizeof(file_stat_t));
|
|
data->_task = task;
|
|
data->_promise = promise;
|
|
data->_request.data = data;
|
|
data->_context = context;
|
|
|
|
int result = uv_fs_stat(tf_task_get_loop(task), &data->_request, path, _file_on_stat_complete);
|
|
if (result)
|
|
{
|
|
tf_task_reject_promise(task, promise, JS_NewInt32(context, result));
|
|
uv_fs_req_cleanup(&data->_request);
|
|
tf_free(data);
|
|
}
|
|
JS_FreeCString(context, path);
|
|
return promise_value;
|
|
}
|
|
|
|
static double _time_spec_to_double(const uv_timespec_t* time_spec)
|
|
{
|
|
return time_spec->tv_sec + (double)(time_spec->tv_nsec) / 1e9;
|
|
}
|
|
|
|
static void _file_on_stat_complete(uv_fs_t* request)
|
|
{
|
|
file_stat_t* data = (file_stat_t*)(request->data);
|
|
JSContext* context = data->_context;
|
|
|
|
if (request->result)
|
|
{
|
|
tf_task_reject_promise(data->_task, data->_promise, JS_NewInt32(context, request->result));
|
|
}
|
|
else
|
|
{
|
|
JSValue result = JS_NewObject(context);
|
|
JS_SetPropertyStr(context, result, "mtime", JS_NewFloat64(context, _time_spec_to_double(&request->statbuf.st_mtim)));
|
|
JS_SetPropertyStr(context, result, "ctime", JS_NewFloat64(context, _time_spec_to_double(&request->statbuf.st_ctim)));
|
|
JS_SetPropertyStr(context, result, "atime", JS_NewFloat64(context, _time_spec_to_double(&request->statbuf.st_atim)));
|
|
JS_SetPropertyStr(context, result, "size", JS_NewFloat64(context, request->statbuf.st_size));
|
|
tf_task_resolve_promise(data->_task, data->_promise, result);
|
|
JS_FreeValue(context, result);
|
|
}
|
|
uv_fs_req_cleanup(request);
|
|
tf_free(data);
|
|
}
|