public const int DEFAULT_BUFFER_SIZE = 8192;

private static readonly object _blockingIOErrorKey = new object();

private static readonly object _unsupportedOperationKey = new object();

private const int O_RDONLY = 0x0000;

private const int O_WRONLY = 0x0001;

private const int O_RDWR = 0x0002;

private const int O_APPEND = 0x0008;

private const int O_CREAT = 0x0100;

private const int O_TRUNC = 0x0200;

private const int O_EXCL = 0x0400;

[SpecialName]

public static void PerformModuleReload(PythonContext/*!*/ context, PythonDictionary/*!*/ dict) {

context.EnsureModuleException(

_unsupportedOperationKey,

new PythonType[] { PythonExceptions.ValueError, PythonExceptions.OSError },

typeof(PythonExceptions.BaseException),

dict,

"UnsupportedOperation",

"io"

);

}

[PythonType, DontMapGetMemberNamesToDir]

public class _IOBase : IDisposable, IEnumerator<object>, IEnumerable<object>, IWeakReferenceable, IDynamicMetaObjectProvider, IPythonExpandable {

private bool _closed;

internal CodeContext/*!*/ context;

public _IOBase(CodeContext/*!*/ context) {

this.context = context;

}

#region Public API

public void __del__(CodeContext/*!*/ context) {

close(context);

}

public _IOBase __enter__() {

_checkClosed();

return this;

}

public void __exit__(CodeContext/*!*/ context, params object[] excinfo) {

close(context);

}

public void _checkClosed() {

_checkClosed(null);

}

public void _checkClosed(string msg) {

if (closed) {

throw PythonOps.ValueError(msg ?? "I/O operation on closed file.");

}

}

public void _checkReadable() {

_checkReadable(null);

}

public void _checkReadable(string msg) {

if (!readable(context)) {

throw PythonOps.ValueError(msg ?? "File or stream is not readable.");

}

}

public void _checkSeekable() {

_checkSeekable(null);

}

public void _checkSeekable(string msg) {

if (!seekable(context)) {

throw PythonOps.ValueError(msg ?? "File or stream is not seekable.");

}

}

public void _checkWritable() {

_checkWritable(null);

}

public void _checkWritable(string msg) {

if (!writable(context)) {

throw PythonOps.ValueError(msg ?? "File or stream is not writable.");

}

}

public virtual void close(CodeContext/*!*/ context) {

try {

if (!_closed) {

flush(context);

}

} finally {

_closed = true;

}

}

public virtual bool closed {

get { return _closed; }

}

public virtual int fileno(CodeContext/*!*/ context) {

throw UnsupportedOperation(context, "fileno");

}

public virtual void flush(CodeContext/*!*/ context) {

_checkClosed();

}

public virtual bool isatty(CodeContext/*!*/ context) {

_checkClosed();

return false;

}

[PythonHidden]

public virtual Bytes peek(CodeContext/*!*/ context, int length=0) {

_checkClosed();

throw AttributeError("peek");

}

[PythonHidden]

public virtual object read(CodeContext/*!*/ context, object length=null) {

throw AttributeError("read");

}

[PythonHidden]

public virtual Bytes read1(CodeContext/*!*/ context, int length=0) {

throw AttributeError("read1");

}

public virtual bool readable(CodeContext/*!*/ context) {

return false;

}

public virtual object readline(CodeContext/*!*/ context, int limit) {

_checkClosed();

List<Bytes> res = new List<Bytes>();

int count = 0;

while (limit < 0 || res.Count < limit) {

object cur = read(context, 1);

if (cur == null) {

break;

}

Bytes curBytes = GetBytes(cur, "read()");

if (curBytes.Count == 0) {

break;

}

res.Add(curBytes);

count += curBytes.Count;

if (((IList<byte>)curBytes)[curBytes.Count - 1] == (byte)'\n') {

break;

}

}

return Bytes.Concat(res, count);

}

public object readline(CodeContext/*!*/ context, object limit=null) {

return readline(context, GetInt(limit, -1));

}

public virtual PythonList readlines() {

return readlines(null);

}

public virtual PythonList readlines(object hint=null) {

int size = GetInt(hint, -1);

PythonList res = new PythonList();

if (size <= 0) {

foreach (object line in this) {

res.AddNoLock(line);

}

return res;

}

int count = 0;

foreach (object line in this) {

if (line is Bytes bytes) {

res.AddNoLock(line);

count += bytes.Count;

if (count >= size) {

break;

}

continue;

}

if (line is string str) {

res.AddNoLock(line);

count += str.Length;

if (count >= size) {

break;

}

continue;

}

throw PythonOps.TypeError("next() should return string or bytes");

}

return res;

}

public virtual BigInteger seek(CodeContext/*!*/ context, BigInteger pos, [Optional]object whence) {

throw UnsupportedOperation(context, "seek");

}

public virtual bool seekable(CodeContext/*!*/ context) {

return false;

}

public virtual BigInteger tell(CodeContext/*!*/ context) {

return seek(context, 0, 1);

}

public virtual BigInteger truncate(CodeContext/*!*/ context, object pos=null) {

throw UnsupportedOperation(context, "truncate");

}

public virtual bool writable(CodeContext/*!*/ context) {

return false;

}

[PythonHidden]

public virtual BigInteger write(CodeContext/*!*/ context, object buf) {

throw AttributeError("write");

}

public virtual void writelines(CodeContext/*!*/ context, object lines) {

_checkClosed();

IEnumerator en = PythonOps.GetEnumerator(context, lines);

while (en.MoveNext()) {

write(context, en.Current);

}

}

#endregion

~_IOBase() {

try {

close(context);

} catch { }

}

#region IEnumerator<object> Members

private object _current;

object IEnumerator<object>.Current {

get { return _current; }

}

#endregion

#region IEnumerator Members

object IEnumerator.Current {

get { return _current; }

}

bool IEnumerator.MoveNext() {

_current = readline(context, -1);

if (_current == null) {

return false;

}

if (_current is Bytes bytes) {

return bytes.Count > 0;

}

if (_current is string str) {

return str.Length > 0;

}

return PythonOps.IsTrue(_current);

}

void IEnumerator.Reset() {

_current = null;

seek(context, 0, 0);

}

#endregion

#region IEnumerable<object> Members

[PythonHidden]

public IEnumerator<object> GetEnumerator() {

_checkClosed();

return this;

}

#endregion

#region IEnumerable Members

IEnumerator IEnumerable.GetEnumerator() {

_checkClosed();

return this;

}

#endregion

#region IDisposable Members

void IDisposable.Dispose() { }

#endregion

#region IWeakReferenceable Members

private WeakRefTracker _weakref;

WeakRefTracker IWeakReferenceable.GetWeakRef() {

return _weakref;

}

bool IWeakReferenceable.SetWeakRef(WeakRefTracker value) {

_weakref = value;

return true;

}

void IWeakReferenceable.SetFinalizer(WeakRefTracker value) {

((IWeakReferenceable)this).SetWeakRef(value);

}

#endregion

#region IPythonExpandable Members

private IDictionary<string, object> _customAttributes;

IDictionary<string, object> IPythonExpandable.EnsureCustomAttributes() {

return _customAttributes = new Dictionary<string, object>();

}

IDictionary<string, object> IPythonExpandable.CustomAttributes {

get { return _customAttributes; }

}

CodeContext/*!*/ IPythonExpandable.Context {

get { return context; }

}

#endregion

#region IDynamicMetaObjectProvider Members

DynamicMetaObject IDynamicMetaObjectProvider.GetMetaObject(Expression parameter) {

return new MetaExpandable<_IOBase>(parameter, this);

}

#endregion

#region Private implementation details

internal Exception UnsupportedOperation(CodeContext/*!*/ context, string attr) {

throw PythonExceptions.CreateThrowable(

(PythonType)context.LanguageContext.GetModuleState(_unsupportedOperationKey),

string.Format("{0}.{1} not supported", PythonTypeOps.GetName(this), attr)

);

}

internal Exception AttributeError(string attrName) {

throw PythonOps.AttributeError("'{0}' object has no attribute '{1}'", PythonTypeOps.GetName(this), attrName);

}

internal Exception InvalidPosition(BigInteger pos) {

return PythonOps.IOError("Raw stream returned invalid position {0}", pos);

}

#endregion

}

[PythonType]

public class _RawIOBase : _IOBase, IDynamicMetaObjectProvider {

public _RawIOBase(CodeContext/*!*/ context) : base(context) { }

#region Public API

public override object read(CodeContext/*!*/ context, object size=null) {

int sizeInt = GetInt(size, -1);

if (sizeInt < 0) {

return readall(context);

}

ByteArray arr = new ByteArray(new byte[sizeInt]);

sizeInt = (int)readinto(context, arr);

List<byte> res = arr.UnsafeByteList;

if (sizeInt < res.Count) {

res.RemoveRange(sizeInt, res.Count - sizeInt);

}

return new Bytes(res);

}

public Bytes readall(CodeContext/*!*/ context) {

List<Bytes> res = new List<Bytes>();

int count = 0;

for (; ; ) {

object cur = read(context, DEFAULT_BUFFER_SIZE);

if (cur == null) {

break;

}

Bytes curBytes = GetBytes(cur, "read()");

if (curBytes.Count == 0) {

break;

}

count += curBytes.Count;

res.Add(curBytes);

}

return Bytes.Concat(res, count);

}

public virtual BigInteger readinto(CodeContext/*!*/ context, object buf) {

throw UnsupportedOperation(context, "readinto");

}

public override BigInteger write(CodeContext/*!*/ context, object buf) {

throw UnsupportedOperation(context, "write");

}

#endregion

#region IDynamicMetaObjectProvider Members

DynamicMetaObject IDynamicMetaObjectProvider.GetMetaObject(Expression parameter) {

return new MetaExpandable<_RawIOBase>(parameter, this);

}

#endregion

}

[PythonType]

public class _BufferedIOBase : _IOBase, IDynamicMetaObjectProvider {

public _BufferedIOBase(CodeContext/*!*/ context) : base(context) { }

#region Public API

public virtual object detach(CodeContext/*!*/ context) {

throw UnsupportedOperation(context, "detach");

}

public override object read(CodeContext/*!*/ context, object length=null) {

throw UnsupportedOperation(context, "read");

}

public virtual BigInteger readinto(CodeContext/*!*/ context, object buf) {

int length = -1;

if (PythonOps.HasAttr(context, buf, "__len__")) {

length = PythonOps.Length(buf);

}

object dataObj = read(context, length);

if (dataObj == null) {

return BigInteger.Zero;

}

Bytes data = GetBytes(dataObj, "read()");

if (buf is IList<byte> bytes) {

for (int i = 0; i < data.Count; i++) {

bytes[i] = ((IList<byte>)data)[i];

}

GC.KeepAlive(this);

return data.Count;

}

object setter;

if (PythonOps.TryGetBoundAttr(buf, "__setitem__", out setter)) {

for (int i = 0; i < data.Count; i++) {

PythonOps.CallWithContext(context, setter, i, data[i]);

}

GC.KeepAlive(this);

return data.Count;

}

throw PythonOps.TypeError("must be read-write buffer, not " + PythonTypeOps.GetName(buf));

}

public override BigInteger write(CodeContext/*!*/ context, object buf) {

throw UnsupportedOperation(context, "write");

}

#endregion

#region IDynamicMetaObjectProvider Members

DynamicMetaObject IDynamicMetaObjectProvider.GetMetaObject(Expression parameter) {

return new MetaExpandable<_BufferedIOBase>(parameter, this);

}

#endregion

}

[PythonType]

public class _TextIOBase : _IOBase, IDynamicMetaObjectProvider {

public _TextIOBase(CodeContext/*!*/ context) : base(context) { }

#region Public API

public virtual object detach(CodeContext/*!*/ context) {

throw UnsupportedOperation(context, "detach");

}

public virtual string encoding {

get { return null; }

}

public virtual string errors {

get { return null; }

}

public virtual object newlines {

get { return null; }

}

public override object read(CodeContext/*!*/ context, [DefaultParameterValue(-1)]object length) {

throw UnsupportedOperation(context, "read");

}

public override object readline(CodeContext/*!*/ context, int limit=-1) {

throw UnsupportedOperation(context, "readline");

}

public override BigInteger truncate(CodeContext/*!*/ context, object pos=null) {

throw UnsupportedOperation(context, "truncate");

}

public override BigInteger write(CodeContext/*!*/ context, object str) {

throw UnsupportedOperation(context, "write");

}

#endregion

#region IDynamicMetaObjectProvider Members

DynamicMetaObject IDynamicMetaObjectProvider.GetMetaObject(Expression parameter) {

return new MetaExpandable<_TextIOBase>(parameter, this);

}

#endregion

}

[PythonType]

public class BufferedReader : _BufferedIOBase, IDynamicMetaObjectProvider {

private _IOBase _rawIO;

private object _raw;

private int _bufSize;

private Bytes _readBuf;

private int _readBufPos;

internal static BufferedReader Create(CodeContext/*!*/ context, object raw, int buffer_size=DEFAULT_BUFFER_SIZE) {

var res = new BufferedReader(context, raw, buffer_size);

res.__init__(context, raw, buffer_size);

return res;

}

public BufferedReader(

CodeContext/*!*/ context,

object raw,

int buffer_size=DEFAULT_BUFFER_SIZE

) : base(context) {

}

public void __init__(

CodeContext/*!*/ context,

object raw,

int buffer_size=DEFAULT_BUFFER_SIZE

) {

this.raw = raw;

if (_rawIO != null) {

if (!_rawIO.readable(context)) {

throw PythonOps.IOError("\"raw\" argument must be readable.");

}

} else {

if (PythonOps.Not(PythonOps.Invoke(context, _raw, "readable"))) {

throw PythonOps.IOError("\"raw\" argument must be readable.");

}

}

if (buffer_size <= 0) {

throw PythonOps.ValueError("invalid buffer size (must be positive)");

}

_bufSize = buffer_size;

_readBuf = Bytes.Empty;

}

#region Public API

public object raw {

get {

return _raw;

}

set {

_rawIO = value as _IOBase;

_raw = value;

}

}

#region _BufferedIOMixin

public override BigInteger truncate(CodeContext/*!*/ context, object pos=null) {

if (_rawIO != null) {

return _rawIO.truncate(context, pos);

}

return GetBigInt(

PythonOps.Invoke(context, _raw, "truncate", pos),

"truncate() should return integer"

);

}

public override void close(CodeContext/*!*/ context) {

if (!closed) {

try {

flush(context);

} finally {

if (_rawIO != null) {

_rawIO.close(context);

} else {

PythonOps.Invoke(context, _raw, "close");

}

}

}

}

public override object detach(CodeContext/*!*/ context) {

if (_raw == null) {

throw PythonOps.ValueError("raw stream already detached");

}

flush(context);

object res = _raw;

raw = null;

return res;

}

public override bool seekable(CodeContext/*!*/ context) {

if (_rawIO != null) {

return _rawIO.seekable(context);

}

return PythonOps.IsTrue(PythonOps.Invoke(context, _raw, "seekable"));

}

public override bool readable(CodeContext/*!*/ context) {

if (_rawIO != null) {

return _rawIO.readable(context);

}

return PythonOps.IsTrue(PythonOps.Invoke(context, _raw, "readable"));

}

public override bool writable(CodeContext/*!*/ context) {

if (_rawIO != null) {

return _rawIO.writable(context);

}

return PythonOps.IsTrue(PythonOps.Invoke(context, _raw, "writable"));

}

public override bool closed {

get {

if (_rawIO != null) {

return _rawIO.closed;

}

return PythonOps.IsTrue(PythonOps.GetBoundAttr(context, _raw, "closed"));

}

}

public object name {

get {

return PythonOps.GetBoundAttr(context, _raw, "name");

}

}

public object mode {

get {

return PythonOps.GetBoundAttr(context, _raw, "mode");

}

}

public override int fileno(CodeContext/*!*/ context) {

if (_rawIO != null) {

return _rawIO.fileno(context);

}

return GetInt(

PythonOps.Invoke(context, _raw, "fileno"),

"fileno() should return integer"

);

}

public override bool isatty(CodeContext/*!*/ context) {

if (_rawIO != null) {

return _rawIO.isatty(context);

}

return PythonOps.IsTrue(PythonOps.Invoke(context, _raw, "isatty"));

}

#endregion

public override object read(CodeContext/*!*/ context, object length=null) {

int len = GetInt(length, -1);

if (len < -1) {

throw PythonOps.ValueError("invalid number of bytes to read");

}

lock (this) {

return ReadNoLock(context, len);

}

}

private Bytes ReadNoLock(CodeContext/*!*/ context, int length, bool read1 = false) {

if (length == 0) {

return Bytes.Empty;

}

if (length < 0) {

List<Bytes> chunks = new List<Bytes>();

int count = 0;

if (_readBuf.Count > 0) {

chunks.Add(ResetReadBuf());

count += chunks[0].Count;

}

for (; ; ) {

object chunkObj;

if (_rawIO != null) {

chunkObj = _rawIO.read(context, -1);

} else {

chunkObj = PythonOps.Invoke(context, _raw, "read", -1);

}

Bytes chunk = GetBytes(chunkObj, "read()");

if (chunk == null || chunk.Count == 0) {

if (count == 0) {

return chunk;

}

break;

}

chunks.Add(chunk);

count += chunk.Count;

if (read1) break;

}

GC.KeepAlive(this);

return Bytes.Concat(chunks, count);

}

if (length <= _readBuf.Count - _readBufPos) {

// requested data is already buffered

byte[] res = new byte[length];

Array.Copy(_readBuf.UnsafeByteArray, _readBufPos, res, 0, length);

_readBufPos += length;

if (_readBufPos == _readBuf.Count) {

ResetReadBuf();

}

GC.KeepAlive(this);

return Bytes.Make(res);

} else {

// a read is required to provide requested amount of data

List<Bytes> chunks = new List<Bytes>();

int remaining = length;

if (_readBuf.Count > 0) {

chunks.Add(ResetReadBuf());

remaining -= chunks[0].Count;

}

while (remaining > 0) {

object chunkObj;

if (_rawIO != null) {

chunkObj = _rawIO.read(context, _bufSize);

} else {

chunkObj = PythonOps.Invoke(context, _raw, "read", _bufSize);

}

Bytes chunk = chunkObj != null ? GetBytes(chunkObj, "read()") : Bytes.Empty;

_readBuf = chunk;

if (_readBuf.Count == 0) {

break;

}

if (remaining >= _readBuf.Count - _readBufPos) {

remaining -= _readBuf.Count - _readBufPos;

chunks.Add(ResetReadBuf());

} else {

byte[] bytes = new byte[remaining];

Array.Copy(_readBuf.UnsafeByteArray, 0, bytes, 0, remaining);

chunks.Add(Bytes.Make(bytes));

_readBufPos = remaining;

remaining = 0;

break;

}

if (read1) break;

}

GC.KeepAlive(this);

return Bytes.Concat(chunks, length - remaining);

}

}

public override Bytes peek(CodeContext/*!*/ context, int length=0) {

_checkClosed();

if (length <= 0 || length > _bufSize) {

length = _bufSize;

}

lock (this) {

return PeekNoLock(context, length);

}

}

private Bytes PeekNoLock(CodeContext/*!*/ context, int length) {

int bufLen = _readBuf.Count - _readBufPos;

byte[] bytes = new byte[length];

if (length <= bufLen) {

Array.Copy(_readBuf.UnsafeByteArray, _readBufPos, bytes, 0, length);

return Bytes.Make(bytes);

}

object nextObj;

if (_rawIO != null) {

nextObj = _rawIO.read(context, length - _readBuf.Count + _readBufPos);

} else {

nextObj = PythonOps.Invoke(context, _raw, "read", length - _readBuf.Count + _readBufPos);

}

Bytes next = nextObj != null ? GetBytes(nextObj, "read()") : Bytes.Empty;