#!/usr/bin/env python
"""
Fix Google Custom Search (GCS) off by one index issue
"""
# Standard library
import csv
import os
import sys
import textwrap
import traceback

# Third-party
from pygments import highlight
from pygments.formatters import TerminalFormatter
from pygments.lexers import PythonTracebackLexer

# Add parent directory so shared can be imported
sys.path.append(os.path.join(os.path.dirname(__file__), ".."))

# First-party/Local
import shared  # noqa: E402

# Setup
LOGGER, PATHS = shared.setup(__file__)

# Constants
FILE1_COUNT = os.path.join(PATHS["data_phase"], "gcs_1_count.csv")
FILE2_LANGUAGE = os.path.join(
    PATHS["data_phase"], "gcs_2_count_by_language.csv"
)
FILE3_COUNTRY = os.path.join(PATHS["data_phase"], "gcs_3_count_by_country.csv")


def read_data(file_path):
    data = []
    with open(file_path, "r", newline="") as file_obj:
        data = list(csv.reader(file_obj, dialect="unix"))
    return data


def increment_index(data):
    for i, row in enumerate(data):
        if i == 0:
            continue
        row[0] = int(row[0]) + 1
        data[i] = row
    return data


def write_data(file_path, data):
    with open(file_path, "w", newline="") as file_obj:
        writer = csv.writer(file_obj, dialect="unix")
        writer.writerows(data)


def main():
    for file_path in [FILE1_COUNT, FILE2_LANGUAGE, FILE3_COUNTRY]:
        data = read_data(file_path)
        data = increment_index(data)
        write_data(file_path, data)


if __name__ == "__main__":
    try:
        main()
    except shared.QuantifyingException as e:
        if e.exit_code == 0:
            LOGGER.info(e.message)
        else:
            LOGGER.error(e.message)
        sys.exit(e.exit_code)
    except SystemExit as e:
        LOGGER.error(f"System exit with code: {e.code}")
        sys.exit(e.code)
    except KeyboardInterrupt:
        LOGGER.info("(130) Halted via KeyboardInterrupt.")
        sys.exit(130)
    except Exception:
        traceback_formatted = textwrap.indent(
            highlight(
                traceback.format_exc(),
                PythonTracebackLexer(),
                TerminalFormatter(),
            ),
            "    ",
        )
        LOGGER.exception(f"(1) Unhandled exception:\n{traceback_formatted}")
        sys.exit(1)