update readme again

.gitignore

@@ -0,0 +1,4 @@
+client.txt
+server.txt
+*.o
+ct

README.md

@@ -1,9 +1,12 @@
 # threaded network chat refactor
 
-The functions have been split into semantically appropriate files but almost nothing whatsoever
+I have split functions semantically into different files and tied all global variables together in
-has been modified.
+`public.h` so they work without modifications.
 
-It APPEARS that this version actually works. I tested it a little.
+Additional changes:
+- Fixed the dreaded bug where you have to launch the program twice to make it work
+- Adjusted the ncurses stuff a bit so it doesn't look as janky
+- Changed server/client mode (`mode` variable) from an integer to an enum (still works as an integer)
 
 ## What now?
 
@@ -21,15 +24,19 @@ and inconsistent. If you set up your editor correctly, it will take care of most
 - Distribute appropriate code to people with those responsibilities
 - Have everyone upload their respective code to their sections of the gitlab
 
-## Code
-
-Everything is in [src/](src).
-A mostly untouched copy of [Scott's original code](https://git.therats.win/scott/threaded_network_chat) is also in [old/](old).
-A reformatted version of that code is also in [modified-example.cc](modified-example.cc) but it can be ignored.
-
 ## Build
 
 ```
-cd src/
 make
 ```
+
+## Usage
+
+```
+./ct # run as server
+```
+
+```
+./ct client # run as client
+```
+

log.cc

@@ -1,10 +1,10 @@
-#include <fstream>
-
-#include <ios>
 #include <sys/stat.h>
 
-#include "public.h"
+#include <fstream>
+#include <ios>
+
 #include "log.h"
+#include "public.h"
 
 // gets the number of lines in a file. returns -1 if there was an error.
 int
@@ -30,21 +30,24 @@ int
 writeToFile(string path, string line, bool incLineNum)
 {
 	ofstream file;
+	struct stat s;
 
 	/*
-	 * Hopefully this fixes the program messing up where the server/client
+	 * It APPEARS this fixes our problem where the program has to be run twice to work correctly.
-	 * doesn't start if the file wasn't created by a previous session.
 	 */
-	file.open(path, ios_base::out|ios_base::app);
+	if (stat(path.c_str(), &s) != 0) {
+		file.open(path, ios_base::out);
+		file << "SYSTEM: No log file found. Created a new one." << endl;
+		file.close();
+		linePos += 5; /* This seems to be what actually fixes everything. Not sure. */
+	}
+
+	file.open(path, ios_base::app);
 
 	if (file.is_open()) {
 		file << line << endl;
 		file.close();
-		// this probably would help but theres too much broken stuff
+		//linePos++ // We might also want this here at some point?
-		// right now to be sure if (incLineNum)
-		//{
-		//     linePos++;
-		// }
 		return 0;
 	} else {
 		// do something if it didn't work

main.cc

@@ -1,15 +1,10 @@
 #include <arpa/inet.h>
 #include <curses.h>
-#include <fstream>
-#include <iostream>
 #include <netdb.h>
 #include <netinet/in.h>
-#include <ostream>
 #include <pthread.h>
-#include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
-#include <string>
 #include <sys/socket.h>
 #include <sys/time.h>
 #include <sys/types.h>
@@ -17,6 +12,7 @@
 #include <sys/wait.h>
 #include <unistd.h>
 
+#include <string>
 #include <vector> //try to make the program work without this without doing anything awful but do it later
 
 #include "client.h"
@@ -34,7 +30,8 @@ const int DEFAULT_CUR_X = 2; // the x position of the preferred default cursor
 const int DEFAULT_CUR_Y = 12; // the x position of the preferred default cursor
 			      // position for message entry
 
-int mode = 0;		    // what mode is this program in. 0 = nothing. 1 = server. 2 = client
+chatmode_t mode = NO_MODE; // what mode is this program in. 0 = nothing. 1 = server. 2 = client
+
 int serverSocketDescriptor; // a global variable for storing the server socket
 			    // descriptor.
 int clientSocketDescriptor;
@@ -64,14 +61,12 @@ struct waitClientArgs {
 int
 main(int argc, char *argv[])
 {
-	if (argc > 1 && argv[1][0] == 'c') {
+	if (argc > 1 && (!strncmp(argv[1], "client", 6))) {
-		// client mode
 		logFileName = "client.txt";
-		mode = 2;
+		mode = CLIENT_MODE;
 	} else {
-		// server mode
 		logFileName = "server.txt";
-		mode = 1;
+		mode = SERVER_MODE;
 	}
 
 	// putting this lower down to circumvent the terminal brick when the
@@ -91,13 +86,20 @@ main(int argc, char *argv[])
 	char *userInput = new char[1024];
 	bool exit = false;
 
-	if (mode == 2) {
+	switch (mode) {
+	case CLIENT_MODE:
 		writeToFile(logFileName, "CLIENT MODE");
 		setupClient();
-	} else if (mode == 1) {
+		linePos++;
+		break;
+	case SERVER_MODE:
 		writeToFile(logFileName, "SERVER MODE");
 		setupServer(PORT_NUM);
-		linePos++; /* No idea why, but this needs to be here. */
+		linePos++;
+		break;
+	default:
+		goto leave;
+		break;
 	}
 
 	while (!exit) {
@@ -111,7 +113,7 @@ main(int argc, char *argv[])
 		/* clear message box / reset cursor */
 		move(12, 0);
 		printw(">\t\t\t");
-		move(12,2);
+		move(12, 2);
 
 		getstr(userInput);
 		writeToFile(logFileName, userInput);
@@ -125,13 +127,23 @@ main(int argc, char *argv[])
 			send(clientSocketDescriptor, (char *)userInput, strlen(userInput), 0);
 	}
 
+leave:
 	endwin();
-	//closeServer();
+	// closeServer();
 
-	if (mode == 1)
+	switch (mode) {
+	case SERVER_MODE:
 		closeServer();
-	else if (mode == 2)
+		break;
-		closeClient();
+	case CLIENT_MODE:
+		closeServer();
+		break;
+	default:
+		puts("Warn: program appears to have successfully finished without ever setting "
+		     "mode.");
+		return 1;
+		break;
+	}
 
 	return 0;
 }

public.h

@@ -5,11 +5,13 @@
 
 using namespace std;
 
+typedef enum { NO_MODE = 0, SERVER_MODE = 1, CLIENT_MODE = 2 } chatmode_t;
+
+extern chatmode_t mode;
 extern const int PORT_NUM;
 extern string IP_ADDRESS;
 extern const int DEFAULT_CUR_X;
 extern const int DEFAULT_CUR_Y;
-extern int mode;
 extern int serverSocketDescriptor;
 extern int clientSocketDescriptor;
 extern struct timeval start1, end1;

server.cc

@@ -1,7 +1,7 @@
-#include <cstdio>
 #include <curses.h>
 #include <netinet/in.h>
 #include <pthread.h>
+#include <stdio.h>
 #include <string.h>
 #include <sys/socket.h>
 #include <sys/time.h>