/****************************************************************************
 * DOH (Dynamic Object Hack)
 * 
 * Author : David Beazley
 *
 * Department of Computer Science        
 * University of Chicago
 * 1100 E 58th Street
 * Chicago, IL  60637
 * beazley@cs.uchicago.edu
 *
 * Please read the file LICENSE for the copyright and terms by which DOH
 * can be used and distributed.
 ****************************************************************************/

#include "dohint.h"
#include <unistd.h>

/* ---------------------------------------------------------------------------
 * $Header$
 * string.c
 *
 * String support.
 * --------------------------------------------------------------------------- */

typedef struct File {
    DOHCOMMON;
    FILE     *filep;       
    int       fd;
    int closeondel;
} File;

/* Forward references */

void    DelFile(DOH *s);
int     File_read(DOH *s, void *buffer, int length);
int     File_write(DOH *s, void *buffer, int length);
int     File_putc(DOH *s, int ch);
int     File_getc(DOH *s);
int     File_ungetc(DOH *s, int ch);
int     File_seek(DOH *s, long offset, int whence);
long    File_tell(DOH *s);


static DohFileMethods FileFileMethods = {
  File_read, 
  File_write,
  File_putc,
  File_getc,
  File_ungetc,
  File_seek,
  File_tell,
  0,              /* close */
};

static DohObjInfo FileType = {
    "File",          /* objname      */
    sizeof(File),    /* objsize      */
    DelFile,         /* doh_del      */
    0,               /* doh_copy     */
    0,               /* doh_clear    */
    0,               /* doh_scope    */
    0,               /* doh_str      */
    0,               /* doh_data     */
    0,               /* doh_dump     */
    0,               /* doh_load     */
    0,               /* doh_len      */
    0,               /* doh_hash     */
    0,               /* doh_cmp      */
    0,               /* doh_mapping  */
    0,               /* doh_sequence */
    &FileFileMethods,/* doh_file     */
    0,               /* doh_string   */
    0,               /* doh_callable */
    0,               /* doh_position */
};

DohObjInfo *File_type() {
  return &FileType;
}

/* -----------------------------------------------------------------------------
 * NewFile(DOH *filename, char *mode)
 * ----------------------------------------------------------------------------- */
DOH *
NewFile(DOH *fn, char *mode)
{
  File *f;
  FILE *file;
  char *filename;

  filename = Char(fn);
  file = fopen(filename,mode);
  if (!file) return 0;

  f = (File *) DohObjMalloc(sizeof(File));
  if (!f) {
    fclose(file);
    return 0;
  }
  f->objinfo = &FileType;
  f->filep = file;
  f->fd = fileno(file);
  f->closeondel = 1;
  return (DOH *) f;
}

/* -----------------------------------------------------------------------------
 * NewFileFromFile(FILE *f)
 * ----------------------------------------------------------------------------- */

DOH *
NewFileFromFile(FILE *file) 
{
  File *f;
  f = (File *) DohObjMalloc(sizeof(File));
  if (!f) return 0;
  f->objinfo = &FileType;
  f->filep = file;
  f->fd = fileno(file);
  f->closeondel = 0;
  return (DOH *) f;
}

/* -----------------------------------------------------------------------------
 * NewFileFromFd(int fd)
 * ----------------------------------------------------------------------------- */

DOH *
NewFileFromFd(int fd)
{
  File *f;
  f = (File *) DohObjMalloc(sizeof(File));
  if (!f) return 0;
  f->objinfo = &FileType;
  f->filep = 0;
  f->fd = fd;
  f->closeondel = 0;
  return (DOH *) f;
}

/* -----------------------------------------------------------------------------
 * DelFile(DOH *s) - Delete a file
 * ----------------------------------------------------------------------------- */
void
DelFile(DOH *so) {
  File *f = (File *) so;
  assert(f->refcount <= 0);
  if (f->closeondel)
    fclose(f->filep);
  DohObjFree(f);
}

/* -----------------------------------------------------------------------------
 * int File_check(DOH *f) - Check if f is a file
 * ----------------------------------------------------------------------------- */
int
File_check(DOH *f) 
{
  File *df;
  if (!DohCheck(f)) return 0;
  
  df = (File *) f;
  if (df->objinfo == &FileType) return 1;
  return 0;
}

/* -----------------------------------------------------------------------------
 * int File_read(DOH *so, void *buffer, int len)
 * 
 * Read data from the File
 * ----------------------------------------------------------------------------- */
int
File_read(DOH *so, void *buffer, int len) {
  File *s = (File *) so;
  if (s->filep) 
    return fread(buffer,1,len,s->filep);
  else 
    return read(s->fd,buffer,len);
}

/* -----------------------------------------------------------------------------
 * int File_write(DOH *so, void *buffer, int len)
 * 
 * Write data to the File
 * ----------------------------------------------------------------------------- */
int
File_write(DOH *so, void *buffer, int len) {
  File *s = (File *) so;
  if (s->filep) 
    return fwrite(buffer,1,len,s->filep);
  else
    return write(s->fd, buffer, len);
}

/* -----------------------------------------------------------------------------
 * int File_seek(DOH *so, long offset, int whence)
 * 
 * Seek to a new position
 * ----------------------------------------------------------------------------- */
int
File_seek(DOH *so, long offset, int whence) {
  File *s = (File *) so;
  if (s->filep) 
    return fseek(s->filep,offset,whence);
  else
    return lseek(s->fd,offset,whence);
}

/* -----------------------------------------------------------------------------
 * long File_tell(DOH *so)
 * 
 * Return current position
 * ----------------------------------------------------------------------------- */
long
File_tell(DOH *so) {
  File *s = (File *) so;
  if (s->filep) 
    return ftell(s->filep);
  else
    return lseek(s->fd,0,SEEK_CUR);
}

/* -----------------------------------------------------------------------------
 * int File_putc(DOH *obj, int ch)
 *
 * Put a character on the output
 * ----------------------------------------------------------------------------- */

int File_putc(DOH *obj, int ch) {
  File *s = (File *) obj;
  if (s->filep) 
    return fputc(ch,s->filep);
  return EOF;
}

/* -----------------------------------------------------------------------------
 * int File_getc(DOH *obj)
 *
 * Get a character
 * ----------------------------------------------------------------------------- */

int File_getc(DOH *obj) {
  File *s = (File *) obj;
  if (s->filep)
    return fgetc(s->filep);
  return EOF;
}

/* -----------------------------------------------------------------------------
 * int File_ungetc(DOH *obj, int ch)
 *
 * Put a character back onto the input
 * ----------------------------------------------------------------------------- */

int File_ungetc(DOH *obj, int ch) {
  File *s = (File *) obj;
  if (s->filep)
    return ungetc(ch, s->filep);
  return EOF;
}