359 lines
12 KiB
Java
359 lines
12 KiB
Java
/*
|
|
* This file is part of ComputerCraft - http://www.computercraft.info
|
|
* Copyright Daniel Ratcliffe, 2011-2019. Do not distribute without permission.
|
|
* Send enquiries to dratcliffe@gmail.com
|
|
*/
|
|
|
|
package dan200.computercraft.core.apis;
|
|
|
|
import dan200.computercraft.api.lua.ILuaAPI;
|
|
import dan200.computercraft.api.lua.ILuaContext;
|
|
import dan200.computercraft.api.lua.LuaException;
|
|
import dan200.computercraft.core.apis.handles.BinaryReadableHandle;
|
|
import dan200.computercraft.core.apis.handles.BinaryWritableHandle;
|
|
import dan200.computercraft.core.apis.handles.EncodedReadableHandle;
|
|
import dan200.computercraft.core.apis.handles.EncodedWritableHandle;
|
|
import dan200.computercraft.core.filesystem.FileSystem;
|
|
import dan200.computercraft.core.filesystem.FileSystemException;
|
|
import dan200.computercraft.core.filesystem.FileSystemWrapper;
|
|
import dan200.computercraft.core.tracking.TrackingField;
|
|
|
|
import javax.annotation.Nonnull;
|
|
import java.io.BufferedReader;
|
|
import java.io.BufferedWriter;
|
|
import java.nio.channels.ReadableByteChannel;
|
|
import java.nio.channels.WritableByteChannel;
|
|
import java.util.HashMap;
|
|
import java.util.Map;
|
|
import java.util.function.Function;
|
|
|
|
import static dan200.computercraft.core.apis.ArgumentHelper.getString;
|
|
|
|
public class FSAPI implements ILuaAPI
|
|
{
|
|
private IAPIEnvironment m_env;
|
|
private FileSystem m_fileSystem;
|
|
|
|
public FSAPI( IAPIEnvironment env )
|
|
{
|
|
m_env = env;
|
|
m_fileSystem = null;
|
|
}
|
|
|
|
@Override
|
|
public String[] getNames()
|
|
{
|
|
return new String[] { "fs" };
|
|
}
|
|
|
|
@Override
|
|
public void startup()
|
|
{
|
|
m_fileSystem = m_env.getFileSystem();
|
|
}
|
|
|
|
@Override
|
|
public void shutdown()
|
|
{
|
|
m_fileSystem = null;
|
|
}
|
|
|
|
@Nonnull
|
|
@Override
|
|
public String[] getMethodNames()
|
|
{
|
|
return new String[] {
|
|
"list",
|
|
"combine",
|
|
"getName",
|
|
"getSize",
|
|
"exists",
|
|
"isDir",
|
|
"isReadOnly",
|
|
"makeDir",
|
|
"move",
|
|
"copy",
|
|
"delete",
|
|
"open",
|
|
"getDrive",
|
|
"getFreeSpace",
|
|
"find",
|
|
"getDir",
|
|
};
|
|
}
|
|
|
|
@Override
|
|
public Object[] callMethod( @Nonnull ILuaContext context, int method, @Nonnull Object[] args ) throws LuaException
|
|
{
|
|
switch( method )
|
|
{
|
|
case 0:
|
|
{
|
|
// list
|
|
String path = getString( args, 0 );
|
|
m_env.addTrackingChange( TrackingField.FS_OPS );
|
|
try
|
|
{
|
|
String[] results = m_fileSystem.list( path );
|
|
Map<Object, Object> table = new HashMap<>();
|
|
for( int i = 0; i < results.length; i++ )
|
|
{
|
|
table.put( i + 1, results[i] );
|
|
}
|
|
return new Object[] { table };
|
|
}
|
|
catch( FileSystemException e )
|
|
{
|
|
throw new LuaException( e.getMessage() );
|
|
}
|
|
}
|
|
case 1:
|
|
{
|
|
// combine
|
|
String pathA = getString( args, 0 );
|
|
String pathB = getString( args, 1 );
|
|
return new Object[] { m_fileSystem.combine( pathA, pathB ) };
|
|
}
|
|
case 2:
|
|
{
|
|
// getName
|
|
String path = getString( args, 0 );
|
|
return new Object[] { FileSystem.getName( path ) };
|
|
}
|
|
case 3:
|
|
{
|
|
// getSize
|
|
String path = getString( args, 0 );
|
|
try
|
|
{
|
|
return new Object[] { m_fileSystem.getSize( path ) };
|
|
}
|
|
catch( FileSystemException e )
|
|
{
|
|
throw new LuaException( e.getMessage() );
|
|
}
|
|
}
|
|
case 4:
|
|
{
|
|
// exists
|
|
String path = getString( args, 0 );
|
|
try
|
|
{
|
|
return new Object[] { m_fileSystem.exists( path ) };
|
|
}
|
|
catch( FileSystemException e )
|
|
{
|
|
return new Object[] { false };
|
|
}
|
|
}
|
|
case 5:
|
|
{
|
|
// isDir
|
|
String path = getString( args, 0 );
|
|
try
|
|
{
|
|
return new Object[] { m_fileSystem.isDir( path ) };
|
|
}
|
|
catch( FileSystemException e )
|
|
{
|
|
return new Object[] { false };
|
|
}
|
|
}
|
|
case 6:
|
|
{
|
|
// isReadOnly
|
|
String path = getString( args, 0 );
|
|
try
|
|
{
|
|
return new Object[] { m_fileSystem.isReadOnly( path ) };
|
|
}
|
|
catch( FileSystemException e )
|
|
{
|
|
return new Object[] { false };
|
|
}
|
|
}
|
|
case 7:
|
|
{
|
|
// makeDir
|
|
String path = getString( args, 0 );
|
|
try
|
|
{
|
|
m_env.addTrackingChange( TrackingField.FS_OPS );
|
|
m_fileSystem.makeDir( path );
|
|
return null;
|
|
}
|
|
catch( FileSystemException e )
|
|
{
|
|
throw new LuaException( e.getMessage() );
|
|
}
|
|
}
|
|
case 8:
|
|
{
|
|
// move
|
|
String path = getString( args, 0 );
|
|
String dest = getString( args, 1 );
|
|
try
|
|
{
|
|
m_env.addTrackingChange( TrackingField.FS_OPS );
|
|
m_fileSystem.move( path, dest );
|
|
return null;
|
|
}
|
|
catch( FileSystemException e )
|
|
{
|
|
throw new LuaException( e.getMessage() );
|
|
}
|
|
}
|
|
case 9:
|
|
{
|
|
// copy
|
|
String path = getString( args, 0 );
|
|
String dest = getString( args, 1 );
|
|
try
|
|
{
|
|
m_env.addTrackingChange( TrackingField.FS_OPS );
|
|
m_fileSystem.copy( path, dest );
|
|
return null;
|
|
}
|
|
catch( FileSystemException e )
|
|
{
|
|
throw new LuaException( e.getMessage() );
|
|
}
|
|
}
|
|
case 10:
|
|
{
|
|
// delete
|
|
String path = getString( args, 0 );
|
|
try
|
|
{
|
|
m_env.addTrackingChange( TrackingField.FS_OPS );
|
|
m_fileSystem.delete( path );
|
|
return null;
|
|
}
|
|
catch( FileSystemException e )
|
|
{
|
|
throw new LuaException( e.getMessage() );
|
|
}
|
|
}
|
|
case 11:
|
|
{
|
|
// open
|
|
String path = getString( args, 0 );
|
|
String mode = getString( args, 1 );
|
|
m_env.addTrackingChange( TrackingField.FS_OPS );
|
|
try
|
|
{
|
|
switch( mode )
|
|
{
|
|
case "r":
|
|
{
|
|
// Open the file for reading, then create a wrapper around the reader
|
|
FileSystemWrapper<BufferedReader> reader = m_fileSystem.openForRead( path, EncodedReadableHandle::openUtf8 );
|
|
return new Object[] { new EncodedReadableHandle( reader.get(), reader ) };
|
|
}
|
|
case "w":
|
|
{
|
|
// Open the file for writing, then create a wrapper around the writer
|
|
FileSystemWrapper<BufferedWriter> writer = m_fileSystem.openForWrite( path, false, EncodedWritableHandle::openUtf8 );
|
|
return new Object[] { new EncodedWritableHandle( writer.get(), writer ) };
|
|
}
|
|
case "a":
|
|
{
|
|
// Open the file for appending, then create a wrapper around the writer
|
|
FileSystemWrapper<BufferedWriter> writer = m_fileSystem.openForWrite( path, true, EncodedWritableHandle::openUtf8 );
|
|
return new Object[] { new EncodedWritableHandle( writer.get(), writer ) };
|
|
}
|
|
case "rb":
|
|
{
|
|
// Open the file for binary reading, then create a wrapper around the reader
|
|
FileSystemWrapper<ReadableByteChannel> reader = m_fileSystem.openForRead( path, Function.identity() );
|
|
return new Object[] { new BinaryReadableHandle( reader.get(), reader ) };
|
|
}
|
|
case "wb":
|
|
{
|
|
// Open the file for binary writing, then create a wrapper around the writer
|
|
FileSystemWrapper<WritableByteChannel> writer = m_fileSystem.openForWrite( path, false, Function.identity() );
|
|
return new Object[] { new BinaryWritableHandle( writer.get(), writer ) };
|
|
}
|
|
case "ab":
|
|
{
|
|
// Open the file for binary appending, then create a wrapper around the reader
|
|
FileSystemWrapper<WritableByteChannel> writer = m_fileSystem.openForWrite( path, true, Function.identity() );
|
|
return new Object[] { new BinaryWritableHandle( writer.get(), writer ) };
|
|
}
|
|
default:
|
|
throw new LuaException( "Unsupported mode" );
|
|
}
|
|
}
|
|
catch( FileSystemException e )
|
|
{
|
|
return new Object[] { null, e.getMessage() };
|
|
}
|
|
}
|
|
case 12:
|
|
{
|
|
// getDrive
|
|
String path = getString( args, 0 );
|
|
try
|
|
{
|
|
if( !m_fileSystem.exists( path ) )
|
|
{
|
|
return null;
|
|
}
|
|
return new Object[] { m_fileSystem.getMountLabel( path ) };
|
|
}
|
|
catch( FileSystemException e )
|
|
{
|
|
throw new LuaException( e.getMessage() );
|
|
}
|
|
}
|
|
case 13:
|
|
{
|
|
// getFreeSpace
|
|
String path = getString( args, 0 );
|
|
try
|
|
{
|
|
long freeSpace = m_fileSystem.getFreeSpace( path );
|
|
if( freeSpace >= 0 )
|
|
{
|
|
return new Object[] { freeSpace };
|
|
}
|
|
return new Object[] { "unlimited" };
|
|
}
|
|
catch( FileSystemException e )
|
|
{
|
|
throw new LuaException( e.getMessage() );
|
|
}
|
|
}
|
|
case 14:
|
|
{
|
|
// find
|
|
String path = getString( args, 0 );
|
|
try
|
|
{
|
|
m_env.addTrackingChange( TrackingField.FS_OPS );
|
|
String[] results = m_fileSystem.find( path );
|
|
Map<Object, Object> table = new HashMap<>();
|
|
for( int i = 0; i < results.length; i++ )
|
|
{
|
|
table.put( i + 1, results[i] );
|
|
}
|
|
return new Object[] { table };
|
|
}
|
|
catch( FileSystemException e )
|
|
{
|
|
throw new LuaException( e.getMessage() );
|
|
}
|
|
}
|
|
case 15:
|
|
{
|
|
// getDir
|
|
String path = getString( args, 0 );
|
|
return new Object[] { FileSystem.getDirectory( path ) };
|
|
}
|
|
default:
|
|
assert false;
|
|
return null;
|
|
}
|
|
}
|
|
}
|