Word count using ctype.h to check for alphabetic and blank characters.

Use defines for buffer sizes.
Added missing fclose.
Return pointer to the user_t struct in user_get.

1 files changed, 36 insertions, 5 deletions

diff --git a/main.c b/main.c
index 79b1eec..02686ec 100644
--- a/main.c
+++ b/main.c
@@ -1,11 +1,16 @@
 #include <stdio.h>
 #include <string.h>
+#include <ctype.h>
 
 #include "config.h"
 #include "regexset.h"
 #include "channel.h"
 #include "user.h"
 
+#define NICK_BUFFER_SIZE 0x100
+#define TEXT_BUFFER_SIZE 0x400
+#define LINE_BUFFER_SIZE 0x400
+
 int main(int argc, char **argv) {
 	/* Regex sets must be initialized before config. */
 	rs_init();
@@ -34,25 +39,51 @@ int main(int argc, char **argv) {
 			} else
 				printf("\tParsing %s\n", file->path);
 
-			char line[0x2ff];
-			while(fgets(line, 0x2ff, f)) {
+			char line[LINE_BUFFER_SIZE];
+			while(fgets(line, LINE_BUFFER_SIZE, f)) {
 				int rc;
 				int ovector[30];
 
 				rc = pcre_exec(rs->text, NULL, line, strlen(line), 0, 0, ovector, 30);
 				if(rc > 0) {
-					char nick[0x20], text[0x200];
-					pcre_copy_named_substring(rs->text, line, ovector, rc, "nick", nick, 0x20);
-					pcre_copy_named_substring(rs->text, line, ovector, rc, "text", text, 0x200);
+					char nick[NICK_BUFFER_SIZE], text[TEXT_BUFFER_SIZE];
+					pcre_copy_named_substring(rs->text, line, ovector, rc, "nick", nick, NICK_BUFFER_SIZE);
+					pcre_copy_named_substring(rs->text, line, ovector, rc, "text", text, TEXT_BUFFER_SIZE);
 					struct user_t *user = user_get(nick);
+					user->lines++;
+
+					/* Count words. */
+					char word[TEXT_BUFFER_SIZE];
+					char *end = strchr(text, '\0');
+					*word = '\0';
+					int len = 0;
+					for(char *pos = text; pos < end; pos++) {
+						if(isblank(*pos)) {
+							user->words++;
+							word[len] = '\0';
+							len = 0;
+							*word = '\0';
+						} else if isalpha(*pos) {
+							word[len++] = *pos;
+						} else {
+							len = 0;
+							*word = '\0';
+						}
+					}
+					if(len)
+						user->words++;
 					continue;
 				}
 
 				rc = pcre_exec(rs->join, NULL, line, strlen(line), 0, 0, ovector, 30);
 				if(rc > 0) {
+					char nick[NICK_BUFFER_SIZE];
+					pcre_copy_named_substring(rs->join, line, ovector, rc, "nick", nick, NICK_BUFFER_SIZE);
+					struct user_t *user = user_get(nick);
 					continue;
 				}
 			}
+			fclose(f);
 
 			file = file->next;
 		}