#include "types.h"
#include <stdio.h>

void print_2d(int matrix[MAT_ROWS][MAT_COLUMNS])
{
    for (int row = 0; row < MAT_ROWS; row++) {
        for (int column = 0; column < MAT_COLUMNS; column++) {
            printf("%d ", matrix[row][column]);
        }
        printf("\n");
    }
}

bool save_to(FILE *output, int matrix[MAT_ROWS][MAT_COLUMNS])
{
    for (int row = 0; row < MAT_ROWS; row++) {
        for (int column = 0; column < MAT_COLUMNS; column++) {
            if (fprintf(output, "%d ", matrix[row][column]) < 0) {
                return false;
            }
        }
        if (fprintf(output, "\n") < 0) {
            return false;
        }
    }

    // TODO: return true when successful
    return true;
}

bool save_to_file(const char *filename, int matrix[MAT_ROWS][MAT_COLUMNS])
{
    // FILE * is the type used to represent files and data streams
    FILE *output = fopen(filename, "w");
    if (output == NULL) {
        return false;
    }

    // TODO: 1. open the destination file
    // TODO: 2. validate that the stream has opened correctly

    bool result = save_to(output, matrix);

    // TODO: 4. Don't forget to close the file!
    fclose(output);
    return result;
}

bool load_from(FILE *input, int matrix[MAT_ROWS][MAT_COLUMNS])
{
    // TODO: Read the matrix items from the file into the matrix
    int row = 0;
    int column = 0;
    int currNumber = 0;
    while (fscanf(input, "%d", &currNumber)) {
        matrix[row][column] = currNumber;
        if (++column >= MAT_COLUMNS && row < MAT_ROWS) {
            row++;
            column = 0;
        } else if (row >= MAT_ROWS) {
            break;
        }
    }

    // TODO: return true when successful
    return true;
}

bool load_from_file(const char *filename, int matrix[MAT_ROWS][MAT_COLUMNS])
{
    FILE *input = fopen(filename, "r");
    if (input == NULL) {
        return false;
    }
    // TODO: 1. open the source file
    // TODO: 2. validate that the stream has opened correctly

    bool result = load_from(input, matrix);

    // TODO: 4. Don't forget to close the file!
    fclose(input);
    return result;
}

#ifndef TEST_BUILD

int main(int argc, char *argv[])
{
    // Create and initialize a matrix
    int matrix[MAT_ROWS][MAT_COLUMNS];
    for (int row = 0; row < MAT_ROWS; row++) {
        for (int column = 0; column < MAT_COLUMNS; column++) {
            matrix[row][column] = (row + 1) * (column + 1);
        }
    }

    // Save the matrix to a file
    if (save_to_file(MAT_FILE_PATH, matrix) == false) {
        printf("Error during saving matrix to file!\n");
        return 1;
    }

    // Load matrix contents from a file
    int matrix_loaded[MAT_ROWS][MAT_COLUMNS];
    if (load_from_file(MAT_FILE_PATH, matrix_loaded) == false) {
        printf("Error during loading matrix from file!\n");
        return 1;
    }

    // Print the matrix contents
    print_2d(matrix_loaded);

    return 0;
}

#endif