#include "mystring.h"

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#define LINE_SIZE 1000 // assume each line has at most 999 characters

int main(int argc, char *argv[])

{

if (argc != 4)

{

printf("usage: %s <command> <inputfile> <outputfile>\n", argv[0]);

return EXIT_FAILURE;

}

FILE *infptr = fopen(argv[2], "r");

if (infptr == NULL)

{

printf("unable to open file %s!\n", argv[2]);

return EXIT_FAILURE;

}

FILE *outfptr = fopen(argv[3], "w");

if (outfptr == NULL)

{

printf("unable to open file %s!\n", argv[3]);

fclose(infptr);

return EXIT_FAILURE;

}

int num_lines = 0;

char buffer[LINE_SIZE];

// count the number of lines in the file

while (fgets(buffer, LINE_SIZE, infptr) != NULL)

{

num_lines++;

}

fseek(infptr, 0, SEEK_SET); // return to the beginning of the file

char **lines = malloc(sizeof(char *) * num_lines);

int i;

for (i = 0; i < num_lines; i++)

{

if (feof(infptr))

{

printf("not enough num_lines in file!\n");

fclose(infptr);

fclose(outfptr);

return EXIT_FAILURE;

}

lines[i] = malloc(sizeof(char) * LINE_SIZE);

fgets(lines[i], LINE_SIZE, infptr);

}

fclose(infptr);

int total_length = 0;

for (i = 0; i < num_lines; i++)

{

total_length += my_strlen(lines[i]);

}

// count the length of each line

if (strcmp(argv[1], "strlen") == 0)

{

for (i = 0; i < num_lines; i++)

{

fprintf(outfptr, "length: %d\n", my_strlen(lines[i]));

}

}

// for each line, count the occurrence of the first letter in the line

if (strcmp(argv[1], "countchar") == 0)

{

for (i = 0; i < num_lines; i++)

{

fprintf(outfptr, "count(%c): %d\n", lines[i][0],

my_countchar(lines[i], lines[i][0]));

}

}

if (strcmp(argv[1], "strupper") == 0)

{

for (i = 0; i < num_lines; i++) {

my_strupper(lines[i]);

fprintf(outfptr, "%s", lines[i]);

}

}

for (i = 0; i < num_lines; i++)

{

free(lines[i]);

}

free(lines);

fclose(outfptr);

return EXIT_SUCCESS;

}