" File.st -- stream-like accessors for files

  Copyright (c) 2006, 2007 Ian Piumarta
  All rights reserved.

  Permission is hereby granted, free of charge, to any person obtaining a
  copy of this software and associated documentation files (the 'Software'),
  to deal in the Software without restriction, including without limitation
  the rights to use, copy, modify, merge, publish, distribute, and/or sell
  copies of the Software, and to permit persons to whom the Software is
  furnished to do so, provided that the above copyright notice(s) and this
  permission notice appear in all copies of the Software and that both the
  above copyright notice(s) and this permission notice appear in supporting
  documentation.

  THE SOFTWARE IS PROVIDED 'AS IS'.  USE ENTIRELY AT YOUR OWN RISK.

  Last edited: 2009-03-07 20:06:47 by piumarta on emilia
"

{ import: Objects }

{ include "tag.h" }

File : Object ( _fd name )

File isFile	[ ^true ]
Object isFile	[ ^false ]

File withFd: fd
[
    self := self new.
    _fd := fd _integerValue.
    name := '<unknown>'.
]

File withFd: fd name: nameString
[
    self := self new.
    _fd := fd _integerValue.
    name := nameString.
]

File withFd_: _theFd name: nameString
[
    self := self new.
    _fd := _theFd.
    name := nameString.
]

File name					[ ^name ]

File print: anObject					[ anObject printOn: self ]
File println: anObject					[ self print: anObject; cr ]
File print: aNumber base: base				[ aNumber printOn: self base: base ]
File print: aNumber base: base width: width		[ aNumber printOn: self base: base width: width ]
File print: aNumber base: base width: width padded: p	[ aNumber printOn: self base: base width: width padded: p ]
File cr							[ self nextPut: $\n ]
File cr: n						[ n timesRepeat: [self cr] ]
File space						[ self nextPut: $   ]
File space: n						[ n timesRepeat: [self space] ]
File tab						[ self nextPut: $\t ]
File tab: n						[ n timesRepeat: [self tab] ]

File nextPut: aByte
[
    { if ((long)v_aByte & 1) { char c= _I(v_aByte);  _return _O(write((long)self->v__fd, (void *)&c, 1)); } }.
    ^self primitiveFailed
]

File nextPutAll: aString
[
    self next: aString size putAll: aString
]

File next: size putAll: aString
[
    | _bytes _size |
    _bytes := aString _bytes.
    _size  := size _integerValue.
    { _return _O(write((long)self->v__fd, (void *)v__bytes, (size_t)v__size)); }.
]

File print_x: _pointer
{
    char buf[1024];
    int size= sprintf(buf, "%lx", (long)v__pointer);
    _return _O(write((long)self->v__fd, (void *)buf, (size_t)size));
}

File  read: aCollection				[ ^self read: aCollection size: aCollection size ]
File  read: aCollection size: size		[ ^self read: aCollection at: 0 size: size ]
File  read: aCollection at: offset size: size	[ ^SmallInteger value_: (self _read_: aCollection _elements at_: offset _integerValue size_: size _integerValue) ]
File _read_: _bytes at_: _offset size_: _size	{ _return (oop)(long)read((long)self->v__fd, (char *)v__bytes + (long)v__offset, (size_t)v__size); }

File  write: aCollection			[ ^self write: aCollection size: aCollection size ]
File  write: aCollection size: size		[ ^SmallInteger value_: (self _write_: aCollection _elements size_: size _integerValue) ]
File _write_: _bytes size_: _size		{ _return (oop)(long)write((long)self->v__fd, (void *)v__bytes, (size_t)v__size); }

File isInteractive			{ _return (oop)(long)isatty((long)self->v__fd); }

"
File waitForInput: msTimeout
{
    fd_set r, w, x;
    int n= 0;
    struct timeval tv= { ((long)v_msTimeout >> 1) / 1000000, ((long)v_msTimeout >> 1) % 1000000 };
    FD_ZERO(&r);
    FD_ZERO(&w);
    FD_ZERO(&x);
    FD_SET((long)self->v__fd, &r);
    n= select((long)self->v__fd + 1, &r, &w, &x, &tv);
    _return (oop)(n << 1 | 1);
}
"

File open: pathString
[
    ^self open: pathString ifAbsent: [self error: pathString, ': No such file or directory'].
]

File openIfPresent: pathString
[
    ^self open: pathString ifAbsent: [].
]

File open: pathString ifAbsent: errorBlock
[
    | descriptor |
    descriptor := self open_: pathString _stringValue.
    ^descriptor >= 0
	ifTrue:  [self withFd: descriptor name: pathString]
	ifFalse: [errorBlock value]
]

File open_: _path
{
    _return (oop)(((long)open((char *)v__path, O_RDONLY, 0)) << 1 | 1);
}

File create: pathString
[
    ^self create: pathString ifAbsent: [self error: pathString, ': Cannot open for writing'].
]

File create: pathString ifAbsent: errorBlock
[
    | descriptor |
    descriptor := self create_: pathString _stringValue.
    ^descriptor >= 0
	ifTrue:  [File withFd: descriptor name: pathString]
	ifFalse: [errorBlock value]
]

File create_: _path
{
    _return (oop)(((long)open((char *)v__path,
			      O_WRONLY | O_CREAT | O_TRUNC,
#			     if defined(WIN32)
			      0644
#			     else
			      S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH
#			     endif
			      )) << 1 | 1);
}

File close
{
    close((long)self->v__fd);
}

File contents
[
    | buf out n |
    buf := ByteArray new: 8192.
    out := WriteStream on: (ByteArray new: 8192).
    [(n := self read: buf) > 0]
	whileTrue:
	   [out next: n putAll: buf].
    ^out contents
]

"----------------------------------------------------------------"

SharedFile : File ( readStream )

SharedFile readStream
[
    readStream isNil ifTrue: [readStream := super readStream].
    ^readStream
]

File readStream
[
    ^(self isInteractive ifTrue: [ConsoleFileStream] ifFalse: [FileStream]) on: self
]