input stream: wrap FILE

1 files changed, 44 insertions, 6 deletions

diff --git a/include/kit.inl.h b/include/kit.inl.h
index c50dd61..968d973 100644
--- a/include/kit.inl.h
+++ b/include/kit.inl.h
@@ -1259,6 +1259,7 @@ static kit_bigint_t kit_bi_from_base58(kit_str_t base58) {
  *********************************************************************/
 #ifndef KIT_INPUT_STREAM_H
 #define KIT_INPUT_STREAM_H
+#include <stdio.h>
 #ifdef __cplusplus
 extern "C" {
 #endif
@@ -1270,17 +1271,21 @@ typedef struct {
 } kit_is_handle_t;
 kit_is_handle_t kit_is_wrap_string(kit_str_t       string,
                                    kit_allocator_t alloc);
+kit_is_handle_t kit_is_wrap_file(FILE *f, kit_allocator_t alloc);
 void kit_is_destroy(kit_is_handle_t in);
 #define KIT_IS_WRAP_STRING(string) \
   kit_is_wrap_string((string), kit_alloc_default())
+#define KIT_IS_WRAP_FILE(f) kit_is_wrap_file((f), kit_alloc_default())
 #define KIT_IS_READ(in, destination) \
   (in).read((in).state, (destination))
 #ifndef KIT_DISABLE_SHORT_NAMES
 #  define is_read_fn kit_is_read_fn
 #  define is_handle_t kit_is_handle_t
 #  define is_wrap_string kit_is_wrap_string
+#  define is_wrap_file kit_is_wrap_file
 #  define is_destroy kit_is_destroy
 #  define IS_WRAP_STRING KIT_IS_WRAP_STRING
+#  define IS_WRAP_FILE KIT_IS_WRAP_FILE
 #  define IS_READ KIT_IS_READ
 #endif
 #ifdef __cplusplus
@@ -2644,7 +2649,7 @@ void kit_da_resize(kit_da_void_t *array, ptrdiff_t element_size,
  *                                                                   *
  *********************************************************************/
 #include <string.h>
-enum { input_stream_str };
+enum { KIT_INPUT_STREAM_STR, KIT_INPUT_STREAM_FILE };
 typedef struct {
   ptrdiff_t       type;
   kit_allocator_t alloc;
@@ -2654,12 +2659,17 @@ typedef struct {
   kit_allocator_t alloc;
   kit_str_t       string;
 } kit_is_state_str_t;
-static int check_type(void *state, ptrdiff_t type) {
+typedef struct {
+  ptrdiff_t       type;
+  kit_allocator_t alloc;
+  FILE           *file;
+} kit_is_state_file_t;
+static int kit_is_check_type_(void *state, ptrdiff_t type) {
   kit_is_state_basic_t *basic = (kit_is_state_basic_t *) state;
   return basic != NULL && basic->type == type;
 }
-static ptrdiff_t read_str(void *state, kit_str_t destination) {
-  if (!check_type(state, input_stream_str))
+static ptrdiff_t kit_read_str_(void *state, kit_str_t destination) {
+  if (!kit_is_check_type_(state, KIT_INPUT_STREAM_STR))
     return 0;
   kit_is_state_str_t *str  = (kit_is_state_str_t *) state;
   ptrdiff_t           size = destination.size < str->string.size
@@ -2670,6 +2680,18 @@ static ptrdiff_t read_str(void *state, kit_str_t destination) {
   str->string.size -= size;
   return size;
 }
+static ptrdiff_t kit_read_file_(void *state, kit_str_t destination) {
+  if (!kit_is_check_type_(state, KIT_INPUT_STREAM_FILE))
+    return 0;
+  kit_is_state_file_t *f = (kit_is_state_file_t *) state;
+  if (f->file == NULL || feof(f->file))
+    return 0;
+  ptrdiff_t size = (ptrdiff_t) fread(destination.values, 1,
+                                     destination.size, f->file);
+  if (size <= 0)
+    return 0;
+  return size;
+}
 kit_is_handle_t kit_is_wrap_string(kit_str_t       string,
                                    kit_allocator_t alloc) {
   kit_is_handle_t in;
@@ -2679,11 +2701,27 @@ kit_is_handle_t kit_is_wrap_string(kit_str_t       string,
                          sizeof(kit_is_state_str_t), 0, NULL);
   if (state != NULL) {
     memset(state, 0, sizeof *state);
-    state->type   = input_stream_str;
+    state->type   = KIT_INPUT_STREAM_STR;
     state->string = string;
     state->alloc  = alloc;
     in.state      = state;
-    in.read       = read_str;
+    in.read       = kit_read_str_;
+  }
+  return in;
+}
+kit_is_handle_t kit_is_wrap_file(FILE *f, kit_allocator_t alloc) {
+  kit_is_handle_t in;
+  memset(&in, 0, sizeof in);
+  kit_is_state_file_t *state = (kit_is_state_file_t *)
+      kit_alloc_dispatch(alloc, KIT_ALLOCATE,
+                         sizeof(kit_is_state_file_t), 0, NULL);
+  if (state != NULL) {
+    memset(state, 0, sizeof *state);
+    state->type  = KIT_INPUT_STREAM_FILE;
+    state->file  = f;
+    state->alloc = alloc;
+    in.state     = state;
+    in.read      = kit_read_file_;
   }
   return in;
 }