add directory state to console (#416)

This commit is contained in:
Saeed Vaziry
2025-01-01 14:52:27 -08:00
committed by GitHub
parent ba069a2db0
commit f5c9d6701b
7 changed files with 149 additions and 50 deletions

View File

@ -4,14 +4,16 @@
use App\Models\Server;
use Illuminate\Http\Request;
use Illuminate\Support\Facades\Cache;
use Illuminate\Validation\Rule;
use Spatie\RouteAttributes\Attributes\Get;
use Spatie\RouteAttributes\Attributes\Middleware;
use Spatie\RouteAttributes\Attributes\Post;
#[Middleware('auth')]
class ConsoleController extends Controller
{
#[Post('/{server}/console', name: 'servers.console.run')]
#[Post('servers/{server}/console/run', name: 'servers.console.run')]
public function run(Server $server, Request $request)
{
$this->authorize('update', $server);
@ -24,15 +26,29 @@ public function run(Server $server, Request $request)
'command' => 'required|string',
]);
$ssh = $server->ssh($request->user);
$log = 'console-'.time();
$user = $request->input('user');
$currentDir = $user == 'root' ? '/root' : '/home/'.$user;
if (Cache::has('console.'.$server->id.'.dir')) {
$currentDir = Cache::get('console.'.$server->id.'.dir');
}
return response()->stream(
function () use ($server, $request) {
$ssh = $server->ssh($request->user);
$log = 'console-'.time();
$ssh->exec(command: $request->command, log: $log, stream: true, streamCallback: function ($output) {
echo $output;
function () use ($server, $request, $ssh, $log, $currentDir) {
$command = 'cd '.$currentDir.' && '.$request->command.' && echo "VITO_WORKING_DIR: $(pwd)"';
$output = '';
$ssh->exec(command: $command, log: $log, stream: true, streamCallback: function ($out) use (&$output) {
echo preg_replace('/^VITO_WORKING_DIR:.*(\r?\n)?/m', '', $out);
$output .= $out;
ob_flush();
flush();
});
// extract the working dir and put it in the session
if (preg_match('/VITO_WORKING_DIR: (.*)/', $output, $matches)) {
Cache::put('console.'.$server->id.'.dir', $matches[1]);
}
},
200,
[
@ -42,4 +58,12 @@ function () use ($server, $request) {
]
);
}
#[Get('servers/{server}/console/working-dir', name: 'servers.console.working-dir')]
public function workingDir(Server $server)
{
return response()->json([
'dir' => Cache::get('console.'.$server->id.'.dir'),
]);
}
}