feat(mcp): AttachMediaFromUploadTool attaches uploaded Media by token

This commit is contained in:
Paulo Castellano 2026-05-15 16:29:05 -03:00
parent 7341852f5c
commit dd1056a413
3 changed files with 168 additions and 0 deletions

View file

@ -12,6 +12,7 @@
use App\Mcp\Tools\Label\ListLabelsTool;
use App\Mcp\Tools\Label\UpdateLabelTool;
use App\Mcp\Tools\Platform\ListContentTypesTool;
use App\Mcp\Tools\Post\AttachMediaFromUploadTool;
use App\Mcp\Tools\Post\AttachMediaFromUrlTool;
use App\Mcp\Tools\Post\CreatePostTool;
use App\Mcp\Tools\Post\DeletePostTool;
@ -52,6 +53,7 @@ class TryPostServer extends Server
DeletePostTool::class,
AttachMediaFromUrlTool::class,
RequestMediaUploadTool::class,
AttachMediaFromUploadTool::class,
GetPostMetricsTool::class,
// Platforms (read-only metadata)

View file

@ -0,0 +1,74 @@
<?php
declare(strict_types=1);
namespace App\Mcp\Tools\Post;
use App\Http\Resources\Api\PostResource;
use App\Models\Media;
use App\Models\Post;
use App\Models\Workspace;
use Illuminate\Contracts\JsonSchema\JsonSchema;
use Laravel\Mcp\Request;
use Laravel\Mcp\Response;
use Laravel\Mcp\ResponseFactory;
use Laravel\Mcp\Server\Attributes\Description;
use Laravel\Mcp\Server\Tool;
#[Description('Attach a Media uploaded via RequestMediaUploadTool to a post. The upload_token is the value returned by RequestMediaUploadTool; the Media is resolved by that token within the current workspace, then appended to the post.')]
class AttachMediaFromUploadTool extends Tool
{
public function handle(Request $request): Response|ResponseFactory
{
$validated = $request->validate([
'post_id' => ['required', 'uuid'],
'upload_token' => ['required', 'uuid'],
]);
$workspaceId = $request->user()->current_workspace_id;
$post = Post::where('workspace_id', $workspaceId)
->find(data_get($validated, 'post_id'));
if (! $post) {
return Response::error('Post not found.');
}
$media = Media::query()
->where('upload_token', data_get($validated, 'upload_token'))
->where('mediable_type', Workspace::class)
->where('mediable_id', $workspaceId)
->first();
if (! $media) {
return Response::error('Upload not found.');
}
if (! in_array($media->type, $post->allowedMediaTypes(), true)) {
return Response::error('No enabled platform on this post accepts this media type.');
}
$post->appendMedia([[
'id' => $media->id,
'path' => $media->path,
'url' => $media->url,
'type' => $media->type,
'mime_type' => $media->mime_type,
'original_filename' => $media->original_filename,
]]);
$post->refresh()->load(['postPlatforms.socialAccount', 'labels']);
return Response::structured([
'post' => (new PostResource($post))->resolve(),
]);
}
public function schema(JsonSchema $schema): array
{
return [
'post_id' => $schema->string()->required()->description('UUID of the post to attach the uploaded media to.'),
'upload_token' => $schema->string()->required()->description('upload_token returned by RequestMediaUploadTool, after the user has POSTed the file to the upload_url.'),
];
}
}

View file

@ -0,0 +1,92 @@
<?php
declare(strict_types=1);
use App\Enums\UserWorkspace\Role;
use App\Mcp\Servers\TryPostServer;
use App\Mcp\Tools\Post\AttachMediaFromUploadTool;
use App\Models\Media;
use App\Models\Post;
use App\Models\User;
use App\Models\Workspace;
use Illuminate\Support\Str;
beforeEach(function () {
$this->user = User::factory()->create();
$this->workspace = Workspace::factory()->create(['user_id' => $this->user->id]);
$this->workspace->members()->attach($this->user->id, ['role' => Role::Member->value]);
$this->user->update(['current_workspace_id' => $this->workspace->id]);
$this->post = Post::factory()->create([
'workspace_id' => $this->workspace->id,
'user_id' => $this->user->id,
]);
$this->token = (string) Str::uuid();
$this->media = Media::factory()->create([
'mediable_type' => Workspace::class,
'mediable_id' => $this->workspace->id,
'collection' => 'assets',
'upload_token' => $this->token,
]);
});
test('attaches the uploaded Media to the post', function () {
$response = TryPostServer::actingAs($this->user)
->tool(AttachMediaFromUploadTool::class, [
'post_id' => $this->post->id,
'upload_token' => $this->token,
]);
$response->assertOk();
expect($this->post->fresh()->media)->toHaveCount(1);
});
test('rejects a token from a different workspace', function () {
$other = User::factory()->create();
$otherWs = Workspace::factory()->create(['user_id' => $other->id]);
$foreignToken = (string) Str::uuid();
Media::factory()->create([
'mediable_type' => Workspace::class,
'mediable_id' => $otherWs->id,
'collection' => 'assets',
'upload_token' => $foreignToken,
]);
$response = TryPostServer::actingAs($this->user)
->tool(AttachMediaFromUploadTool::class, [
'post_id' => $this->post->id,
'upload_token' => $foreignToken,
]);
$response->assertHasErrors();
expect($this->post->fresh()->media)->toHaveCount(0);
});
test('rejects an unknown upload_token', function () {
$response = TryPostServer::actingAs($this->user)
->tool(AttachMediaFromUploadTool::class, [
'post_id' => $this->post->id,
'upload_token' => (string) Str::uuid(),
]);
$response->assertHasErrors();
});
test('rejects a post from another workspace', function () {
$other = User::factory()->create();
$otherWs = Workspace::factory()->create(['user_id' => $other->id]);
$otherPost = Post::factory()->create([
'workspace_id' => $otherWs->id,
'user_id' => $other->id,
]);
$response = TryPostServer::actingAs($this->user)
->tool(AttachMediaFromUploadTool::class, [
'post_id' => $otherPost->id,
'upload_token' => $this->token,
]);
$response->assertHasErrors();
});