using System;
using System.Collections.Generic;
using System.Linq;
using UnityEditor;
using UnityEngine;
using UnityEngine.SceneManagement;
using Random = UnityEngine.Random;

public class GameController : MonoBehaviour
{
    public static GameController i;
    public SceneAsset deployScene;
    public List<LevelInfo> levels;
    public List<EnemyInfo> enemies;
    public Dictionary<BulletType, int> inventory = new();

    public void Start() {
        if (i) { Destroy(gameObject); return; }
        i = GetComponent<GameController>();
        DontDestroyOnLoad(gameObject);
        levels = levels.Select(li => {
            li.hp = li.maxHp;
            li.enemies = new();
            return li;
        }).ToList();
        levels.ForEach(li => {
            AddEnemyInLevel(SelectEnemyWeighted().prefab, li.scene.name);
        });
        foreach (BulletType i in Enum.GetValues(typeof(BulletType))) {
            if (i == BulletType.None) continue;
            inventory[i] = 10;
        }
    }

    public Dictionary<GameObject, int> PopEnemies(string sceneName) {
        LevelInfo li = levels.Find(x => x.scene.name == sceneName);
        Dictionary<GameObject, int> levelEnemies = new(li.enemies);
        li.enemies.Clear();
        return levelEnemies;
    }

    public void AddEnemyInLevel(GameObject enemy, string sceneName) {
        Dictionary<GameObject, int> levelEnemies = levels.Find(x => x.scene.name == sceneName).enemies;
        if (levelEnemies.ContainsKey(enemy)) levelEnemies[enemy] += 1;
        else levelEnemies[enemy] = 1;
    }

    public LevelInfo GetLevel(string sceneName) {
        return levels.Find(x => x.scene.name == sceneName);
    }

    public void OnLevelClear(string sceneName, float timeSpent) {

    }

    public void OnLevelFailed(string sceneName, float timeSpent) {

    }

    EnemyInfo SelectEnemyWeighted() {
        float sumWeights = enemies.Sum(enemy => enemy.weight);
        float r = Random.Range(0, sumWeights);
        float curSum = 0;
        foreach (EnemyInfo en in enemies) {
            curSum += en.weight;
            if (curSum >= r) return en;
        }
        return enemies[^1];
    }

    public void LoadDeployScene() {
        SceneManager.LoadScene(deployScene.name);
    }

    public static void SetCursorState(bool locked) {
        Cursor.lockState = locked ? CursorLockMode.Locked : CursorLockMode.None;
        Cursor.visible = !locked;
    }

    public int GetBulletCount(BulletType bulletType) {
        return inventory[bulletType];
    }

    public bool ConsumeBullet(BulletType bulletType, int n = 1) {
        if (inventory[bulletType] >= n) {
            inventory[bulletType] -= n;
            return true;
        }
        return false;
    }
}

[Serializable]
public struct LevelInfo {
    public SceneAsset scene;
    public int hp;
    public int maxHp;
    public BulletType factory;
    public Dictionary<GameObject, int> enemies;
}

public enum BulletType {
    None,
    Rifle,
    Rocket,
    Minigun,
    Railgun
}

[Serializable]
public struct EnemyInfo {
    public GameObject prefab;
    public float weight;
}