#!/usr/bin/python
# -*- coding:utf-8 -*-
import RPi.GPIO as GPIO
import paho.mqtt.client as mqtt
import serial
import time, os
import csv
from datetime import datetime

# MQTT Broker settings
mqttBroker ="65.108.199.212"
myhost = os.uname()[1]
client = mqtt.Client(myhost)
# client.connect(mqttBroker, 1883)

ser = serial.Serial('/dev/ttyS0',115200)
ser.flushInput()

rec_buff = ''
time_count = 0

# CSV file settings
csv_filename = f"gps_data_{myhost}.csv"
csv_headers = ["Timestamp", "Latitude", "Longitude"]

def convert_to_decimal(coord_str):
    coord = coord_str[:-2]
    direction = coord_str[-1]
    
    degrees = float(coord[:2] if direction in ['N', 'S'] else coord[:3])
    minutes = float(coord[2:] if direction in ['N', 'S'] else coord[3:])
    
    decimal = degrees + (minutes / 60)
    
    if direction in ['S', 'W']:
        decimal = -decimal
    
    return decimal

def parse_gps_data(gps_string):
    parts = gps_string.split(',')
    if len(parts) != 4:
        return "Invalid GPS data"
    
    lat = convert_to_decimal(parts[0] + ',' + parts[1])
    lon = convert_to_decimal(parts[2] + ',' + parts[3])
    
    return f"{lat:.6f}", f"{lon:.6f}"

def write_to_csv(lat, lon):
    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
    with open(csv_filename, mode='a', newline='') as file:
        writer = csv.writer(file)
        writer.writerow([timestamp, lat, lon])

def send_at(command,back,timeout):
    rec_buff = ''
    ser.write((command+'\r\n').encode())
    time.sleep(timeout)
    if ser.inWaiting():
        time.sleep(0.01)
        rec_buff = ser.read(ser.inWaiting())
    if rec_buff != '':
        if back not in rec_buff.decode():
            print(command + ' ERROR')
            print(command + ' back:\t' + rec_buff.decode())
            return 0
        else:
            gps_data = rec_buff.decode()[13:-36]
            print("Raw GPS data:", gps_data)
            lat, lon = parse_gps_data(gps_data)
            converted_gps = f"{lat}, {lon}"
            print("Converted GPS data for Google Maps:", converted_gps)
            client.publish("iiot/"+ myhost +"/gps", converted_gps)
            write_to_csv(lat, lon)
            return 1
    else:
        print('GPS no está listo')
        return 0

def get_gps_position():
    rec_null = True
    answer = 0
    print('Iniciando GPS')
    rec_buff = ''
    send_at('AT+CGPS=1,1','OK',1)
    time.sleep(2)
    while rec_null:
        answer = send_at('AT+CGPSINFO','+CGPSINFO: ',1)
        client.publish("iiot/"+ myhost +"/gps", rec_buff, 0)
        if 1 == answer:
            answer = 0
            if ',,,,,,' in rec_buff:
                print('GPS no está listo')
                rec_null = False
                time.sleep(1)
        else:
            print('error %d'%answer)
            rec_buff = ''
            send_at('AT+CGPS=0','OK',1)
            return False
        time.sleep(1.5)

def power_on(power_key):
    print('SIM7600X is starting:')
    GPIO.setmode(GPIO.BCM)
    GPIO.setwarnings(False)
    GPIO.setup(power_key,GPIO.OUT)
    time.sleep(0.1)
    GPIO.output(power_key,GPIO.HIGH)
    time.sleep(2)
    GPIO.output(power_key,GPIO.LOW)
    time.sleep(20)
    ser.flushInput()
    print('SIM7600X is ready')

def power_down(power_key):
    print('SIM7600X is loging off:')
    GPIO.output(power_key,GPIO.HIGH)
    time.sleep(3)
    GPIO.output(power_key,GPIO.LOW)
    time.sleep(18)
    print('Good bye')

# Create CSV file with headers if it doesn't exist
if not os.path.exists(csv_filename):
    with open(csv_filename, mode='w', newline='') as file:
        writer = csv.writer(file)
        writer.writerow(csv_headers)

try:
    #power_on(power_key)
    get_gps_position()
    #power_down(power_key)
except:
    if ser != None:
        ser.close()
    #power_down(power_key)
        GPIO.cleanup()
if ser != None:
        ser.close()
        GPIO.cleanup()