local maincampaign = LUI.MenuBuilder.m_types_build["main_campaign"]
LUI.MenuBuilder.m_types_build["main_campaign"] = function(...)
    local initlist = LUI.Options.InitScrollingList
    LUI.Options.InitScrollingList = function(list) 
    	initlist(list, nil ,{
            rows = 10
        })
    end

    local menu = maincampaign(...)
    LUI.Options.InitScrollingList = initlist
    return menu
end

LUI.MenuBuilder.registerType("achievements_menu", function(root, controller)
    local menuwidth = 1129
	local menu = LUI.MenuTemplate.new(root, {
		menu_title = "@LUA_MENU_ACHIEVEMENTS",
		exclusiveController = 0,
		menu_width = menuwidth,
		menu_top_indent = LUI.MenuTemplate.spMenuOffset,
		showTopRightSmallBar = true,
		uppercase_title = true
	})

    local itemwidth = 80
    local itemspacing = 10
    local maxrowelements = 1
    
    local rowelements = 0
    local newrow = function()
        local currentrow = LUI.UIElement.new({
            topAnchor = true,
            leftAnchor = true,
            width = itemwidth * maxrowelements,
            height = itemwidth,
        })
    
        menu.list.currentrow = currentrow
        menu.list:addElement(currentrow)
        menu.list.currentrow:makeFocusable()

    end

    newrow()

    local addelement = function(element)
        local container = LUI.UIElement.new({
            topAnchor = true,
            leftAnchor = true,
            width = itemwidth,
            height = itemwidth,
            left = rowelements * (itemwidth + itemspacing)
        })
        container:addElement(element)
        container:makeFocusable()
        menu.list.currentrow:addElement(container)
        rowelements = rowelements + 1
        if (rowelements == maxrowelements) then
            newrow()
            rowelements = 0
        end
    end

    local achievementtable = achievements.table()
    local unlockcount = 0
    for i = 0, #achievementtable do
        if (achievementtable[i]) then
            unlockcount = unlockcount + 1
        end
    end

    for i = 0, achievements.count() - 1 do
        local trophyimage = LUI.UIImage.new({
            topAnchor = true,
            leftAnchor = true,
            width = menuwidth,
            height = itemwidth,
            material = RegisterMaterial("h2_btn_unfocused")  
        })

        trophyimage:setup9SliceImage()

        trophyimage:registerAnimationState("locked", {
            material = RegisterMaterial("h2_btn_unfocused_locked")
        })

        local stroke = LUI.UIImage.new({
            topAnchor = true,
            leftAnchor = true,
            width = menuwidth,
            height = itemwidth,
            material = RegisterMaterial("h2_btn_focused_stroke")  
        })

        stroke:registerAnimationState("hide", {
            alpha = 0
        })

        stroke:registerAnimationState("show", {
            alpha = 1
        })

        stroke:setup9SliceImage()

        local glow = LUI.UIImage.new({
            topAnchor = true,
            leftAnchor = true,
            left = -20,
            top = -20,
            width = menuwidth + 40,
            height = itemwidth + 40,
            material = RegisterMaterial("h2_btn_focused_outerglow")  
        })

        glow:registerAnimationState("hide", {
            alpha = 0
        })

        glow:registerAnimationState("show", {
            alpha = 1
        })

        glow:setup9SliceImage()

        local pattern = LUI.UIImage.new({
            topAnchor = true,
            leftAnchor = true,
            width = menuwidth,
            height = itemwidth,
            material = RegisterMaterial("h2_btn_dot_pattern")  
        })

        pattern:registerAnimationState("hide", {
            alpha = 0
        })

        pattern:registerAnimationState("show", {
            alpha = 1
        })

        stroke:animateToState("hide")
        pattern:animateToState("hide")
        glow:animateToState("hide")

        local achievementcontainer = LUI.UIElement.new({
            topAnchor = true,
            leftAnchor = true,
            width = menuwidth,
            height = itemwidth,
        })

        local image = LUI.UIImage.new({
            topAnchor = true,
            leftAnchor = true,
            top = itemspacing / 2,
            left = itemspacing / 2,
            width = itemwidth - itemspacing,
            height = itemwidth - itemspacing,
            material = RegisterMaterial("trophy_" .. i) 
        })

        image:registerAnimationState("locked", {
            topAnchor = true,
            leftAnchor = true,
            top = itemspacing / 2 + 15,
            left = itemspacing / 2 + 15,
            width = itemwidth - itemspacing - 30,
            height = itemwidth - itemspacing - 30,
            material = RegisterMaterial("icon_lock_mini") 
        })

        local textwidth = menuwidth - itemwidth - itemspacing * 2
        local title = LUI.UIText.new({
            topAnchor = true,
            leftAnchor = true,
            top = itemspacing + 5,
            left = itemwidth + 5,
            color = {
                r = 0.7,
                g = 0.7,
                b = 0.7,
            },
            width = textwidth,
            font = CoD.TextSettings.Font23.Font,
            height = CoD.TextSettings.Font23.Height
        })

        local desc = LUI.UIText.new({
            topAnchor = true,
            leftAnchor = true,
            top = itemspacing + CoD.TextSettings.Font23.Height + 10,
            left = itemwidth + 5,
            width = textwidth,
            alignment = LUI.Alignment.Left,
            color = {
                r = 0.7,
                g = 0.7,
                b = 0.7,
            },
            font = CoD.TextSettings.Font21.Font,
            height = CoD.TextSettings.Font21.Height
        })

        local focusedtext = {
            color = {r = 1, g = 1, b = 1}
        }
        desc:registerAnimationState("focused", focusedtext)
        title:registerAnimationState("focused", focusedtext)

        title:setText(Engine.ToUpperCase(Engine.Localize(achievements.getname(i))))
        desc:setText(Engine.Localize(achievements.getdetails(i)))

        achievementcontainer:setHandleMouse(true)
        local locked = not achievementtable[i]

        achievementcontainer:registerEventHandler("mouseenter", function()
            Engine.PlaySound(CoD.SFX.MouseOver)

            if (locked) then
                trophyimage:setImage(RegisterMaterial("h2_btn_focused_locked"))
            else
                stroke:animateToState("show")
                pattern:animateToState("show")
                title:animateToState("focused")
                desc:animateToState("focused")
                glow:animateToState("show")
            end
        end)
        achievementcontainer:registerEventHandler("mouseleave", function()
            if (locked) then
                trophyimage:setImage(RegisterMaterial("h2_btn_unfocused_locked"))
            else
                stroke:animateToState("hide")
                pattern:animateToState("hide")
                title:animateToState("default")
                desc:animateToState("default")
                glow:animateToState("hide")
            end
        end)
    
        achievementcontainer:addElement(trophyimage)
        achievementcontainer:addElement(pattern)
        achievementcontainer:addElement(glow)
        achievementcontainer:addElement(stroke)
        achievementcontainer:addElement(image)
        achievementcontainer:addElement(title)
        achievementcontainer:addElement(desc)

        if (locked) then
            trophyimage:animateToState("locked")
            image:animateToState("locked")
            title:setText(Engine.Localize("@LUA_MENU_UNKNOWN_ACHIEVEMENT"))
            desc:setText("")
        end

        addelement(achievementcontainer)
    end

    if (rowelements == 0) then
        menu.list.currentrow:close()
    end

	LUI.Options.InitScrollingList(menu.list, nil, {
        rows = 5
    })

    local createprogressbar = function()
        local barwidth = menuwidth + 17
        local progressbar = LUI.UIElement.new({
            bottomAnchor = true,
            leftAnchor = true,
            width = barwidth,
            bottom = -90,
            height = 15
        })

        local bg = LUI.UIImage.new({
            topAnchor = true,
            leftAnchor = true,
            rightAnchor = true,
            bottomAnchor = true,
            alpha = 0.55,
            color = Colors.grey_14,
        })

        local bar = LUI.UIImage.new({
            topAnchor = true,
            leftAnchor = true,
            bottomAnchor = true,
            width = 0,
            material = RegisterMaterial("h1_ui_progressbar_green")
        })

        progressbar.setprogress = function(fraction)
            bar:registerAnimationState("progress", {
                topAnchor = true,
                leftAnchor = true,
                bottomAnchor = true,
                width = fraction * (barwidth),
            })
            
            bar:animateToState("progress", 300)
        end

        progressbar.setprogress(unlockcount / achievements.count())

        progressbar:addElement(bg)
        progressbar:addElement(bar)
        progressbar:addElement(LUI.DecoFrame.new(nil, LUI.DecoFrame.Grey))

        return progressbar
    end

    menu:addElement(createprogressbar())
    menu:AddBackButton()

    return menu
end)