main.cpp

#include "difference.hpp"
#include "sum.hpp"

int main() {
    int stop = 0;
    while (stop == 0) {
        cout << "-----------------------MENU------------------------\n";
        cout << "1. Sum of Two Integers of Optional Base.(2 <= base <= 10)\n";
        cout << "2. Difference of Two Integers of Optional Base.\n";
        cout << "---------------------------------------------------\n";

        int option;
        cout << "Choose option: ";
        cin >> option;

        switch (option) {
            case 1:
            {
                int x, y;
                int base;

                cout << "Enter the base: ";
                cin >> base;

                if(2 <= base && base <= 10){
                    cout << "Enter the 1st number: ";
                    cin >> x;
                    cout << "Enter the 2nd number: ";
                    cin >> y;

                    while(checkValid(x,y,base))
                    {
                        cout << "Input Digits Should Be Smaller Than The Base.\n";
                        cout << "Enter the 1st number: ";
                        cin >> x;
                        cout << "Enter the 2nd number: ";
                        cin >> y;
                    }

                    stack<long> s = sumofBase(x,y,base);
                    cout << "Result: ";
                    printStack(s);
                }
                else
                {
                    string a, b;

                    cout << "Enter 1st number: ";
                    cin >> a;
                    a = upperCase(a); 
                    
                    cout << "Enter 2nd number: ";
                    cin >> b;
                    b = upperCase(b);
                    
                    while(!checkValid_36(a,b,base))
                    {
                        cout << "Input Digits Should Be Smaller Than The Base.\n";

                        cout << "Enter the 1st number: ";
                        cin >> a;
                        a = upperCase(a); 

                        cout << "Enter the 2nd number: ";
                        cin >> b;
                        b = upperCase(b);
                    }

                    stack<char> s;
                    s = sumOfBase_36(a, b, base);
                    cout << "Result: ";
                    printStack(s);

                }
                
                cout << "Stop? (1 - YES, 0 - NO): ";
                cin >> stop;
                break;
            }
            case 2:
            {
                string a, b;
                int base;

                cout << "Input base: ";
                cin >> base;
                cout << "Input number a: ";
                cin >> a;
                cout << "Input number b: ";
                cin >> b;
                
                string result = difference(a, b, base);
                if (result == "NaN") {
                    cout << "Function 'difference':" << endl;
                    cout << "Error: Cannot find the difference since the input is invalid." << endl;
                    return 0;
                }
                else {
                    cout << "(a - b)_" << base << " = " << result << "_" << base << endl;
                }

                cout << "Stop? (1 - YES, 0 - NO): ";
                cin >> stop;
                break;
            }
            default:
            {
                cout << "Option doesn't exist." << endl;
                break;
            }
        }
    }
   
    return 0;
}