-
Notifications
You must be signed in to change notification settings - Fork 39
/
DiskUsageChecker.ts
91 lines (74 loc) · 2.23 KB
/
DiskUsageChecker.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
import { execFile } from 'child_process';
import Logger from '../Logger';
import { Emojis } from './Markup';
import NotificationClient from './clients/NotificationClient';
type DiskUsage = {
total: number;
available: number;
};
class DiskUsageChecker {
private alertSent = false;
private static rootDir = '/';
private static gigabyte = 1024 ** 2;
private static warningThreshold = 0.9;
constructor(
private logger: Logger,
private notificationClient: NotificationClient,
) {}
public checkUsage = async (): Promise<void> => {
const { available, total } = await this.getUsage();
const used = total - available;
const usedPercentage = used / total;
if (usedPercentage >= DiskUsageChecker.warningThreshold) {
if (!this.alertSent) {
const message =
`${Emojis.RotatingLight} Disk usage is **${this.formatNumber(
usedPercentage * 100,
)}%**: ` +
`**${this.formatNumber(
this.convertToGb(available),
)} GB** of **${this.formatNumber(
this.convertToGb(total),
)} GB** available ${Emojis.RotatingLight}`;
this.logger.warn(message);
await this.notificationClient.sendMessage(message);
this.alertSent = true;
}
} else {
this.alertSent = false;
}
};
private getUsage = async (): Promise<DiskUsage> => {
return new Promise((resolve, reject) => {
execFile(
'df',
['-P', '-k', DiskUsageChecker.rootDir],
(error, stdout) => {
if (error) {
reject(error);
return;
}
const lines = stdout.split('\n');
if (lines.length < 2) {
throw new Error(`unexpected df output: ${stdout}`);
}
const parts = lines[1].split(' ').filter((x) => {
return x !== '';
});
resolve({
total: Number(parts[1]),
available: Number(parts[3]),
});
},
);
});
};
private formatNumber = (toFormat: number) => {
return Number(toFormat.toFixed(2));
};
private convertToGb = (bytes: number) => {
return bytes / DiskUsageChecker.gigabyte;
};
}
export default DiskUsageChecker;
export { DiskUsage };