X-Git-Url: https://harrygodden.com/git/?a=blobdiff_plain;f=vg_io.h;h=06aee757179ca82c525c8c9c3f49368603351a5e;hb=13737a7a9faa5b31696c711f153b7de4201c404e;hp=779b8726c09db9e81d45063422518bde1f3bbe92;hpb=651edda3736812c89f43c11319c6b485ff14ea19;p=vg.git

diff --git a/vg_io.h b/vg_io.h
index 779b872..06aee75 100644
--- a/vg_io.h
+++ b/vg_io.h
@@ -3,12 +3,131 @@
 #ifndef VG_IO_H
 #define VG_IO_H
 
+#include "vg.h"
 #include "vg_stdint.h"
 #include "vg_platform.h"
 #include "vg_log.h"
 #include "vg_mem.h"
-#include <stdio.h>
-#include <errno.h>
+
+
+typedef struct vg_dir vg_dir;
+#ifndef _WIN32
+ #include <dirent.h>
+ struct vg_dir{
+   DIR *h;
+   struct dirent *data;
+   u32 index;
+ };
+#else
+ #include <windows.h>
+ #include <fileapi.h>
+ struct vg_dir{
+   HANDLE h;
+   WIN32_FIND_DATA data;
+   u32 index;
+ };
+#endif
+
+enum vg_entry_type{
+   k_vg_entry_type_unknown,
+   k_vg_entry_type_file,
+   k_vg_entry_type_dir
+};
+
+static int vg_dir_open( vg_dir *dir, const char *name ){
+#ifdef _WIN32
+   char q_buf[4096];
+   vg_str q;
+   vg_strnull( &q, q_buf, 4096 );
+   vg_strcat( &q, name );
+   vg_strcat( &q, "/*" );
+   if( !vg_strgood(&q) ) return 0;
+
+   vg_info( "FindFirstFile( '%s' )\n", q.buffer );
+   dir->h = FindFirstFile( q.buffer, &dir->data );
+   if( dir->h == INVALID_HANDLE_VALUE ){
+      if( GetLastError() == ERROR_FILE_NOT_FOUND ){
+         dir->index = 0;
+         return 1;
+      }
+      else return 0;
+   }
+#else
+   dir->h = opendir( name );
+   if( !dir->h ) return 0;
+#endif
+   dir->index = 1;
+   return 1;
+}
+
+static const char *vg_dir_entry_name( vg_dir *dir ){
+#ifdef _WIN32
+   return dir->data.cFileName;
+#else
+   return dir->data->d_name;
+#endif
+}
+
+static int vg_dirskip( vg_dir *dir ){
+   const char *s = vg_dir_entry_name(dir);
+#ifdef _WIN32
+   if( dir->data.dwFileAttributes & FILE_ATTRIBUTE_HIDDEN ) return 1;
+#endif
+   if( s[0] == '.' ){
+      if( s[1] == '\0' ) return 1;
+      else if( s[1] == '.' ){
+         if( s[2] == '\0' ) return 1;
+      }
+   }
+   return 0;
+}
+
+static int vg_dir_next_entry( vg_dir *dir ){
+#ifdef _WIN32
+   if( dir->index == 0 ) return 0;
+   if( dir->index > 1 ) {
+      dir->index ++;
+      if( !FindNextFile( dir->h, &dir->data ) ) return 0;
+   }
+   while( vg_dirskip(dir) ){
+      dir->index ++;
+      if( !FindNextFile( dir->h, &dir->data ) ) return 0;
+   }
+   if( dir->index == 1 ) dir->index ++;
+   return 1;
+#else
+   while( (dir->data = readdir(dir->h)) ){
+      dir->index ++;
+      if( !vg_dirskip(dir) ) break;
+   }
+   if( dir->data ) return 1;
+   else return 0;
+#endif
+}
+
+static enum vg_entry_type vg_dir_entry_type( vg_dir *dir ){
+#ifdef _WIN32
+   if( dir->data.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY ) 
+      return k_vg_entry_type_dir;
+   return k_vg_entry_type_file; /* sketchy? */
+#else
+   if( dir->data->d_type == DT_DIR ) return k_vg_entry_type_dir;
+   if( dir->data->d_type == DT_REG ) return k_vg_entry_type_file;
+#endif
+   return 0;
+}
+
+static void vg_dir_close( vg_dir *dir ){
+#ifdef _WIN32
+   if( dir->index ) FindClose( dir->h );
+   dir->h = INVALID_HANDLE_VALUE;
+#else
+   closedir( dir->h );
+   dir->h = NULL;
+   dir->data = NULL;
+#endif
+   dir->index = 0;
+}
 
 /*
  * File I/O
@@ -18,12 +137,10 @@
 
 VG_STATIC void vg_file_print_invalid( FILE *fp )
 {
-   if( feof( fp )) 
-   {
+   if( feof( fp )) {
       vg_error( "mdl_open: header too short\n" );
    }
-   else
-   {
+   else{
       if( ferror( fp ))
          vg_error( "mdl_open: %s\n", strerror(errno) );
       else
@@ -68,11 +185,9 @@ VG_STATIC void *vg_file_read( void *lin_alloc, const char *path, u32 *size )
 		fclose( f );
 
       *size = (u32)current;
-      
       return buffer;
 	}
-	else
-	{
+	else{
       vg_error( "vg_disk_open_read: %s\n", strerror(errno) );
 		return NULL;
 	}
@@ -96,19 +211,39 @@ VG_STATIC char *vg_file_read_text( void *lin_alloc, const char *path, u32 *sz )
 }
 
 
-VG_STATIC int vg_asset_write( const char *path, void *data, i64 size )
-{
+VG_STATIC int vg_asset_write( const char *path, void *data, i64 size ){
 	FILE *f = fopen( path, "wb" );
-	if( f )
-	{
+	if( f ){
 		fwrite( data, size, 1, f );
 		fclose( f );
 		return 1;
 	}
-	else
-	{
+	else{
 		return 0;
 	}
 }
 
+/* TODO: error handling if read fails */
+VG_STATIC int vg_file_copy( const char *src, const char *dst, void *lin_alloc )
+{
+   vg_info( "vg_file_copy( %s -> %s )\n", src, dst );
+   u32 size;
+   void *data = vg_file_read( lin_alloc, src, &size );
+   return vg_asset_write( dst, data, size );
+}
+
+VG_STATIC const char *vg_path_filename( const char *path )
+{
+   const char *base = path;
+   
+   for( int i=0; i<1024; i++ ){
+      if( path[i] == '\0' ) break;
+      if( path[i] == '/' ){
+         base = path+i+1;
+      }
+   }
+
+   return base;
+}
+
 #endif /* VG_IO_H */