File: foundry-cli-builtin-mcp.c

package info (click to toggle)
foundry 1.1~beta-2
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 15,552 kB
  • sloc: ansic: 167,487; xml: 417; makefile: 21; sh: 19; javascript: 10
file content (124 lines) | stat: -rw-r--r-- 4,167 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
/* foundry-cli-builtin-mcp.c
 *
 * Copyright 2025 Christian Hergert <chergert@redhat.com>
 *
 * This library is free software; you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation; either version 2.1 of the
 * License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 * SPDX-License-Identifier: LGPL-2.1-or-later
 */

#include "config.h"

#include <glib/gi18n-lib.h>
#include <glib-unix.h>

#include <gio/gunixinputstream.h>
#include <gio/gunixoutputstream.h>

#include "foundry-cli-builtin-private.h"
#include "foundry-cli-command-tree.h"
#include "foundry-command-line.h"
#include "foundry-context.h"
#include "foundry-mcp-server.h"
#include "foundry-util-private.h"

static int
foundry_cli_builtin_mcp_run (FoundryCommandLine *command_line,
                             const char * const *argv,
                             FoundryCliOptions  *options,
                             DexCancellable     *cancellable)
{
  g_autoptr(FoundryContext) context = NULL;
  g_autoptr(FoundryMcpServer) server = NULL;
  g_autoptr(GIOStream) stream = NULL;
  g_autoptr(GInputStream) input = NULL;
  g_autoptr(GOutputStream) output = NULL;
  g_autoptr(GError) error = NULL;
  int stdin_fd;
  int stdout_fd;

  g_assert (FOUNDRY_IS_COMMAND_LINE (command_line));
  g_assert (argv != NULL);
  g_assert (options != NULL);
  g_assert (!cancellable || DEX_IS_CANCELLABLE (cancellable));

  stdin_fd = foundry_command_line_get_stdin (command_line);
  stdout_fd = foundry_command_line_get_stdout (command_line);

  if (stdin_fd < 0 || stdout_fd < 0)
    {
      foundry_command_line_printerr (command_line, "%s\n", _("Failed to get stdin/stdout file descriptors"));
      return EXIT_FAILURE;
    }

  if (!g_unix_set_fd_nonblocking (stdin_fd, TRUE, &error) ||
      !g_unix_set_fd_nonblocking (stdout_fd, TRUE, &error))
    {
      foundry_command_line_printerr (command_line, "%s\n", error->message);
      return EXIT_FAILURE;
    }

  input = g_unix_input_stream_new (dup (stdin_fd), TRUE);
  output = g_unix_output_stream_new (dup (stdout_fd), TRUE);
  stream = g_simple_io_stream_new (input, output);

  if (!(context = dex_await_object (foundry_cli_options_load_context (options, command_line), &error)))
    goto handle_error;

  server = foundry_mcp_server_new (context, stream);

  if (server == NULL)
    {
      foundry_command_line_printerr (command_line, "%s\n", _("Failed to create MCP server"));
      return EXIT_FAILURE;
    }

  foundry_mcp_server_start (server);

  if (cancellable != NULL)
    {
      if (!dex_await (dex_ref (cancellable), &error))
        goto handle_error;
    }

  foundry_mcp_server_stop (server);

  return EXIT_SUCCESS;

handle_error:

  if (server != NULL)
    foundry_mcp_server_stop (server);

  foundry_command_line_printerr (command_line, "%s\n", error->message);
  return EXIT_FAILURE;
}

void
foundry_cli_builtin_mcp (FoundryCliCommandTree *tree)
{
  foundry_cli_command_tree_register (tree,
                                     FOUNDRY_STRV_INIT ("foundry", "mcp"),
                                     &(FoundryCliCommand) {
                                       .options = (GOptionEntry[]) {
                                         { "help", 0, 0, G_OPTION_ARG_NONE },
                                         {0}
                                       },
                                       .run = foundry_cli_builtin_mcp_run,
                                       .prepare = NULL,
                                       .complete = NULL,
                                       .gettext_package = GETTEXT_PACKAGE,
                                       .description = N_("Run MCP server"),
                                     });
}