Dynamic hostname support

monitor/monitor.rb

@@ -13,6 +13,8 @@ StatsD.server = 'monitor:8125'
 StatsD.logger = Logger.new(STDERR)
 StatsD.mode = 'production'
 
+hostname = Socket.gethostname
+
 loop do
   server_count = 0
   client_count = 0
@@ -30,9 +32,9 @@ loop do
     end
   end
 
-  StatsD.gauge('tmate.servers', server_count)
+  StatsD.gauge("tmate.#{hostname}.servers", server_count)
-  StatsD.gauge('tmate.clients', client_count)
+  StatsD.gauge("tmate.#{hostname}.clients", client_count)
-  StatsD.gauge('tmate.unique_ips', ips.uniq.count)
+  StatsD.gauge("tmate.#{hostname}.unique_ips", ips.uniq.count)
 
   sleep 10
 end

tmate-decoder.c

@@ -1,4 +1,5 @@
 #include <ctype.h>
+#include <unistd.h>
 #include "tmate.h"
 
 char *tmate_left_status, *tmate_right_status;
@@ -97,6 +98,7 @@ static void unpack_array(struct tmate_unpacker *uk,
 
 static void tmate_header(struct tmate_unpacker *uk)
 {
+	char hostname[128];
 	int protocol = unpack_int(uk);
 
 	if (protocol != 1)
@@ -104,8 +106,11 @@ static void tmate_header(struct tmate_unpacker *uk)
 
 	tmate_debug("new master, protocol version: %d", protocol);
 
-	tmate_notify("Remote session: ssh %s@%s",
+	if (gethostname(hostname, sizeof(hostname)) < 0)
-		     tmate_session_token, TMATE_HOST);
+		tmate_fatal("cannot get hostname");
+
+	tmate_notify("Remote session: ssh %s@%s.%s",
+		     tmate_session_token, hostname, TMATE_DOMAIN);
 }
 
 extern u_int next_window_pane_id;

tmate.h

@@ -17,7 +17,7 @@ do {							\
 
 /* tmate-encoder.c */
 
-#define TMATE_HOST "a.tmate.io"
+#define TMATE_DOMAIN "tmate.io"
 
 enum tmate_client_commands {
 	TMATE_NOTIFY,