viarno reshenie energy booster

function energyBooster(input) {
    let fruit = input[0];
    let setSize = input[1];
    let setsCount = Number(input[2]);
    let price = 0;

    switch (fruit) {
        case "Watermelon":
            switch (setSize) {
                case "small":
                    price = setsCount * 2 * 56;
                    break;
                case "big":
                    price = setsCount * 5 * 28.70;
                    break;
            }
            break;

        case "Mango":
            switch (setSize) {
                case "small":
                    price = setsCount * 2 * 36.66;
                    break;
                case "big":
                    price = setsCount * 5 * 19.60;
                    break;
            }
            break;
        case "Pineapple":
            switch (setSize) {
                case "small":
                    price = setsCount * 2 * 42.10;
                    break;
                case "big":
                    price = setsCount * 5 * 24.80;
                    break;
            }
            break;
        case "Raspberry":
            switch (setSize) {
                case "small":
                    price = setsCount * 2 * 20;
                    break;
                case "big":
                    price = setsCount * 5 * 15.20;
                    break;
            }
    }


    if (price >= 400 && price <= 1000) {
        price = price - (price * 0.15);
        console.log(`${(price).toFixed(2)} lv.`);
    } else if (price > 1000) {
        price = price - (price * 0.50);
        console.log(`${(price).toFixed(2)} lv.`);
    } else {
        console.log(`${(price).toFixed(2)} lv.`);
    }
}