ESOUI SVN ZAMNotebook

-- ZAM_Notebook © ZAM Network LLC

-- All Rights Reserved

local wm = WINDOW_MANAGER

local em = EVENT_MANAGER

local znb, db, indexPool, currentlyViewing

local tinsert = table.insert

local tremove = table.remove

local defaults = {

        anchor = {

                a = CENTER,

                b = CENTER,

                x = 0,

                y = 0,

        }

}

local function ProtectText(text)        --protect against backslashes corrupting our saved variables

        return text:gsub([[\]], [[%%92]])

end

local function UnprotectText(text)      --convert back to what should be displayed

        return text:gsub([[%%92]], [[\]])

end

local buttonCount = 1

local function CreateIndexButton(indexPool)

        local button = ZO_ObjectPool_CreateControl("ZAM_NotebookIndex", indexPool, znb.menu.scrollChild)

        local anchorBtn = buttonCount == 1 and znb.menu.scrollChild or indexPool:AcquireObject(buttonCount-1)

        button:SetAnchor(TOPLEFT, anchorBtn, buttonCount == 1 and TOPLEFT or BOTTOMLEFT)

        button:SetHorizontalAlignment(TEXT_ALIGN_LEFT)

        button:SetWidth(190)

        button:SetHandler("OnClicked", function(self)

                                znb.titleEdit:SetText(UnprotectText(self.data.title))

                                znb.edit:SetText(UnprotectText(self.data.text))

                                currentlyViewing = self.id

                                znb.save.highlight:SetHidden(true)

                                znb.selectedHL:SetHidden(false)

                                znb.selectedHL:ClearAnchors()

                                znb.selectedHL:SetAnchorFill(self)

                        end)

        buttonCount = buttonCount + 1

        return button

end

local function RemoveIndexButton(button)

        button:SetHidden(true)

end

local function PopulateScrollList()

        local numPages = #db.pages

        for i = 1, numPages do

                local button = indexPool:AcquireObject(i)

                button.data = db.pages[i]

                button.id = i

                button:SetText(UnprotectText(button.data.title))

                button:SetHidden(false)

        end

        local activePages = indexPool:GetActiveObjectCount()

        if activePages > numPages then

                for i = numPages+1, activePages do

                        indexPool:ReleaseObject(i)

                end

        end

end

local function CreateNotebook()

        --main window

        znb = wm:CreateTopLevelWindow("ZAM_Notebook")

        znb:SetDimensions(800, 650)

        znb:SetAnchor(db.anchor.a, GuiRoot, db.anchor.b, db.anchor.x, db.anchor.y)

        znb:SetClampedToScreen(true)

        znb:SetMouseEnabled(true)

        znb:SetMovable(true)

        znb:SetHandler("OnReceiveDrag", function(self)

                        --if not db.locked then

                                self:StartMoving()

                        --end

                end)

        znb:SetHandler("OnMouseUp", function(self)

                        self:StopMovingOrResizing()

                        local _,a,_,b,x,y = self:GetAnchor()

                        db.anchor = {["a"]=a, ["b"]=b, ["x"]=x, ["y"]=y}

                end)

        --window bg

        znb.bg = wm:CreateControl("ZAM_NotebookBG", znb, CT_TEXTURE)

        znb.bg:SetTexture("ZAM_Notebook\\NotebookBG.dds")

        znb.bg:SetTextureCoords(0, 0.98, 0, 0.85)

        znb.bg:SetAnchorFill(znb)

        --window title

        znb.title = wm:CreateControl("ZAM_NotebookTitle", znb, CT_LABEL)

        znb.title:SetAnchor(TOPLEFT, znb, TOPLEFT, 50, 50)

        znb.title:SetFont("ZoFontWindowTitle")

        znb.title:SetText("ZAM Notebook")

        --divider below title

        znb.titleDiv = wm:CreateControl("ZAM_NotebookTitleDiv", znb, CT_TEXTURE)

        znb.titleDiv:SetDimensions(800, 4)

        znb.titleDiv:SetAnchor(TOPLEFT, znb.title, BOTTOMLEFT, -50, 5)

        znb.titleDiv:SetTexture("EsoUI\\Art\\Miscellaneous\\centerscreen_topDivider.dds")

        --pages menu

        znb.menu = wm:CreateControlFromVirtual("ZAM_NotebookMenu", znb, "ZO_ScrollContainer")

        znb.menu:SetAnchor(TOPLEFT, znb.titleDiv, BOTTOMLEFT, 70, 30)

        znb.menu:SetAnchor(BOTTOMLEFT, znb, BOTTOMLEFT, 70, -60)

        znb.menu:SetWidth(200)

        znb.menu.scrollChild = znb.menu:GetNamedChild("ScrollChild")

        --page selected highlight

        znb.selectedHL = wm:CreateControl(nil, znb.menu.scrollChild, CT_TEXTURE)

        local selectedHL = znb.selectedHL

        selectedHL:SetTexture("EsoUI\\Art\\Buttons\\generic_highlight.dds")

        selectedHL:SetHidden(true)

        --divider between menu and notebook editbox

        znb.divider = wm:CreateControl("ZAM_NotebookDivider", znb, CT_TEXTURE)

        znb.divider:SetAnchor(TOPLEFT, znb.menu, TOPRIGHT, 15, -15)

        znb.divider:SetAnchor(BOTTOMLEFT, znb.menu, BOTTOMRIGHT)

        znb.divider:SetWidth(4)

        znb.divider:SetTexture("EsoUI\\Art\\Miscellaneous\\window_edge.dds")

        --page title editbox bg

        znb.titleEditbg = wm:CreateControlFromVirtual("ZAM_NotebookTitleEditBG", znb, "ZO_EditBackdrop")

        znb.titleEditbg:SetDimensions(250, 28)

        znb.titleEditbg:SetAnchor(TOPLEFT, znb.divider, TOPRIGHT, 15, 15)

        --page title editbox

        znb.titleEdit = wm:CreateControlFromVirtual("ZAM_NotebookTitleEdit", znb.titleEditbg, "ZO_DefaultEditForBackdrop")

        znb.titleEdit:SetHandler("OnTab", function() znb.edit:TakeFocus() end)

        znb.titleEdit:SetHandler("OnEscape", znb.titleEdit.LoseFocus)

        --editbox bg

        znb.editbg = wm:CreateControlFromVirtual("ZAM_NotebookEditBG", znb, "ZO_EditBackdrop")

        znb.editbg:SetDimensions(450, 350)

        znb.editbg:SetAnchor(TOPLEFT, znb.titleEditbg, BOTTOMLEFT, 0, 15)

        --editbox

        znb.edit = wm:CreateControlFromVirtual("ZAM_NotebookEdit", znb.editbg, "ZO_DefaultEditMultiLineForBackdrop")

        znb.edit:SetMaxInputChars(3000)

        znb.edit:SetHandler("OnTab", function() if IsShiftKeyDown() then znb.titleEdit:TakeFocus() end end)

        znb.edit:SetHandler("OnEscape", znb.edit.LoseFocus)

        znb.edit:SetHandler("OnMouseWheel", function(self, delta)

                        if self:HasFocus() then --only set focus to new spots if the editbox is currently in use

                                local cursorPos = self:GetCursorPosition()

                                local text = self:GetText()

                                local textLen = text:len()

                                local newPos

                                if delta > 0 then       --scrolling up

                                        local reverseText = text:reverse()

                                        local revCursorPos = textLen - cursorPos

                                        local revPos = reverseText:find("\n", revCursorPos+1)

                                        newPos = revPos and textLen - revPos

                                else    --scrolling down

                                        newPos = text:find("\n", cursorPos+1)

                                end

                                if newPos then  --if we found a new line, then scroll, otherwise don't

                                        self:SetCursorPosition(newPos)

                                end

                        end

                end)

        --run script button

        znb.run = wm:CreateControl("ZAM_NotebookRun", znb, CT_BUTTON)

        znb.run:SetDimensions(34, 34)

        znb.run:SetAnchor(TOPLEFT, znb.editbg, BOTTOMLEFT, 0, 12)

        znb.run:SetNormalTexture("EsoUI\\Art\\Buttons\\edit_up.dds")

        znb.run:SetPressedTexture("EsoUI\\Art\\Buttons\\edit_down.dds")

        znb.run:SetMouseOverTexture("EsoUI\\Art\\Buttons\\edit_over.dds")

        znb.run.data = {tooltipText = "Run this page as a Lua script."}

        znb.run:SetHandler("OnMouseEnter", ZO_Options_OnMouseEnter)

        znb.run:SetHandler("OnMouseExit", ZO_Options_OnMouseExit)

        --setting to run at startup?

        znb.run:SetHandler("OnClicked", function()

                        local znbScript = zo_loadstring(znb.edit:GetText())

                        if znbScript then

                                znbScript()

                        end

                end)

        --undo changes button

        znb.undo = wm:CreateControl("ZAM_NotebookUndo", znb, CT_BUTTON)

        znb.undo:SetDimensions(30, 30)

        znb.undo:SetAnchor(LEFT, znb.run, RIGHT, 15, 1)

        znb.undo:SetNormalTexture("EsoUI\\Art\\Buttons\\decline_up.dds")

        znb.undo:SetPressedTexture("EsoUI\\Art\\Buttons\\decline_down.dds")

        znb.undo:SetMouseOverTexture("EsoUI\\Art\\Buttons\\decline_over.dds")

        znb.undo.data = {tooltipText = "Undo changes made to this page."}

        znb.undo:SetHandler("OnMouseEnter", ZO_Options_OnMouseEnter)

        znb.undo:SetHandler("OnMouseExit", ZO_Options_OnMouseExit)

        znb.undo:SetHandler("OnClicked", function()

                        if currentlyViewing then

                                znb.titleEdit:SetText(db.pages[currentlyViewing].title)

                                znb.edit:SetText(db.pages[currentlyViewing].text)

                        end

                end)

        --save changes button

        znb.save = wm:CreateControl("ZAM_NotebookSave", znb, CT_BUTTON)

        znb.save.highlight = wm:CreateControl("ZAM_NotebookSaveHL", znb.save, CT_TEXTURE)

        local highlight = znb.save.highlight

        znb.save:SetDimensions(30, 30)

        znb.save:SetAnchor(LEFT, znb.undo, RIGHT, 15, 1)

        znb.save:SetNormalTexture("EsoUI\\Art\\Buttons\\accept_up.dds")

        znb.save:SetPressedTexture("EsoUI\\Art\\Buttons\\accept_down.dds")

        znb.save:SetMouseOverTexture("EsoUI\\Art\\Buttons\\accept_over.dds")

        znb.save.data = {tooltipText = "Save changes made to this page."}

        znb.save:SetHandler("OnMouseEnter", ZO_Options_OnMouseEnter)

        znb.save:SetHandler("OnMouseExit", ZO_Options_OnMouseExit)

        znb.save:SetHandler("OnClicked", function(self)

                        local titleText = znb.titleEdit:GetText()

                        if titleText == "" then

                                znb.titleEdit:SetText("New Page "..#db.pages+1)

                                titleText = znb.titleEdit:GetText()

                        end

                        local pageText = znb.edit:GetText()

                        --with the changes to how text is stored in saved variables, literal strings work again.

                        --if pageText:find("%[%[") then

                        --      error([=[ZAM_Notebook does not support literal strings.

                        --

                        --              This would cause a nested literal string error and erase your saved variables data.

                        --              Please edit your page text before saving to remove the literal string. ( [[...]] )]=], 0)

                        --      return

                        --end]]

                        --so now we need to prevent escaped quotes instead...

                        --if pageText:find("%\\\"") then

                        --      error([=[ZAM_Notebook does not support escaped double-quotes.

                        --

                        --              Because ESO saved variables currently escape any double-quotes found in their contents,

                        --              this would cause a malformed table error and erase your saved variables data.

                        --

                        --              Please edit your page text before saving to remove the escaped double-quote ( \" ).

                        --              You may use a single quote instead ( \' ) or change to use a literal string. ( [[ "quote" ]] )]=], 0)

                        --      return

                        --end

                        local safe_pageText = ProtectText(pageText)

                        local safe_titleText = ProtectText(titleText)

                        if currentlyViewing == nil then --if this was a new page

                                tinsert(db.pages, {["title"] = safe_titleText, ["text"]=safe_pageText})

                                currentlyViewing = #db.pages

                                znb.selectedHL:SetHidden(false)

                                znb.selectedHL:ClearAnchors()

                                self.new = true

                        else

                                db.pages[currentlyViewing].title = safe_titleText

                                db.pages[currentlyViewing].text = safe_pageText

                                self.new = false

                        end

                        PopulateScrollList()

                        if self.new then

                                znb.selectedHL:SetAnchorFill(_G["ZAM_NotebookIndex"..currentlyViewing])

                        end

                        highlight:SetHidden(true)

                end)

        highlight:SetAnchor(TOPLEFT, znb.save, TOPLEFT, -15, -15)

        highlight:SetAnchor(BOTTOMRIGHT, znb.save, BOTTOMRIGHT, 15, 15)

        highlight:SetTexture("EsoUI\\Art\\Miscellaneous\\icon_highlight_pulse.dds")

        highlight:SetAlpha(.75)

        highlight:SetHidden(true)

        znb.titleEdit:SetHandler("OnTextChanged", function(self)

                        local page = db.pages[currentlyViewing]

                        if not page or self:GetText() ~= page.title or self:GetText() then

                                highlight:SetHidden(false)

                        else

                                highlight:SetHidden(true)

                        end

                end)

        znb.edit:SetHandler("OnTextChanged", function(self)

                        local page = db.pages[currentlyViewing]

                        if not page or self:GetText() ~= page.text or self:GetText() then

                                highlight:SetHidden(false)

                        else

                                highlight:SetHidden(true)

                        end

                end)

        --close window button

        znb.close = wm:CreateControlFromVirtual("ZAM_NotebookClose", znb, "ZO_DefaultTextButton")

        znb.close:SetText("Close")

        znb.close:SetAnchor(TOPRIGHT, znb.editbg, BOTTOMRIGHT, 0, 15)

        znb.close:SetWidth(75)

        znb.close:SetHandler("OnClicked", function() znb:SetHidden(true) end)

        --new page button

        znb.new = wm:CreateControl("ZAM_NotebookNew", znb, CT_BUTTON)

        znb.new:SetDimensions(30, 30)

        znb.new:SetAnchor(LEFT, znb.titleEditbg, RIGHT, 15, 0)

        znb.new:SetNormalTexture("EsoUI\\Art\\Buttons\\pointsplus_up.dds")

        znb.new:SetPressedTexture("EsoUI\\Art\\Buttons\\pointsplus_down.dds")

        znb.new:SetMouseOverTexture("EsoUI\\Art\\Buttons\\pointsplus_over.dds")

        znb.new.data = {tooltipText = "Create a new page."}

        znb.new:SetHandler("OnMouseEnter", ZO_Options_OnMouseEnter)

        znb.new:SetHandler("OnMouseExit", ZO_Options_OnMouseExit)

        znb.new:SetHandler("OnClicked", function()

                        currentlyViewing = nil

                        znb.titleEdit:SetText("New Page "..#db.pages+1)

                        znb.titleEdit:SelectAll()

                        znb.titleEdit:TakeFocus()

                        znb.edit:Clear()

                end)

        --delete page button

        znb.delete = wm:CreateControl("ZAM_NotebookDelete", znb, CT_BUTTON)

        znb.delete:SetDimensions(32, 32)

        znb.delete:SetAnchor(LEFT, znb.new, RIGHT, 15, 0)

        znb.delete:SetNormalTexture("EsoUI\\Art\\Buttons\\minus_up.dds")

        znb.delete:SetPressedTexture("EsoUI\\Art\\Buttons\\minus_down.dds")

        znb.delete:SetMouseOverTexture("EsoUI\\Art\\Buttons\\minus_over.dds")

        znb.delete.data = {tooltipText = "Delete this page."}

        znb.delete:SetHandler("OnMouseEnter", ZO_Options_OnMouseEnter)

        znb.delete:SetHandler("OnMouseExit", ZO_Options_OnMouseExit)

        znb.delete:SetHandler("OnClicked", function()

                        if currentlyViewing then

                                tremove(db.pages, currentlyViewing)

                                currentlyViewing = nil

                                PopulateScrollList()

                                znb.save.highlight:SetHidden(true)

                                znb.selectedHL:SetHidden(true)

                        end

                        znb.titleEdit:Clear()

                        znb.edit:Clear()

                        highlight:SetHidden(true)

                end)

        znb:SetHidden(true)

end

--global for keybind

function ZAMNotebookToggleVisibility()

        if znb:IsHidden() then

                znb:SetHidden(false)

                if not SCENE_MANAGER:IsInUIMode() then

                        SCENE_MANAGER:SetInUIMode(true)

                end

        else

                znb:SetHidden(true)

        end

end

local function Initialize()

        ZAM_NotebookDB = ZAM_NotebookDB or {}

        for k,v in pairs(defaults) do

            if type(ZAM_NotebookDB[k]) == "nil" then

                ZAM_NotebookDB[k] = v

            end

        end

        ZAM_NotebookDB.pages = ZAM_NotebookDB.pages or {}

        db = ZAM_NotebookDB

        indexPool = ZO_ObjectPool:New(CreateIndexButton, RemoveIndexButton)

        ZO_CreateStringId("SI_BINDING_NAME_ZAM_NOTEBOOK_TOGGLE", "Toggle ZAM Notebook")

        SLASH_COMMANDS["/znb"] = ZAMNotebookToggleVisibility

end

em:RegisterForEvent("ZAM_Notebook", EVENT_ADD_ON_LOADED, function(event, addon)

--em:RegisterForEvent("ZAM_Notebook", EVENT_PLAYER_ACTIVATED, function(event, addon)

                if addon == "ZAM_Notebook" then

                        Initialize()

                        CreateNotebook()

                        PopulateScrollList()

                end

        end)