IntenseFinance/neovim
3
0
Fork 0
mirror of https://github.com/neovim/neovim.git synced 2025-02-25 18:55:25 -06:00
Code Issues Packages Projects Releases Wiki Activity

refactor(main): separate connection code from --remote execution code

Browse Source
This commit is contained in:
Björn Linse 2019-11-02 15:06:32 +01:00 committed by bfredl
parent ab456bc304
commit 3a12737e6c
2 changed files with 90 additions and 77 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/nvim/api/vim.c
View File

@ -1997,9 +1997,8 @@ Array nvim_get_proc_children(Integer pid, Error *err)
DLOG("fallback to vim._os_proc_children()"); DLOG("fallback to vim._os_proc_children()");
Array a = ARRAY_DICT_INIT; Array a = ARRAY_DICT_INIT;
ADD(a, INTEGER_OBJ(pid)); ADD(a, INTEGER_OBJ(pid));
String s = cstr_to_string("return vim._os_proc_children(...)"); String s = STATIC_CSTR_AS_STRING("return vim._os_proc_children(...)");
Object o = nlua_exec(s, a, err); Object o = nlua_exec(s, a, err);
api_free_string(s);
api_free_array(a); api_free_array(a);
if (o.type == kObjectTypeArray) { if (o.type == kObjectTypeArray) {
rvobj = o.data.array; rvobj = o.data.array;

164
src/nvim/main.c
View File

@ -269,7 +269,7 @@ int main(int argc, char **argv)
server_init(params.listen_addr); server_init(params.listen_addr);
if (params.remote) { if (params.remote) {
handle_remote_client(&params, params.remote, remote_request(&params, params.remote,
params.server_addr, argc, argv); params.server_addr, argc, argv);
} }
@ -807,91 +807,105 @@ static void init_locale(void)
} }
#endif #endif
/// Handle remote subcommands
static void handle_remote_client(mparm_T *params, int remote_args, static uint64_t server_connect(char *server_addr, const char **errmsg)
char *server_addr, int argc, char **argv)
{ {
Object rvobj = OBJECT_INIT; if (server_addr == NULL) {
rvobj.data.dictionary = (Dictionary)ARRAY_DICT_INIT; *errmsg = "no address specified";
rvobj.type = kObjectTypeDictionary; return 0;
CallbackReader on_data = CALLBACK_READER_INIT; }
const char *connect_error = NULL; CallbackReader on_data = CALLBACK_READER_INIT;
uint64_t rc_id = 0; const char *error = NULL;
if (server_addr != NULL) { bool is_tcp = strrchr(server_addr, ':') ? true : false;
rc_id = channel_connect(false, server_addr, true, on_data, 50, &connect_error); // connected to channel
} uint64_t chan = channel_connect(is_tcp, server_addr, true, on_data, 50, &error);
if (error) {
*errmsg = error;
return 0;
}
return chan;
}
int t_argc = remote_args; /// Handle remote subcommands
Array args = ARRAY_DICT_INIT; static void remote_request(mparm_T *params, int remote_args,
String arg_s; char *server_addr, int argc, char **argv)
for (; t_argc < argc; t_argc++) { {
arg_s = cstr_to_string(argv[t_argc]); const char *connect_error = NULL;
ADD(args, STRING_OBJ(arg_s)); uint64_t chan = server_connect(server_addr, &connect_error);
} Object rvobj = OBJECT_INIT;
Error err = ERROR_INIT; int t_argc = remote_args;
Array a = ARRAY_DICT_INIT; Array args = ARRAY_DICT_INIT;
ADD(a, INTEGER_OBJ((int)rc_id)); String arg_s;
ADD(a, CSTR_TO_OBJ(server_addr)); for (; t_argc < argc; t_argc++) {
ADD(a, CSTR_TO_OBJ(connect_error)); arg_s = cstr_to_string(argv[t_argc]);
ADD(a, ARRAY_OBJ(args)); ADD(args, STRING_OBJ(arg_s));
String s = STATIC_CSTR_AS_STRING("return vim._cs_remote(...)"); }
Object o = nlua_exec(s, a, &err);
api_free_array(a); Error err = ERROR_INIT;
if (ERROR_SET(&err)) { Array a = ARRAY_DICT_INIT;
mch_errmsg(err.msg); ADD(a, INTEGER_OBJ((int)chan));
ADD(a, CSTR_TO_OBJ(server_addr));
ADD(a, CSTR_TO_OBJ(connect_error));
ADD(a, ARRAY_OBJ(args));
String s = STATIC_CSTR_AS_STRING("return vim._cs_remote(...)");
Object o = nlua_exec(s, a, &err);
api_free_array(a);
if (ERROR_SET(&err)) {
mch_errmsg(err.msg);
mch_errmsg("\n");
os_exit(2);
}
if (o.type == kObjectTypeDictionary) {
rvobj.data.dictionary = o.data.dictionary;
} else {
mch_errmsg("vim._cs_remote returned unexpected value\n");
os_exit(2);
}
TriState should_exit = kNone;
TriState tabbed = kNone;
for (size_t i = 0; i < rvobj.data.dictionary.size ; i++) {
if (strcmp(rvobj.data.dictionary.items[i].key.data, "errmsg") == 0) {
if (rvobj.data.dictionary.items[i].value.type != kObjectTypeString) {
mch_errmsg("vim._cs_remote returned an unexpected type for 'errmsg'\n");
os_exit(2);
}
mch_errmsg(rvobj.data.dictionary.items[i].value.data.string.data);
mch_errmsg("\n"); mch_errmsg("\n");
os_exit(2); os_exit(2);
} } else if (strcmp(rvobj.data.dictionary.items[i].key.data, "tabbed") == 0) {
if (rvobj.data.dictionary.items[i].value.type != kObjectTypeBoolean) {
if (o.type == kObjectTypeDictionary) { mch_errmsg("vim._cs_remote returned an unexpected type for 'tabbed'\n");
rvobj.data.dictionary = o.data.dictionary;
} else {
mch_errmsg("vim._cs_remote returned unexpected value\n");
os_exit(2);
}
TriState should_exit = kNone;
TriState tabbed = kNone;
for (size_t i = 0; i < rvobj.data.dictionary.size ; i++) {
if (strcmp(rvobj.data.dictionary.items[i].key.data, "errmsg") == 0) {
if (rvobj.data.dictionary.items[i].value.type != kObjectTypeString) {
mch_errmsg("vim._cs_remote returned an unexpected type for 'errmsg'\n");
os_exit(2);
}
mch_errmsg(rvobj.data.dictionary.items[i].value.data.string.data);
mch_errmsg("\n");
os_exit(2); os_exit(2);
} else if (strcmp(rvobj.data.dictionary.items[i].key.data, "tabbed") == 0) {
if (rvobj.data.dictionary.items[i].value.type != kObjectTypeBoolean) {
mch_errmsg("vim._cs_remote returned an unexpected type for 'tabbed'\n");
os_exit(2);
}
tabbed = rvobj.data.dictionary.items[i].value.data.boolean ? kTrue : kFalse;
} else if (strcmp(rvobj.data.dictionary.items[i].key.data, "should_exit") == 0) {
if (rvobj.data.dictionary.items[i].value.type != kObjectTypeBoolean) {
mch_errmsg("vim._cs_remote returned an unexpected type for 'should_exit'\n");
os_exit(2);
}
should_exit = rvobj.data.dictionary.items[i].value.data.boolean ? kTrue : kFalse;
} }
tabbed = rvobj.data.dictionary.items[i].value.data.boolean ? kTrue : kFalse;
} else if (strcmp(rvobj.data.dictionary.items[i].key.data, "should_exit") == 0) {
if (rvobj.data.dictionary.items[i].value.type != kObjectTypeBoolean) {
mch_errmsg("vim._cs_remote returned an unexpected type for 'should_exit'\n");
os_exit(2);
}
should_exit = rvobj.data.dictionary.items[i].value.data.boolean ? kTrue : kFalse;
} }
if (should_exit == kNone || tabbed == kNone) { }
mch_errmsg("vim._cs_remote didn't return a value for should_exit or tabbed, bailing\n"); if (should_exit == kNone || tabbed == kNone) {
os_exit(2); mch_errmsg("vim._cs_remote didn't return a value for should_exit or tabbed, bailing\n");
} os_exit(2);
api_free_object(o); }
api_free_object(o);
if (should_exit == kTrue) { if (should_exit == kTrue) {
os_exit(0); os_exit(0);
} }
if (tabbed == kTrue) { if (tabbed == kTrue) {
params->window_count = argc - remote_args - 1; params->window_count = argc - remote_args - 1;
params->window_layout = WIN_TABS; params->window_layout = WIN_TABS;
} }
} }
/// Decides whether text (as opposed to commands) will be read from stdin. /// Decides whether text (as opposed to commands) will be read from stdin.
/// @see EDIT_STDIN /// @see EDIT_STDIN
static bool edit_stdin(bool explicit, mparm_T *parmp) static bool edit_stdin(bool explicit, mparm_T *parmp)