# Conflicts: # package-lock.json # package.json # src/util-frontend.jspull/2632/head
Louis Lam
2 years ago
commit
c47b6c5995
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,113 @@
|
|||||||
|
const NotificationProvider = require("./notification-provider");
|
||||||
|
const axios = require("axios");
|
||||||
|
const { UP, DOWN, getMonitorRelativeURL } = require("../../src/util");
|
||||||
|
const { setting } = require("../util-server");
|
||||||
|
let successMessage = "Sent Successfully.";
|
||||||
|
|
||||||
|
class Splunk extends NotificationProvider {
|
||||||
|
name = "Splunk";
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @inheritdoc
|
||||||
|
*/
|
||||||
|
async send(notification, msg, monitorJSON = null, heartbeatJSON = null) {
|
||||||
|
try {
|
||||||
|
if (heartbeatJSON == null) {
|
||||||
|
const title = "Uptime Kuma Alert";
|
||||||
|
const monitor = {
|
||||||
|
type: "ping",
|
||||||
|
url: "Uptime Kuma Test Button",
|
||||||
|
};
|
||||||
|
return this.postNotification(notification, title, msg, monitor, "trigger");
|
||||||
|
}
|
||||||
|
|
||||||
|
if (heartbeatJSON.status === UP) {
|
||||||
|
const title = "Uptime Kuma Monitor ✅ Up";
|
||||||
|
return this.postNotification(notification, title, heartbeatJSON.msg, monitorJSON, "recovery");
|
||||||
|
}
|
||||||
|
|
||||||
|
if (heartbeatJSON.status === DOWN) {
|
||||||
|
const title = "Uptime Kuma Monitor 🔴 Down";
|
||||||
|
return this.postNotification(notification, title, heartbeatJSON.msg, monitorJSON, "trigger");
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
this.throwGeneralAxiosError(error);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Check if result is successful, result code should be in range 2xx
|
||||||
|
* @param {Object} result Axios response object
|
||||||
|
* @throws {Error} The status code is not in range 2xx
|
||||||
|
*/
|
||||||
|
checkResult(result) {
|
||||||
|
if (result.status == null) {
|
||||||
|
throw new Error("Splunk notification failed with invalid response!");
|
||||||
|
}
|
||||||
|
if (result.status < 200 || result.status >= 300) {
|
||||||
|
throw new Error("Splunk notification failed with status code " + result.status);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Send the message
|
||||||
|
* @param {BeanModel} notification Message title
|
||||||
|
* @param {string} title Message title
|
||||||
|
* @param {string} body Message
|
||||||
|
* @param {Object} monitorInfo Monitor details (For Up/Down only)
|
||||||
|
* @param {?string} eventAction Action event for PagerDuty (trigger, acknowledge, resolve)
|
||||||
|
* @returns {string}
|
||||||
|
*/
|
||||||
|
async postNotification(notification, title, body, monitorInfo, eventAction = "trigger") {
|
||||||
|
|
||||||
|
let monitorUrl;
|
||||||
|
if (monitorInfo.type === "port") {
|
||||||
|
monitorUrl = monitorInfo.hostname;
|
||||||
|
if (monitorInfo.port) {
|
||||||
|
monitorUrl += ":" + monitorInfo.port;
|
||||||
|
}
|
||||||
|
} else if (monitorInfo.hostname != null) {
|
||||||
|
monitorUrl = monitorInfo.hostname;
|
||||||
|
} else {
|
||||||
|
monitorUrl = monitorInfo.url;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (eventAction === "recovery") {
|
||||||
|
if (notification.splunkAutoResolve === "0") {
|
||||||
|
return "No action required";
|
||||||
|
}
|
||||||
|
eventAction = notification.splunkAutoResolve;
|
||||||
|
} else {
|
||||||
|
eventAction = notification.splunkSeverity;
|
||||||
|
}
|
||||||
|
|
||||||
|
const options = {
|
||||||
|
method: "POST",
|
||||||
|
url: notification.splunkRestURL,
|
||||||
|
headers: { "Content-Type": "application/json" },
|
||||||
|
data: {
|
||||||
|
message_type: eventAction,
|
||||||
|
state_message: `[${title}] [${monitorUrl}] ${body}`,
|
||||||
|
entity_display_name: "Uptime Kuma Alert: " + monitorInfo.name,
|
||||||
|
routing_key: notification.pagerdutyIntegrationKey,
|
||||||
|
entity_id: "Uptime Kuma/" + monitorInfo.id,
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
const baseURL = await setting("primaryBaseURL");
|
||||||
|
if (baseURL && monitorInfo) {
|
||||||
|
options.client = "Uptime Kuma";
|
||||||
|
options.client_url = baseURL + getMonitorRelativeURL(monitorInfo.id);
|
||||||
|
}
|
||||||
|
|
||||||
|
let result = await axios.request(options);
|
||||||
|
this.checkResult(result);
|
||||||
|
if (result.statusText != null) {
|
||||||
|
return "Splunk notification succeed: " + result.statusText;
|
||||||
|
}
|
||||||
|
|
||||||
|
return successMessage;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
module.exports = Splunk;
|
||||||
@ -0,0 +1,32 @@
|
|||||||
|
<template>
|
||||||
|
<div class="mb-3">
|
||||||
|
<label for="splunk-rest-url" class="form-label">{{ $t("Splunk Rest URL") }}</label>
|
||||||
|
<HiddenInput id="splunk-rest-url" v-model="$parent.notification.splunkRestURL" :required="true" autocomplete="false"></HiddenInput>
|
||||||
|
</div>
|
||||||
|
<div class="mb-3">
|
||||||
|
<label for="splunk-severity" class="form-label">{{ $t("Severity") }}</label>
|
||||||
|
<select id="splunk-severity" v-model="$parent.notification.splunkSeverity" class="form-select">
|
||||||
|
<option value="INFO">{{ $t("info") }}</option>
|
||||||
|
<option value="WARNING">{{ $t("warning") }}</option>
|
||||||
|
<option value="CRITICAL" selected="selected">{{ $t("critical") }}</option>
|
||||||
|
</select>
|
||||||
|
</div>
|
||||||
|
<div class="mb-3">
|
||||||
|
<label for="splunk-resolve" class="form-label">{{ $t("Auto resolve or acknowledged") }}</label>
|
||||||
|
<select id="splunk-resolve" v-model="$parent.notification.splunkAutoResolve" class="form-select">
|
||||||
|
<option value="0" selected="selected">{{ $t("do nothing") }}</option>
|
||||||
|
<option value="ACKNOWLEDGEMENT">{{ $t("auto acknowledged") }}</option>
|
||||||
|
<option value="RECOVERY">{{ $t("auto resolve") }}</option>
|
||||||
|
</select>
|
||||||
|
</div>
|
||||||
|
</template>
|
||||||
|
|
||||||
|
<script>
|
||||||
|
import HiddenInput from "../HiddenInput.vue";
|
||||||
|
|
||||||
|
export default {
|
||||||
|
components: {
|
||||||
|
HiddenInput,
|
||||||
|
},
|
||||||
|
};
|
||||||
|
</script>
|
||||||
@ -0,0 +1,33 @@
|
|||||||
|
import { hostNameRegexPattern } from "../../../src/util-frontend";
|
||||||
|
|
||||||
|
describe("Test util-frontend.js", () => {
|
||||||
|
|
||||||
|
describe("hostNameRegexPattern()", () => {
|
||||||
|
it('should return a valid regex for non mqtt hostnames', () => {
|
||||||
|
const regex = new RegExp(hostNameRegexPattern(false));
|
||||||
|
|
||||||
|
expect(regex.test("www.test.com")).to.be.true;
|
||||||
|
expect(regex.test("127.0.0.1")).to.be.true;
|
||||||
|
expect(regex.test("192.168.1.156")).to.be.true;
|
||||||
|
|
||||||
|
["mqtt", "mqtts", "ws", "wss"].forEach(schema => {
|
||||||
|
expect(regex.test(`${schema}://www.test.com`)).to.be.false;
|
||||||
|
expect(regex.test(`${schema}://127.0.0.1`)).to.be.false;
|
||||||
|
});
|
||||||
|
});
|
||||||
|
it('should return a valid regex for mqtt hostnames', () => {
|
||||||
|
const hostnameString = hostNameRegexPattern(false);
|
||||||
|
console.log('*********', hostnameString, '***********');
|
||||||
|
const regex = new RegExp(hostNameRegexPattern(true));
|
||||||
|
|
||||||
|
expect(regex.test("www.test.com")).to.be.true;
|
||||||
|
expect(regex.test("127.0.0.1")).to.be.true;
|
||||||
|
expect(regex.test("192.168.1.156")).to.be.true;
|
||||||
|
|
||||||
|
["mqtt", "mqtts", "ws", "wss"].forEach(schema => {
|
||||||
|
expect(regex.test(`${schema}://www.test.com`)).to.be.true;
|
||||||
|
expect(regex.test(`${schema}://127.0.0.1`)).to.be.true;
|
||||||
|
});
|
||||||
|
});
|
||||||
|
});
|
||||||
|
});
|
||||||
Loading…
Reference in new issue