[CC] turtleFactory

function makeEnum(vals)
    -- Expects vals to be a numeric array, ie vals = {blah,blah2, ...}
    out = {}
    for k,v in pairs(vals) do
        out[v]=k
    end
    return out
end

--================ ASSEMBLY LINE FORMAT ================--
-- Each var describes the position of a block along the production line.
-- The turtle will attempt to seek those blocks on its bottom, and starts
-- at position 0.
INPUT_CHEST_POS = 0
MF_EXTRUDE_POS = 1
TEMP_CHEST_POS = 2
MF_ROLL_POS = 3
OUTPUT_CHEST_POS = 4

PROD_LINE_ORIENTATION = 'up' -- the direction the turtle is from the
    -- production line's point of view.

--================ GENERATE RECIPES =====================--
--[[ Recipe format : table 't'
    * t.tool = raw | craft | mf_roll | mf_extrude
    * if raw : the material is a raw material, gathered from input chest.
    * if craft:
        o t.recipe = 3*3 array, recipe. Values belong to recipes enum.
        o Must be crafted as turtle craft.
    * if mf_roll:
        o Must be processed by a metal former in roll mode
        o t.material : belongs to the recipes enum, to be fed to the MF
    * if mf_extrude:
        o Must be processed by a metal former in extrude mode
        o t.material : idem mf_roll
--]]

RECIPES_NAMES = { 'air','iron_plate','tin_plate','tin_casing','tin_cable',
    'tin_ins_cable','re_batt','copper_cable','copper_ins_cable',
    'elec_circ','iron_casing','coil','elec_motor','power_unit',
    'chainsaw','drill','tin','iron','rubber','copper','redstone' }

NAMES_ENUM = makeEnum(RECIPES_NAMES)

CC_NAMES = {'','ic2.itemplateiron','ic2.itemplatetin','ic2.itemcasingtin',
    'ic2.itemtincable','ic2.itemtincablei','ic2.itembatredischarged',
    'ic2.itemcableo','ic2.itemcable','ic2.itempartcircuit','ic2.itemironcasing',
    'ic2.itempartcoil','ic2.itempartelemotor','ic2.itempowerunit',
    'ic2.itemtoolchainsaw','ic2.itemtooldrill', 'ic2.itemingottin',
    'item.ingotiron','ic2.itemrubber','ic2.itemingotcopper','item.redstone'
}

STACK_SIZES = {64,64,64,64,64,64,16,64,64,64,64,64,64,64,1,1,64,64,64,64,64}
CRAFT_RATIO = {1,1,1,2,3,1,1,3,1,1,2,1,1,1,1,1,1,1,1,1,1}

RECIPES = {}
RECIPES[NAMES_ENUM['air']] = { tool='raw' }
RECIPES[NAMES_ENUM['iron_plate']] = { tool='mf_roll', material=NAMES_ENUM['iron'] }
RECIPES[NAMES_ENUM['tin_plate']] = { tool='mf_roll', material=NAMES_ENUM['tin'] }
RECIPES[NAMES_ENUM['tin_casing']] = { tool='mf_roll',
    material=NAMES_ENUM['tin_plate'] }
RECIPES[NAMES_ENUM['tin_cable']] = { tool='mf_extrude', material=NAMES_ENUM['tin'] }
RECIPES[NAMES_ENUM['tin_ins_cable']] = { tool='craft', recipe={
    {NAMES_ENUM['tin_cable'],NAMES_ENUM['rubber'],1},{1,1,1},{1,1,1}} }
RECIPES[NAMES_ENUM['re_batt']] = { tool='craft', recipe={
    {1,NAMES_ENUM['tin_ins_cable'],1},
    {NAMES_ENUM['tin_casing'],NAMES_ENUM['redstone'],NAMES_ENUM['tin_casing']},
    {NAMES_ENUM['tin_casing'],NAMES_ENUM['redstone'],NAMES_ENUM['tin_casing']}}}
RECIPES[NAMES_ENUM['copper_cable']] = { tool='mf_extrude',
    material=NAMES_ENUM['copper'] }
RECIPES[NAMES_ENUM['copper_ins_cable']] = { tool='craft', recipe={
    {NAMES_ENUM['copper_cable'],NAMES_ENUM['rubber'],1},{1,1,1},{1,1,1}} }
RECIPES[NAMES_ENUM['elec_circ']] = { tool='craft', recipe={
    {NAMES_ENUM['copper_ins_cable'],NAMES_ENUM['copper_ins_cable'],NAMES_ENUM['copper_ins_cable']},
    {NAMES_ENUM['redstone'],NAMES_ENUM['iron_plate'],NAMES_ENUM['redstone']},
    {NAMES_ENUM['copper_ins_cable'],NAMES_ENUM['copper_ins_cable'],NAMES_ENUM['copper_ins_cable']}}}
RECIPES[NAMES_ENUM['iron_casing']] = { tool='mf_roll',
    material=NAMES_ENUM['iron_plate'] }
RECIPES[NAMES_ENUM['coil']] = { tool='craft', recipe={
    {NAMES_ENUM['copper_cable'],NAMES_ENUM['copper_cable'],NAMES_ENUM['copper_cable']},
    {NAMES_ENUM['copper_cable'],NAMES_ENUM['iron'],NAMES_ENUM['copper_cable']},
    {NAMES_ENUM['copper_cable'],NAMES_ENUM['copper_cable'],NAMES_ENUM['copper_cable']}}}
RECIPES[NAMES_ENUM['elec_motor']] = { tool='craft', recipe={
    {1,NAMES_ENUM['tin_casing'],1},
    {NAMES_ENUM['coil'],NAMES_ENUM['iron'],NAMES_ENUM['coil']},
    {1,NAMES_ENUM['tin_casing'],1}}}
RECIPES[NAMES_ENUM['power_unit']] = { tool='craft', recipe={
    {NAMES_ENUM['re_batt'],NAMES_ENUM['copper_cable'],NAMES_ENUM['iron_casing']},
    {NAMES_ENUM['re_batt'],NAMES_ENUM['elec_circ'],NAMES_ENUM['elec_motor']},
    {NAMES_ENUM['re_batt'],NAMES_ENUM['copper_cable'],NAMES_ENUM['iron_casing']}}}
RECIPES[NAMES_ENUM['chainsaw']] = { tool='craft', recipe={
    {1,NAMES_ENUM['iron_plate'],NAMES_ENUM['iron_plate']},
    {NAMES_ENUM['iron_plate'],NAMES_ENUM['iron_plate'],NAMES_ENUM['iron_plate']},
    {NAMES_ENUM['power_unit'],NAMES_ENUM['iron_plate'],1}}}
RECIPES[NAMES_ENUM['drill']] = { tool='craft', recipe={
    {1,NAMES_ENUM['iron_plate'],1},
    {NAMES_ENUM['iron_plate'],NAMES_ENUM['iron_plate'],NAMES_ENUM['iron_plate']},
    {NAMES_ENUM['iron_plate'],NAMES_ENUM['power_unit'],NAMES_ENUM['iron_plate']}}}
RECIPES[NAMES_ENUM['tin']] = { tool='raw' }
RECIPES[NAMES_ENUM['iron']] = { tool='raw' }
RECIPES[NAMES_ENUM['rubber']] = { tool='raw' }
RECIPES[NAMES_ENUM['copper']] = { tool='raw' }
RECIPES[NAMES_ENUM['redstone']] = { tool='raw' }

--=================== END CONSTANTS ========================--

turtlePos = -1

function goToPos(pos)
    while(pos > turtlePos) do
        turtle.forward()
        turtlePos = turtlePos + 1
    end
    while(pos < turtlePos) do
        turtle.back()
        turtlePos = turtlePos - 1
    end
end

function getFromBlock(recipeId, number)
    while true do
        inv = peripheral.wrap('bottom')
        inv.condenseItems()
        nbSlots = inv.getInventorySize()
        for slot=1,nbSlots do
            obj = inv.getStackInSlot(slot)
            if(obj ~= nil and obj.raw_name == CC_NAMES[recipeId]) then
                --Since we've condensed and we are assured that number < stackSize
                --we can safely take only from the first slot we find
                inv.pushItemIntoSlot(PROD_LINE_ORIENTATION,slot,number,1)
                return
            end
        end
        -- We've tried the whole inventory! The item is not present
        print('Missing '..RECIPES_NAMES[recipeId].." x"..number)
        print('Press enter to try again...')
        read()
    end
end

function getFirstFreeSlot(inv)
    size = inv.getInventorySize()
    for i=1,size do
        if inv.getStackInSlot(i) == nil then
            return i
        end
    end
    print('woah, temp buffer full!')
    error()
end

function stashItems(fromSlot, toSlot)
    goToPos(TEMP_CHEST_POS)
    inv = peripheral.wrap('bottom')
    if(toSlot == nil) then
        toSlot = getFirstFreeSlot(inv)
    end
    inv.pullItemIntoSlot(PROD_LINE_ORIENTATION,fromSlot,nil,toSlot)
    return toSlot
end

function pushToOutput(number)
    -- Pushes <number> items from #1 and puts it into output chest
    goToPos(OUTPUT_CHEST_POS)

    ch = peripheral.wrap('bottom')
    ch.pullItem(PROD_LINE_ORIENTATION, 1, number)
end

function processInMF()
    -- Assumes the turtle to be above the right MF, puts everything from #1
    -- into MF and waits to take it back.

    -- MF input slot: 7 ; output slot: 2
    mf = peripheral.wrap('bottom')
    mf.pullItem(PROD_LINE_ORIENTATION, 1, nil, 7)
    while mf.getStackInSlot(7) ~= nil or mf.getProgress() < 100 do
        os.sleep(1)
    end -- We're now done
    mf.pushItem(PROD_LINE_ORIENTATION, 2, nil, 1)
end

function craftRecipe(recipeId,number,final)
    -- Crafts [number] items, then put them in the 1st slot (or at least attempts)
    -- Assumes this slot is empty! Also assumes number <= stackSize
    if(final == nil) then
        final=false
    end

    if(RECIPES[recipeId].tool == 'raw') then
        if(recipeId == 1) then -- air, ie nothing
            return -- We're done ;)
        end

        goToPos(INPUT_CHEST_POS)
        getFromBlock(recipeId,number)
    elseif(RECIPES[recipeId].tool == 'craft') then
        local recipe = RECIPES[recipeId].recipe
        local maxProcessPack = 0
        for i=1,3 do for j=1,3 do
            maxProcessPack = math.max(STACK_SIZES[recipe[i][j]], maxProcessPack)
        end end

        local stashSlot = nil
        while number > 0 do
            local tempCraftSlots = {{-1,-1,-1},{-1,-1,-1},{-1,-1,-1}}
            for i=1,3 do
                for j=1,3 do
                    craftRecipe(recipe[i][j],math.min(maxProcessPack,number))
                    if(recipe[i][j] ~= 1) then -- not air
                        goToPos(TEMP_CHEST_POS)
                        tempCraftSlots[i][j]=stashItems(1)
                    end
                end
            end
            -- subcrafts done, now assemble
            goToPos(TEMP_CHEST_POS)
            tempCh = peripheral.wrap('bottom')
            for i=1,3 do
                for j=1,3 do
                    if(recipe[i][j] ~= 1) then --not air
                        tempCh.pushItem(PROD_LINE_ORIENTATION,
                            tempCraftSlots[i][j], nil, i*4+j+1)
                    end
                end
            end
            turtle.craft(maxProcessPack)

            if(number > 0) then -- Drop it in temp chest
                goToPos(TEMP_CHEST_POS)
                stashSlot = stashItems(1,stashSlot)
            end
            number = number - maxProcessPack -- Might be below the real number
            -- but it is okay
        end
        if stashSlot ~= nil then -- Must get stashed items
            -- Already at temp chest's pos
            inv = peripheral.wrap('bottom')
            inv.pushItemIntoSlot(PROD_LINE_ORIENTATION,stashSlot,nil,1)
        end
    elseif(RECIPES[recipeId].tool == 'mf_roll') then
        craftRecipe(RECIPES[recipeId].material,
                    math.ceil(number / CRAFT_RATIO[recipeId]),
                    false)
        -- Now we have all we want in first slot (maybe more?)
        goToPos(MF_ROLL_POS)
        processInMF()

        remainder = (CRAFT_RATIO[recipeId] - number % CRAFT_RATIO[recipeId]) %
            CRAFT_RATIO[recipeId]
        if(remainder > 0) then
            pushToOutput(remainder)
        end
    elseif(RECIPES[recipeId].tool == 'mf_extrude') then
        craftRecipe(RECIPES[recipeId].material,
                    math.ceil(number / CRAFT_RATIO[recipeId]),
                    false)
        -- Now we have all we want in first slot (maybe more?)
        goToPos(MF_EXTRUDE_POS)
        processInMF()

        remainder = (CRAFT_RATIO[recipeId] - number % CRAFT_RATIO[recipeId]) %
            CRAFT_RATIO[recipeId]
        if(remainder > 0) then
            pushToOutput(remainder)
        end
    end

    if(final) then
        -- Drop everything into the output chest
        pushToOutput(nil) -- nil : everything
        goToPos(-1)
    end
end

function craftName(name, number)
    if(NAMES_ENUM[name] == nil) then
        print("Invalid name")
    else
        craftRecipe(NAMES_ENUM[name],number,true)
    end
end