local gc=love.graphics
local msIsDown,kbIsDown,tcTouches=love.mouse.isDown,love.keyboard.isDown,love.touch.getTouches
local setColor,rectangle=gc.setColor,gc.rectangle

local floor,rnd=math.floor,math.random
local ins,rem=table.insert,table.remove
local setFont,mStr=FONT.set,GC.mStr

local scene={}

local previewX={245,186,129,78,35}
local previewY={435,442,449,456,463}
local tileColor={
    {.39, 1.0, .39},
    {.39, .39, 1.0},
    {1.0, .70, .31},
    {.94, .31, .31},
    {.00, .71, .12},
    {.90, .20, .90},
    {.94, .47, .39},
    {.90, .00, .00},
    {.86, .86, .31},
    {.78, .31, .00},
    {.78, .55, .04},
    {.12, .12, .51},
}

local board,preview,cx,cy
local failPos
local startTime,time
local maxTile,maxNew
local state,progress
local fallingTimer
local score

local nexts
local invis
local fast

local function reset()
    progress={}
    state=0
    score=0
    time=0
    maxTile=2
    maxNew=2
    for i=1,5 do
        preview[i]=rnd(2)
    end
    for i=1,5 do for j=1,5 do
        board[i][j]=rnd(2)
    end end
    board[rnd(5)][rnd(5)]=2
    fallingTimer=false
    failPos=false
end
function scene.enter()
    BG.set('rainbow2')
    BGM.play('truth')
    preview={}
    board={{},{},{},{},{}}
    cx,cy=3,3
    startTime=0
    invis=false
    nexts=true
    reset()
end

local function merge()
    if fallingTimer or state==2 or not cx then return end
    if state==0 then
        state=1
        startTime=TIME()
    end
    if failPos==cy*10+cx then return end
    local chosen=board[cy][cx]
    local connected={{cy,cx}}
    local count=1
    repeat
        local c=rem(connected)
        local y,x=c[1],c[2]
        if board[y][x]~=0 then
            board[y][x]=0
            SYSFX.newShade(2,320+x*128-128,40+y*128-128,128,128)
            if x>1 and board[y][x-1]==chosen then ins(connected,{y,x-1})count=count+1 end
            if x<5 and board[y][x+1]==chosen then ins(connected,{y,x+1})count=count+1 end
            if y>1 and board[y-1][x]==chosen then ins(connected,{y-1,x})count=count+1 end
            if y<5 and board[y+1][x]==chosen then ins(connected,{y+1,x})count=count+1 end
        end
    until not connected[1]
    if count>1 then
        board[cy][cx]=chosen+1
        local getScore=3^(chosen-1)*math.min(floor(.5+count/2),4)
        score=score+getScore
        TEXT.show(getScore,cx*128+256,cy*128-40,40,'score')
        SYSFX.newRectRipple(2,320+cx*128-128,40+cy*128-128,128,128)
        SFX.play('lock')
        if chosen==maxTile then
            maxTile=chosen+1
            if maxTile>=6 then
                ins(progress,("%s - %.3fs"):format(maxTile,TIME()-startTime))
            end
            maxNew=
                maxTile<=4 and 2 or
                maxTile<=8 and 3 or
                maxTile<=11 and 4 or
                5
            SFX.play('reach')
        end
        if chosen>=5 then
            SFX.play(
                chosen>=9 and 'ren_mega' or
                chosen>=8 and 'spin_3' or
                chosen>=7 and 'spin_2' or
                chosen>=6 and 'spin_1' or
                'spin_0'
            )
        end
        fallingTimer=fast and 8 or 12
        failPos=false
    else
        board[cy][cx]=chosen
        failPos=cy*10+cx
    end
end
function scene.keyDown(key,isRep)
    if isRep then return end
    if key=='up' or key=='down' or key=='left' or key=='right' then
        if state==2 then return end
        if not cx then
            cx,cy=3,3
        else
            if key=='up' then
                if cy>1 then cy=cy-1 end
            elseif key=='down' then
                if cy<5 then cy=cy+1 end
            elseif key=='left' then
                if cx>1 then cx=cx-1 end
            elseif key=='right' then
                if cx<5 then cx=cx+1 end
            end
        end
    elseif key=='z' or key=='space' then
        merge()
    elseif key=='r' then
        reset()
    elseif key=='q' then
        if state==0 then
            nexts=not nexts
        end
    elseif key=='w' then
        if state==0 then
            invis=not invis
        end
    elseif key=='e' then
        if state==0 then
            fast=not fast
        end
    elseif key=='escape' then
        if tryBack() then
            SCN.back()
        end
    end
end
function scene.mouseMove(x,y)
    cx,cy=floor((x-192)/128),floor((y+88)/128)
    if cx<1 or cx>5 or cy<1 or cy>5 then
        cx,cy=false
    end
end
function scene.mouseDown(x,y)
    scene.mouseMove(x,y)
    merge()
end

scene.touchMove=scene.mouseMove
scene.touchDown=scene.mouseMove
function scene.touchClick(x,y)
    scene.mouseDown(x,y)
end

function scene.update()
    if state==1 then
        time=TIME()-startTime
        if fallingTimer then
            fallingTimer=fallingTimer-1
            if fallingTimer==0 then
                for i=5,2,-1 do
                    for j=1,5 do
                        if board[i][j]==0 then
                            board[i][j]=board[i-1][j]
                            board[i-1][j]=0
                        end
                    end
                end
                local noNewTile=true
                for i=1,5 do
                    if board[1][i]==0 then
                        board[1][i]=rem(preview,1)
                        preview[5]=
                            maxTile<=4 and rnd(2) or
                            maxTile<=8 and rnd(1+rnd(2)) or
                            maxTile<=11 and rnd(2+rnd(2)) or
                            rnd(2+rnd(3))
                        noNewTile=false
                    end
                end
                if noNewTile then
                    fallingTimer=false
                    for i=1,4 do for j=1,5 do if board[i][j]==board[i+1][j] then return end end end
                    for i=1,5 do for j=1,4 do if board[i][j]==board[i][j+1] then return end end end
                    state=2
                    SFX.play('fail')
                else
                    fallingTimer=fast and 4 or 5
                    SFX.play('touch')
                end
            end
        elseif fast and (
            msIsDown(1) or
            #tcTouches()>0 or
            kbIsDown('space')
        ) then
            merge()
        end
    end
end

function scene.draw()
    setFont(40)
    setColor(1,1,1)
    gc.print(("%.3f"):format(time),1026,50)
    gc.print(score,1026,100)

    -- Progress time list
    setFont(25)
    setColor(.7,.7,.7)
    for i=1,#progress do
        gc.print(progress[i],1000,140+30*i)
    end

    -- Previews
    if nexts then
        gc.setColor(COLOR.dX)
        rectangle('fill',20,450,280,75)
        gc.setLineWidth(6)
        setColor(1,1,1)
        rectangle('line',20,450,280,75)
        for i=1,5 do
            setFont(85-10*i)
            gc.setColor(tileColor[preview[i]])
            gc.print(preview[i],previewX[i],previewY[i])
        end
    end

    if state==2 then
        -- Draw no-setting area
        setColor(1,0,0,.3)
        rectangle('fill',15,200,285,210)
    end
    gc.setLineWidth(10)
    setColor(COLOR[
        state==1 and (fast and 'R' or 'W') or
        state==0 and 'G' or
        state==2 and 'Y'
    ])
    rectangle('line',315,35,650,650)

    gc.setLineWidth(4)
    setFont(70)
    local hide=invis and state==1
    for i=1,5 do
        for j=1,5 do
            local N=board[i][j]
            if N>0 then
                if hide and N>maxNew then
                    setColor(COLOR.dH)
                    rectangle('fill',320+j*128-128,40+i*128-128,128,128)
                    setColor(1,1,1,.3)
                    mStr("?",j*128+256,i*128-75)
                else
                    if N<=12 then
                        setColor(tileColor[N])
                    elseif N<=14 then
                        setColor(COLOR.rainbow(4*TIME()-i-j))
                    else
                        setColor(0,0,0,1-math.abs(TIME()%.5-.25)*6-.25)
                    end
                    rectangle('fill',320+j*128-128,40+i*128-128,128,128)
                    setColor(1,1,1,.9)
                    mStr(N,j*128+256,i*128-75)
                end
            end
        end
    end
    if state<2 and cx then
        setColor(1,1,1,.6)
        gc.setLineWidth(10)
        rectangle('line',325+cx*128-128,45+cy*128-128,118,118)
    end
    setFont(50)
    setColor(1,1,1)
    mStr("Just Get Ten",160,580)
end

scene.widgetList={
    WIDGET.newButton{name='reset',x=160,y=100,w=180,h=100,color='lG',font=50,fText=CHAR.icon.retry_spin,code=pressKey'r'},
    WIDGET.newSwitch{name='next', x=240,y=235,lim=200,font=40,disp=function() return nexts end,code=pressKey'q',hideF=function() return state==1 end},
    WIDGET.newSwitch{name='invis',x=240,y=305,lim=200,font=40,disp=function() return invis end,code=pressKey'w',hideF=function() return state==1 end},
    WIDGET.newSwitch{name='fast', x=240,y=375,lim=200,font=30,disp=function() return fast end,code=pressKey'e',hideF=function() return state==1 end},
    WIDGET.newButton{name='back', x=1140,y=640,w=170,h=80,sound='back',font=60,fText=CHAR.icon.back,code=backScene},
}

return scene