From 877e4acf7d8f397b64ac76cdb1504fa0bd9fda43 Mon Sep 17 00:00:00 2001
From: Nachtalb <info@nachtalb.io>
Date: Thu, 26 Nov 2020 00:37:46 +0100
Subject: [PATCH] Add network specific leave message

---
 client/components/NetworkForm.vue | 10 ++++++++++
 defaults/config.js                |  2 ++
 src/client.js                     |  1 +
 src/models/network.js             |  5 +++++
 src/plugins/inputs/part.js        |  2 +-
 test/models/network.js            |  1 +
 6 files changed, 20 insertions(+), 1 deletion(-)

diff --git a/client/components/NetworkForm.vue b/client/components/NetworkForm.vue
index ee9ff0d3..a2ea9282 100644
--- a/client/components/NetworkForm.vue
+++ b/client/components/NetworkForm.vue
@@ -167,6 +167,16 @@
 					maxlength="300"
 				/>
 			</div>
+			<div class="connect-row">
+				<label for="connect:leaveMessage">Away message</label>
+				<input
+					id="connect:leaveMessage"
+					v-model="defaults.leaveMessage"
+					class="input"
+					name="leaveMessage"
+					placeholder="Smell ya later"
+				/>
+			</div>
 			<template v-if="defaults.uuid && !$store.state.serverConfiguration.public">
 				<div class="connect-row">
 					<label for="connect:commands">
diff --git a/defaults/config.js b/defaults/config.js
index 264d1f5d..9f8a5493 100644
--- a/defaults/config.js
+++ b/defaults/config.js
@@ -218,6 +218,7 @@ module.exports = {
 	//   numbers from 0 to 9. For example, `Guest%%%` may become `Guest123`.
 	// - `username`: User name.
 	// - `realname`: Real name.
+	// - `leaveMessage`: Network specific leave message (overrides global leaveMessage)
 	// - `join`: Comma-separated list of channels to auto-join once connected.
 	//
 	// This value is set to connect to the official channel of The Lounge on
@@ -248,6 +249,7 @@ module.exports = {
 		username: "thelounge",
 		realname: "The Lounge User",
 		join: "#thelounge",
+		leaveMessage: "",
 	},
 
 	// ### `lockNetwork`
diff --git a/src/client.js b/src/client.js
index 6f9fd023..3f28b97e 100644
--- a/src/client.js
+++ b/src/client.js
@@ -241,6 +241,7 @@ Client.prototype.connect = function (args, isStartup = false) {
 		nick: String(args.nick || ""),
 		username: String(args.username || ""),
 		realname: String(args.realname || ""),
+		leaveMessage: String(args.leaveMessage || ""),
 		sasl: String(args.sasl || ""),
 		saslAccount: String(args.saslAccount || ""),
 		saslPassword: String(args.saslPassword || ""),
diff --git a/src/models/network.js b/src/models/network.js
index e137a137..97dea399 100644
--- a/src/models/network.js
+++ b/src/models/network.js
@@ -35,6 +35,7 @@ function Network(attr) {
 		commands: [],
 		username: "",
 		realname: "",
+		leaveMessage: "",
 		sasl: "",
 		saslAccount: "",
 		saslPassword: "",
@@ -82,6 +83,7 @@ Network.prototype.validate = function (client) {
 
 	this.username = cleanString(this.username) || "thelounge";
 	this.realname = cleanString(this.realname) || "The Lounge User";
+	this.leaveMessage = cleanString(this.leaveMessage);
 	this.password = cleanString(this.password);
 	this.host = cleanString(this.host).toLowerCase();
 	this.name = cleanString(this.name);
@@ -267,6 +269,7 @@ Network.prototype.edit = function (client, args) {
 	this.password = String(args.password || "");
 	this.username = String(args.username || "");
 	this.realname = String(args.realname || "");
+	this.leaveMessage = String(args.leaveMessage || "");
 	this.sasl = String(args.sasl || "");
 	this.saslAccount = String(args.saslAccount || "");
 	this.saslPassword = String(args.saslPassword || "");
@@ -447,6 +450,7 @@ Network.prototype.exportForEdit = function () {
 		"password",
 		"username",
 		"realname",
+		"leaveMessage",
 		"sasl",
 		"saslAccount",
 		"saslPassword",
@@ -481,6 +485,7 @@ Network.prototype.export = function () {
 		"password",
 		"username",
 		"realname",
+		"leaveMessage",
 		"sasl",
 		"saslAccount",
 		"saslPassword",
diff --git a/src/plugins/inputs/part.js b/src/plugins/inputs/part.js
index f049d5b8..d4ba095a 100644
--- a/src/plugins/inputs/part.js
+++ b/src/plugins/inputs/part.js
@@ -48,7 +48,7 @@ exports.input = function (network, chan, cmd, args) {
 		});
 		this.save();
 	} else {
-		const partMessage = args.join(" ") || Helper.config.leaveMessage;
+		const partMessage = args.join(" ") || network.leaveMessage || Helper.config.leaveMessage;
 		network.irc.part(target.name, partMessage);
 	}
 
diff --git a/test/models/network.js b/test/models/network.js
index 3308bbc6..6d6d8119 100644
--- a/test/models/network.js
+++ b/test/models/network.js
@@ -40,6 +40,7 @@ describe("Network", function () {
 				password: "",
 				username: "",
 				realname: "",
+				leaveMessage: "",
 				sasl: "plain",
 				saslAccount: "testaccount",
 				saslPassword: "testpassword",