//TankCreationManager.cs allows players to change their tanks’ bases, towers and materials in order to customize it to their own playstyles and aesthetic preferences. It also allows players to preview the changes they’re making before committing to them.


using System.Collections;
using System.Collections.Generic;
using UnityEngine;
using UnityEngine.SceneManagement;
using UnityEngine.UI;
using UnityEngine.EventSystems;


public class TankCreationManager : MonoBehaviour
{
    [SerializeField]
    GameObject tankPrefab, confirmExitMenu;


    [SerializeField]
    Text baseNameText, baseMatText, towerNameText, towerMatText;


    MeshFilter prefabTankBaseMesh, prefabTankTowerMesh, previewTankBaseMesh, previewTankTowerMesh;


    int baseIndex = 0, towerIndex = 0, baseMatIndex = 0, towerMatIndex = 0;


    Mesh[][] allMeshes = new Mesh[2][];


    Dictionary<string, Material[]>[] allMats = new Dictionary<string, Material[]>[2];


    GameObject previewTank;


    float rotAmount = 0f, rotDir = 0f;


    private void Start()
    {
        Object[] loadedBaseMeshes = Resources.LoadAll("Meshes/Bases");
        Object[] loadedTowerMeshes = Resources.LoadAll("Meshes/Towers");


        allMeshes[0] = new Mesh[loadedBaseMeshes.Length];
        allMeshes[1] = new Mesh[loadedTowerMeshes.Length];


        for (int i = 0; i < loadedBaseMeshes.Length; i++)
            allMeshes[0][i] = (loadedBaseMeshes[i] as GameObject).GetComponent<MeshFilter>().sharedMesh;


        for (int i = 0; i < loadedTowerMeshes.Length; i++)
            allMeshes[1][i] = (loadedTowerMeshes[i] as GameObject).GetComponent<MeshFilter>().sharedMesh;


        prefabTankBaseMesh = tankPrefab.GetComponentInChildren<MeshFilter>();
        prefabTankTowerMesh = prefabTankBaseMesh.GetComponentInChildren<MeshFilter>();


        previewTank = Instantiate(tankPrefab);


        previewTank.GetComponent<TankScript>().enabled = false;


        previewTankBaseMesh = previewTank.GetComponentInChildren<MeshFilter>();
        previewTankTowerMesh = previewTankBaseMesh.GetComponentInChildren<MeshFilter>();


        previewTankBaseMesh.GetComponentInParent<TankScript>().enabled = false;


        foreach (ParticleSystem p in previewTankBaseMesh.GetComponentsInChildren<ParticleSystem>())
            p.Stop();


        previewTank.GetComponentInChildren<Canvas>().gameObject.SetActive(false);


        allMats[0] = new Dictionary<string, Material[]>();
        allMats[1] = new Dictionary<string, Material[]>();


        /*
        foreach (Mesh baseMesh in loadedBaseMeshes)
        {
            allMats[0].Add(baseMesh.name, (Material[])Resources.LoadAll("Materials/Bases/" + baseMesh.name, typeof(Material)));
        }
        foreach (Mesh towerMesh in loadedTowerMeshes)
        {
            allMats[1].Add(towerMesh.name, (Material[])Resources.LoadAll("Materials/Towers/" + towerMesh.name, typeof(Material)));
        }
        */
    }


    private void Update()
    {
        if (rotDir != 0f)
        {
            if (Input.GetButton("Submit"))
            {
                rotAmount = rotDir;
            }
            else if(!Input.GetKey(KeyCode.Mouse0))
                rotAmount = 0f;
        }
        if (rotAmount != 0f)
            previewTank.transform.Rotate(0f, rotAmount * Time.deltaTime, 0f);
    }


    public void RotationButtonSelected(float dir)
    {
        rotDir = dir;
        rotAmount = 0f;
    }


    public void ChangeBaseMat(int next)
    {
        baseMatIndex += next;
        if (baseMatIndex >= allMats[0][allMeshes[0][baseIndex].name].Length)
            baseMatIndex = 0;
        else if (baseMatIndex < 0)
            baseMatIndex = allMats[0][allMeshes[0][baseIndex].name].Length - 1;
        baseMatText.text = allMats[0][allMeshes[0][baseIndex].name][baseMatIndex].name;
    }


    public void ChangeTowerMat(int next)
    {
        towerMatIndex += next;
        if (towerMatIndex >= allMats[1][allMeshes[1][towerIndex].name].Length)
            towerMatIndex = 0;
        else if (towerMatIndex < 0)
            towerMatIndex = allMats[1][allMeshes[1][towerIndex].name].Length - 1;
        towerMatText.text = allMats[1][allMeshes[1][towerIndex].name][towerMatIndex].name;
    }


    public void ChangeBaseMesh(int next)
    {
        baseIndex += next;
        if (baseIndex >= allMeshes[0].Length)
            baseIndex = 0;
        else if (baseIndex < 0)
            baseIndex = allMeshes[0].Length - 1;
        previewTankBaseMesh.sharedMesh = allMeshes[0][baseIndex];
        ChangeBaseMat(next);
    }


    public void ChangeTowerMesh(int next)
    {
        towerIndex += next;
        if (towerIndex >= allMeshes[1].Length)
            towerIndex = 0;
        else if (towerIndex < 0)
            towerIndex = allMeshes[1].Length - 1;
        previewTankTowerMesh.sharedMesh = allMeshes[1][towerIndex];
    }


    public void RotateTank(float amount)
    {
        rotAmount = amount;
    }


    public void ReturnButtonPressed()
    {
        if (
            prefabTankBaseMesh.sharedMesh != previewTankBaseMesh.sharedMesh
            || prefabTankTowerMesh.sharedMesh != previewTankTowerMesh.sharedMesh
            || prefabTankBaseMesh.GetComponent<MeshRenderer>().sharedMaterial != previewTankBaseMesh.GetComponent<MeshRenderer>().sharedMaterial
            || prefabTankTowerMesh.GetComponent<MeshRenderer>().sharedMaterial != previewTankTowerMesh.GetComponent<MeshRenderer>().sharedMaterial
           )
        {
            ShowConfirmExitMenu(true);
        }
        else
            ReturnToMainMenu();
    }


    public void ShowConfirmExitMenu(bool show)
    {
        confirmExitMenu.SetActive(show);
    }


    public void ReturnToMainMenu()
    {
        SceneManager.LoadScene(0);
    }


    public void SaveSettings()
    {
        prefabTankBaseMesh.sharedMesh = previewTankBaseMesh.sharedMesh;
        prefabTankBaseMesh.GetComponent<MeshRenderer>().material = previewTankBaseMesh.GetComponent<MeshRenderer>().material;
        prefabTankTowerMesh.sharedMesh = prefabTankTowerMesh.sharedMesh;
        prefabTankTowerMesh.GetComponent<MeshRenderer>().material = previewTankTowerMesh.GetComponent<MeshRenderer>().material;
    }
}