#include "../include/NumberHelper.h"
#include <iostream>
#include <string>
#include <sstream>
#include <vector>

vector<int> NumberHelper::getRolls(string csv){
    vector<int> rolls;
    stringstream ss(csv);
    string number;
    
    // Get number from the CSV file, split by comma, and convert to int.
    while (getline(ss, number, ',')) {
        rolls.push_back(stoi(number));
    }
    
    return rolls;
}

bool NumberHelper::validateRolls(vector<int> rolls){
    int rollCount = rolls.size();
    
    for(int i = 0; i < rollCount; ++i) {
        if(rolls[i] > 10 || rolls[i] < 0) {
            cerr << "Number: " << rolls[i] << " is invalid.";
            return false;
        }
    }
    
    // A game can not have less than 12 rolls, and more than 21.
    if (rollCount < 12 || rollCount > 21) {
        cerr << "Incorrect amount of rolls.";
        return false;
    }
    
    // 10th frame rule.
    if (rollCount == 21) {
        if (rolls[18] != 10 && rolls[18] + rolls[19] != 10) {
            return false;
        }
    }
    
    return true;
}

vector<Frame> NumberHelper::createFrames(vector<int> rolls){
    vector<Frame> frame;
        
    int roll = 0;
    
    while (roll != rolls.size())
    {
        // Strike
        if (rolls[roll] == 10)
        {
            // If we're on our last frame, and roll a strike, we're given two bonus rolls.
            if (roll + 3 == rolls.size() && rolls[roll - 3] == 10) {
                frame.push_back(CreateBonusFrame(rolls[roll], rolls[roll + 1], rolls[roll + 2]));
                break;
            }
            
            frame.push_back(CreateStrikeFrame(10));
            
            roll += 1;
        }
        
        // Spare
        else if (rolls[roll] + rolls[roll + 1] == 10)
        {
            // If we're on our last frame, and roll a spare, we're given a bonus roll.
            if (roll + 3 == rolls.size() && rolls[roll] + rolls[roll + 1] == 10)
            {
                frame.push_back(CreateBonusFrame(rolls[roll], rolls[roll + 1], rolls[roll + 2]));
                break;
            }
            
            frame.push_back(CreateFreeFrame(rolls[roll], rolls[roll + 1]));
            
            roll += 2;
        }
        
        // Open Frame
        else
        {
            frame.push_back(CreateFreeFrame(rolls[roll], rolls[roll + 1]));
            
            roll += 2;
        }
    }
    
    return frame;
}

Frame NumberHelper::CreateStrikeFrame(int i){
    struct Frame frame = Frame();
    
    vector<int> rolls;
    
    rolls.push_back(i);
    
    frame.Roll = rolls;
    
    return frame;
}

Frame NumberHelper::CreateFreeFrame(int i, int j){
    struct Frame frame = Frame();
    
    vector<int> rolls;
    
    rolls.push_back(i);
    rolls.push_back(j);
    
    frame.Roll = rolls;
    
    return frame;
}

Frame NumberHelper::CreateBonusFrame(int i, int j, int k){
    struct Frame frame = Frame();
    
    vector<int> rolls;
    
    rolls.push_back(i);
    rolls.push_back(j);
    rolls.push_back(k);
    
    frame.Roll = rolls;
    
    return frame;
}