mirror of
https://github.com/tmate-io/tmate.git
synced 2024-11-08 01:04:06 +01:00
If no target client is specified to commands which accept one, try to guess the
current client, in a similar manner to how sessions already work: if the current session can be established and has only one client, use that; otherwise use the most recently created client.
This commit is contained in:
parent
45043ebf3e
commit
4ca2200d83
51
cmd.c
51
cmd.c
@ -110,6 +110,7 @@ const struct cmd_entry *cmd_table[] = {
|
||||
};
|
||||
|
||||
struct session *cmd_newest_session(struct sessions *);
|
||||
struct client *cmd_newest_client(void);
|
||||
struct client *cmd_lookup_client(const char *);
|
||||
struct session *cmd_lookup_session(const char *, int *);
|
||||
struct winlink *cmd_lookup_window(struct session *, const char *, int *);
|
||||
@ -369,17 +370,63 @@ cmd_newest_session(struct sessions *ss)
|
||||
return (snewest);
|
||||
}
|
||||
|
||||
/* Find the newest client. */
|
||||
struct client *
|
||||
cmd_newest_client(void)
|
||||
{
|
||||
struct client *c, *cnewest;
|
||||
struct timeval *tv = NULL;
|
||||
u_int i;
|
||||
|
||||
cnewest = NULL;
|
||||
for (i = 0; i < ARRAY_LENGTH(&clients); i++) {
|
||||
if ((c = ARRAY_ITEM(&clients, i)) == NULL)
|
||||
continue;
|
||||
if (c->session == NULL)
|
||||
continue;
|
||||
|
||||
if (tv == NULL || timercmp(&c->tv, tv, >)) {
|
||||
cnewest = c;
|
||||
tv = &c->tv;
|
||||
}
|
||||
}
|
||||
|
||||
return (cnewest);
|
||||
}
|
||||
|
||||
/* Find the target client or report an error and return NULL. */
|
||||
struct client *
|
||||
cmd_find_client(struct cmd_ctx *ctx, const char *arg)
|
||||
{
|
||||
struct client *c;
|
||||
struct session *s;
|
||||
char *tmparg;
|
||||
size_t arglen;
|
||||
u_int i;
|
||||
|
||||
/* A NULL argument means the current client. */
|
||||
if (arg == NULL)
|
||||
return (ctx->curclient);
|
||||
if (arg == NULL) {
|
||||
if (ctx->curclient != NULL)
|
||||
return (ctx->curclient);
|
||||
/*
|
||||
* No current client set. Find the current session and see if
|
||||
* it has only has one client.
|
||||
*/
|
||||
s = cmd_current_session(ctx);
|
||||
if (s != NULL) {
|
||||
c = NULL;
|
||||
for (i = 0; i < ARRAY_LENGTH(&clients); i++) {
|
||||
if (ARRAY_ITEM(&clients, i)->session == s) {
|
||||
if (c != NULL)
|
||||
break;
|
||||
c = ARRAY_ITEM(&clients, i);
|
||||
}
|
||||
}
|
||||
if (i == ARRAY_LENGTH(&clients) && c != NULL)
|
||||
return (c);
|
||||
}
|
||||
return (cmd_newest_client());
|
||||
}
|
||||
tmparg = xstrdup(arg);
|
||||
|
||||
/* Trim a single trailing colon if any. */
|
||||
|
3
server.c
3
server.c
@ -90,6 +90,9 @@ server_create_client(int fd)
|
||||
c = xcalloc(1, sizeof *c);
|
||||
c->references = 0;
|
||||
imsg_init(&c->ibuf, fd);
|
||||
|
||||
if (gettimeofday(&c->tv, NULL) != 0)
|
||||
fatal("gettimeofday failed");
|
||||
|
||||
ARRAY_INIT(&c->prompt_hdata);
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user