from csv import reader
import config
from domain.accelerometer import Accelerometer



class FileReader:
    def __init__(
        self, data_filename: str,
    ) -> None:
        self.file_path = data_filename
        pass

    def read(self):
        return self.getNextValue()

    def startReading(self, *args, **kwargs):
        self.file = open(self.file_path, newline='')
        self.file_reader = reader(self.file)
        file_header = next(self.file_reader)
        
        self.x_idx = file_header.index('X')
        self.y_idx = file_header.index('Y')
        self.z_idx = file_header.index('Z')

    def getNextValue(self):
        row = next(self.file_reader, None)
        if row is None:
            self._rewind_file()
            row = next(self.file_reader)
        x = int(row[self.x_idx])
        y = int(row[self.y_idx])
        z = int(row[self.z_idx])
        return Accelerometer(x=x, y=y, z=z)
        
    def _rewind_file(self):
        self.file.seek(0)
        self.file_reader = reader(self.file)
        next(self.file_reader)
        
    def stopReading(self, *args, **kwargs):
        if self.file:
            self.file.close()
        self.file_reader = None