Improve stats API

1 files changed, 14 insertions, 1 deletions

diff --git a/src/server/api/endpoints/stats.ts b/src/server/api/endpoints/stats.ts
index 56549848dc..2f0a8b4662 100644
--- a/src/server/api/endpoints/stats.ts
+++ b/src/server/api/endpoints/stats.ts
@@ -1,5 +1,7 @@
 import Meta from '../../../models/meta';
 import define from '../define';
+import driveChart from '../../../chart/drive';
+import federationChart from '../../../chart/federation';
 
 export const meta = {
 	requireCredential: false,
@@ -15,5 +17,16 @@ export const meta = {
 export default define(meta, () => new Promise(async (res, rej) => {
 	const meta = await Meta.findOne();
 
-	res(meta ? meta.stats : {});
+	const stats: any = meta ? meta.stats : {};
+
+	const driveStats = await driveChart.getChart('hour', 1);
+	stats.driveUsageLocal = driveStats.local.totalSize[0];
+	stats.driveUsageRemote = driveStats.remote.totalSize[0];
+
+	console.log(driveStats);
+
+	const federationStats = await federationChart.getChart('hour', 1);
+	stats.instances = federationStats.instance.total[0];
+
+	res(stats);
 }));