package com.team5.spatial.repository;

import com.team5.spatial.Helpers.GeometryParser;
import com.team5.spatial.models.Request;
import com.team5.spatial.repository.contracts.ShapeRepository;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.PropertySource;
import org.springframework.core.env.Environment;
import org.springframework.dao.EmptyResultDataAccessException;
import org.springframework.stereotype.Repository;
import org.springframework.web.client.ResourceAccessException;

import java.io.IOException;
import java.sql.*;
import java.text.ParseException;
import java.util.HashMap;

@Repository
@PropertySource("classpath:application.properties")
public class ShapeRepositoryImpl implements ShapeRepository {
    private final String NO_RESULT_FOUND = "No data found for this location...";
    private String dbUrl, username, password;
    private GeometryParser geometryParser;

    @Autowired
    ShapeRepositoryImpl(Environment env, GeometryParser geometryParser) {
        dbUrl = env.getProperty("database.url");
        username = env.getProperty("database.username");
        password = env.getProperty("database.password");
        this.geometryParser = geometryParser;
    }

    @Override
    public HashMap getShape(Request request) throws Exception {
        String sql = "SELECT * FROM " + request.getRequestType() + " WHERE CONTAINS(SHAPE,Point(" + request.getX() + "," + request.getY() + "));";
        try (
                Connection connection = DriverManager.getConnection(dbUrl, username, password);
                Statement statement = connection.createStatement()
        ) {
            ResultSet propertiesSet = statement.executeQuery(sql); //Needs closing

            if (!propertiesSet.isBeforeFirst()) {
                System.out.println(NO_RESULT_FOUND);
                throw new EmptyResultDataAccessException(NO_RESULT_FOUND, 1);
            }

            ResultSetMetaData md = propertiesSet.getMetaData();

            int columns = md.getColumnCount();

            HashMap properties = new HashMap<>();

            while (propertiesSet.next()) {
                for (int i = 1; i <= columns; i++) {
                    if (md.getColumnName(i).equals("SHAPE")) {
                        properties.put("SHAPE", geometryParser.getGeometryFromInputStream(propertiesSet.getBinaryStream("SHAPE")).toString());
                    } else {
                        properties.put(md.getColumnName(i), propertiesSet.getObject(i));
                    }
                }
            }

            return properties;
        } catch (EmptyResultDataAccessException e) {
            throw new EmptyResultDataAccessException(e.getMessage(), e.getActualSize());
        } catch (ParseException | IOException e) {
            throw new IllegalArgumentException("FAILED_TO_PARSE_SHAPE");
        }
    }
}