from .__init__ import *


def basicAlgebraFunc(maxVariable=10, format='string'):
    a = random.randint(1, maxVariable)
    b = random.randint(1, maxVariable)
    c = random.randint(b, maxVariable)

    # calculate gcd
    def calculate_gcd(x, y):
        while (y):
            x, y = y, x % y
        return x

    i = calculate_gcd((c - b), a)
    x = f"{(c - b)//i}/{a//i}"

    if (c - b == 0):
        x = "0"
    elif a == 1 or a == i:
        x = f"{c - b}"

    if format == 'string':
        problem = f"{a}x + {b} = {c}"
        solution = x
        return problem, solution
    elif format == 'latex':
        problem = f"\\({a}x + {b} = {c}\\)"
        solution = "\\(" + x + "\\)"
        return problem, solution
    else:
        return a, b, c, x



basic_algebra = Generator("Basic Algebra", 11,
                          basicAlgebraFunc, ["maxVariable=10"])