#pragma once

#include <cstdlib>
#include <cstring>
#include <stdio.h>
#include <dirent.h>
#include <sys/stat.h>
#include "typedef.h"
#include "allocator.h"

inline void* file_read( const char* file ) {
  FILE* f = fopen( file, "rb" );
  if( !f )
    return 0;

  defer( fclose( f ) );

  fseek( f, 0, SEEK_END );
  U64 size = ftell( f );
  rewind( f );
  if( !size )
    return 0;

  void* block = malloc( size + 1 );
  fread( block, size, 1, f );

  ( (U8*)block )[size] = 0;
  return block;
}

inline void file_write( const char* file, U8* data, U32 size ) {
  FILE* f = fopen( file, "wb" );
  if( !f )
    return;

  fwrite( data, size, 1, f );
  fclose( f );
}

inline const char* file_path_last_of( const char* path ) {
  const char* last_slash = strrchr( path, '/' );
  return last_slash? last_slash + 1 : path;
}

struct FILE_ENTRY {
  char name[256];
  U8 dir;
};

inline LIST<FILE_ENTRY> dir_get_entries( const char* path ) {
  LIST<FILE_ENTRY> entries;

  DIR* dir = opendir( path );
  if( !dir )
    return entries;

  struct dirent* entry;
  struct stat file_stat;
  char full_path[512];

  while( !!( entry = readdir( dir ) ) ) {
    if( !strcmp( entry->d_name, "." ) || !strcmp( entry->d_name, ".." ) )
      continue;

    snprintf( full_path, sizeof(full_path), "%s/%s", path, entry->d_name);
    if( stat( full_path, &file_stat ) )
      continue;

    FILE_ENTRY file_entry;
    strncpy( file_entry.name, entry->d_name, sizeof(file_entry.name) - 1 );
    file_entry.name[sizeof(file_entry.name) - 1] = '\0';
    file_entry.dir = S_ISDIR( file_stat.st_mode ) ? 1 : 0;
    entries.push( file_entry );
  }

  closedir( dir );
  return entries;
}