X-Git-Url: https://git.xinqibao.xyz/slstatus.git/blobdiff_plain/61e44e894890c1521a01148fbf969cbd4dbb4cae..7246dc4381c6c95454672a5c1aff65a02d6d3747:/components/num_files.c?ds=inline

diff --git a/components/num_files.c b/components/num_files.c
new file mode 100644
index 0000000..a8a3894
--- /dev/null
+++ b/components/num_files.c
@@ -0,0 +1,30 @@
+/* See LICENSE file for copyright and license details. */
+#include <dirent.h>
+#include <err.h>
+#include <stdio.h>
+#include <string.h>
+
+#include "../util.h"
+
+const char *
+num_files(const char *dir)
+{
+	struct dirent *dp;
+	DIR *fd;
+	int num = 0;
+
+	if ((fd = opendir(dir)) == NULL) {
+		warn("Failed to get number of files in directory %s", dir);
+		return NULL;
+	}
+
+	while ((dp = readdir(fd)) != NULL) {
+		if (!strcmp(dp->d_name, ".") || !strcmp(dp->d_name, ".."))
+			continue; /* skip self and parent */
+		num++;
+	}
+
+	closedir(fd);
+
+	return bprintf("%d", num);
+}