opus-apps/common/edit.lua

local UI = require('opus.ui')

local colors     = _G.colors
local fs         = _G.fs
local multishell = _ENV.multishell
local os         = _G.os
local shell      = _ENV.shell
local term       = _G.term
local textutils  = _G.textutils

local x, y      = 1, 1
local w, h      = term.getSize()
local scrollX   = 0
local scrollY   = 0
local lastPos   = { x = 1, y = 1 }
local tLines    = { }
local fileInfo
local actions
local lastSave
local dirty     = { y = 1, ey = h }
local mark      = { }
local searchPattern
local undo      = { chain = { }, pointer = 0 }

h = h - 1

local color = {
	textColor    = '0',
	keywordColor = '4',
	commentColor = 'd',
	stringColor  = 'e',
	statusColor  = colors.gray,
	panelColor   = colors.cyan,
}

if not term.isColor() then
	color = {
		textColor    = '0',
		keywordColor = '8',
		commentColor = '8',
		stringColor  = '8',
		statusColor  = colors.white,
		panelColor   = colors.white,
	}
end

local keyMapping = {
	-- movement
	up                        = 'up',
	down                      = 'down',
	left                      = 'left',
	right                     = 'right',
	pageUp                    = 'pageUp',
	[ 'control-b'           ] = 'pageUp',
	pageDown                  = 'pageDown',
--  [ 'control-f'           ] = 'pageDown',
	home                      = 'home',
	[ 'end'                 ] = 'toend',
	[ 'control-home'        ] = 'top',
	[ 'control-end'         ] = 'bottom',
	[ 'control-right'       ] = 'word',
	[ 'control-left'        ] = 'backword',
	[ 'scroll_up'           ] = 'scroll_up',
	[ 'control-up'          ] = 'scroll_up',
	[ 'scroll_down'         ] = 'scroll_down',
	[ 'control-down'        ] = 'scroll_down',
	[ 'mouse_click'         ] = 'go_to',
	[ 'control-g'           ] = 'goto_line',

	-- marking
	[ 'shift-up'            ] = 'mark_up',
	[ 'shift-down'          ] = 'mark_down',
	[ 'shift-left'          ] = 'mark_left',
	[ 'shift-right'         ] = 'mark_right',
	[ 'mouse_drag'          ] = 'mark_to',
	[ 'shift-mouse_click'   ] = 'mark_to',
	[ 'control-a'           ] = 'mark_all',
	[ 'control-shift-right' ] = 'mark_word',
	[ 'control-shift-left'  ] = 'mark_backword',
	[ 'shift-end'           ] = 'mark_end',
	[ 'shift-home'          ] = 'mark_home',
	[ 'mouse_down'          ] = 'mark_anchor',
	[ 'mouse_doubleclick'   ] = 'mark_current_word',

	-- editing
	delete                    = 'delete',
	backspace                 = 'backspace',
	enter                     = 'enter',
	char                      = 'char',
	paste                     = 'paste',
	tab                       = 'tab',
	[ 'control-z'           ] = 'undo',
	[ 'control-space'       ] = 'autocomplete',

	-- copy/paste
	[ 'control-x'           ] = 'cut',
	[ 'control-c'           ] = 'copy',
--	[ 'control-shift-paste' ] = 'paste_internal',

	-- file
	[ 'control-s'           ] = 'save',
	[ 'control-S'           ] = 'save_as',
	[ 'control-q'           ] = 'exit',
	[ 'control-enter'       ] = 'run',

	-- search
	[ 'control-f'           ] = 'find_prompt',
	[ 'control-slash'       ] = 'find_prompt',
	[ 'control-n'           ] = 'find_next',

	-- misc
	[ 'control-i'           ] = 'status',
	[ 'control-r'           ] = 'refresh',
}

local page = UI.Page {
	backgroundColor = color.panelColor,
	menuBar = UI.MenuBar {
		transitionHint = 'slideLeft',
		buttons = {
			{ text = 'File', dropdown = {
				{ text = 'New             ', event = 'menu_action', action = 'file_new' },
				{ text = 'Open            ', event = 'menu_action', action = 'file_open' },
				{ spacer = true },
				{ text = 'Save          ^s', event = 'menu_action', action = 'save' },
				{ text = 'Save As...    ^S', event = 'menu_action', action = 'save_as' },
				{ spacer = true },
				{ text = 'Run',              event = 'menu_action', action = 'run' },
				{ spacer = true },
				{ text = 'Quit          ^q', event = 'menu_action', action = 'exit' },
			} },
			{ text = 'Edit', dropdown = {
				{ text = 'Cut           ^x', event = 'menu_action', action = 'cut' },
				{ text = 'Copy          ^c', event = 'menu_action', action = 'copy' },
				{ text = 'Paste         ^V', event = 'menu_action', action = 'paste_internal' },
				{ spacer = true },
				{ text = 'Find...       ^f', event = 'menu_action', action = 'find_prompt' },
				{ text = 'Find Next     ^n', event = 'menu_action', action = 'find_next' },
				{ spacer = true },
				{ text = 'Go to line... ^g', event = 'menu_action', action = 'goto_line' },
				{ text = 'Mark all      ^a', event = 'menu_action', action = 'mark_all' },
			} },
			{ text = 'Code', dropdown = {
				{ text = 'Complete  ^space', event = 'menu_action', action = 'autocomplete' },
			} },
		},
		status = UI.Text {
			textColor = color.statusColor,
			x = -9, width = 9,
			align = 'right',
		},
	},
	gotoLine = UI.MiniSlideOut {
		x = -15, y = -2,
		label = 'Line',
		lineNo = UI.TextEntry {
			x = 7, width = 7,
			limit = 5,
			backgroundFocusColor = colors.gray,
			backgroundColor = colors.gray,
			transform = 'number',
			accelerators = {
				[ 'enter' ] = 'accept',
			},
		},
		show = function(self)
			self.lineNo:reset()
			UI.MiniSlideOut.show(self)
		end,
		eventHandler = function(self, event)
			if event.type == 'accept' then
				if self.lineNo.value then
					actions.process('go_to', 1, self.lineNo.value)
				end
				self:hide()
				return true
			end
			return UI.MiniSlideOut.eventHandler(self, event)
		end,
	},
	search = UI.MiniSlideOut {
		x = -20, y = -2,
		label = 'Find',
		search = UI.TextEntry {
			x = 7, width = 12,
			limit = 512,
			markBackgroundColor = colors.lightGray,
			backgroundFocusColor = colors.gray,
			backgroundColor = colors.gray,
			accelerators = {
				[ 'enter' ] = 'accept',
			},
		},
		show = function(self)
			self.search:markAll()
			UI.MiniSlideOut.show(self)
		end,
		eventHandler = function(self, event)
			if event.type == 'accept' then
				local text = self.search.value
				if text and #text > 0 then
					searchPattern = text:lower()
					if searchPattern then
						actions.unmark()
						actions.process('find', searchPattern, x)
					end
				end
				self:hide()
				return true
			end
			return UI.MiniSlideOut.eventHandler(self, event)
		end,
	},
	save_as = UI.MiniSlideOut {
		x = -24, y = -2,
		label = 'Save',
		filename = UI.TextEntry {
			x = 7, width = 16,
			limit = 512,
			markBackgroundColor = colors.lightGray,
			backgroundFocusColor = colors.gray,
			backgroundColor = colors.gray,
			accelerators = {
				[ 'enter' ] = 'accept',
			},
		},
		show = function(self)
			self.filename.value = fileInfo.abspath
			self.filename:setPosition(#self.filename.value)
			UI.MiniSlideOut.show(self)
		end,
		eventHandler = function(self, event)
			if event.type == 'accept' then
				local text = self.filename.value
				if text and #text > 0 then
					actions.save('/' .. text)
				end
				self:hide()
				return true
			end
			return UI.MiniSlideOut.eventHandler(self, event)
		end,
	},
	unsaved = UI.Question {
		x = -25, y = -2,
		label = 'Save',
		cancel = UI.Button {
			x = 16,
			text = 'Cancel',
			backgroundColor = color.panelColor,
			event = 'question_cancel',
		},
		show = function(self, action)
			self.action = action
			UI.MiniSlideOut.show(self)
		end,
		eventHandler = function(self, event)
			if event.type == 'question_yes' then
				if actions.save() then
					self:hide()
					actions.process(self.action)
				end
			elseif event.type == 'question_no' then
				actions.process(self.action, true)
				self:hide()
			elseif event.type == 'question_cancel' then
				self:hide()
			end
			return UI.MiniSlideOut.eventHandler(self, event)
		end,
	},
	file_open = UI.FileSelect {
		modal = true,
		enable = function() end,
		show = function(self)
			UI.FileSelect.enable(self, fs.getDir(fileInfo.abspath))
			self:focusFirst()
			self:draw()
			self:addTransition('expandUp', { easing = 'outBounce', ticks = 12 })
		end,
		eventHandler = function(self, event)
			if event.type == 'select_cancel' then
				self:disable()
			elseif event.type == 'select_file' then
				self:disable()
				actions.process('open', event.file)
			end
			return UI.FileSelect.eventHandler(self, event)
		end,
	},
	completions = UI.SlideOut {
		x = -12, y = 2,
		transitionHint = 'slideLeft',
		grid = UI.Grid {
			x = 2, y = 2, ey = -2,
			columns = {
				{ key = 'text', heading = 'Completion' },
			},
			accelerators = {
				[ ' ' ] = 'down',
				backspace = 'slide_hide',
			}
		},
		show = function(self, values)
			local m = 12
			for _, v in pairs(values) do
				m = #v.text > m and #v.text or m
			end
			m = m + 3
			m = m > self.parent.width and self.parent.width or m
			self.ox = -m
			self:resize()
			self.grid:setValues(values)
			self.grid:setIndex(1)
			UI.SlideOut.show(self)
		end,
		cancel = UI.Button {
			y = -1, x = -9,
			text = 'Cancel',
			backgroundColor = colors.black,
			backgroundFocusColor = colors.black,
			textColor = colors.lightGray,
			event = 'slide_hide',
		},
		eventHandler = function(self, event)
			if event.type == 'grid_select' then
				actions.process('insertText', x, y, event.selected.complete)
				self:hide()
				return true
			end
			return UI.SlideOut.eventHandler(self, event)
		end,
	},
	editor = UI.Window {
		y = 2,
		backgroundColor = colors.black,
		transitionHint = 'slideRight',
		focus = function(self)
			if self.focused then
				self:setCursorPos(x - scrollX, y - scrollY)
				self:setCursorBlink(true)
			else
				self:setCursorBlink(false)
			end
		end,
		resize = function(self)
			UI.Window.resize(self)

			w, h = self.width, self.height
			actions.setCursor(x, y)
			actions.dirty_all()
			actions.redraw()
		end,
		draw = function()
			actions.redraw()
		end,
		eventHandler = function(_, event)
			if event.ie then
				local action, param, param2
				local ie = event.ie

				if ie.code == 'char' then
					action = keyMapping.char
					param = ie.ch

				elseif ie.code == "mouse_click" or
					ie.code == 'mouse_drag' or
					--ie.code == 'mouse_up' or
					ie.code == 'shift-mouse_click' or
					ie.code == 'mouse_down' or
					ie.code == 'mouse_doubleclick' then

					action = keyMapping[ie.code]
					param = ie.x + scrollX
					param2 = ie.y + scrollY

				elseif event.type == 'paste' then
					action = keyMapping.paste
					param = event.text

				else
					action = keyMapping[ie.code]
				end

				if action then
					actions.process(action, param, param2)
					return true
				end
			end
		end,
	},
	notification = UI.Notification { },
	enable = function(self)
		UI.Page.enable(self)
		self:setFocus(self.editor)
	end,
	checkFocus = function(self)
		if not self.focused or not self.focused.enabled then
			-- if no current focus, set it to the editor
			self:setFocus(self.editor)
		end
	end,
	eventHandler = function(self, event)
		if event.type == 'menu_action' then
			actions.process(event.element.action)
			return true
		end
		return UI.Page.eventHandler(self, event)
	end,
}

local function getFileInfo(path)
	local abspath = shell.resolve(path)

	local fi = {
		abspath = abspath,
		path = path,
		isNew = not fs.exists(abspath),
		dirExists = fs.exists(fs.getDir(abspath)),
		modified = false,
	}
	if fi.isDir then
		fi.isReadOnly = true
	else
		fi.isReadOnly = fs.isReadOnly(fi.abspath)
	end

	if multishell then
		multishell.setTitle(multishell.getCurrent(), fs.getName(fi.path))
	end

	return fi
end

local function setStatus(pattern, ...)
	page.notification:info(string.format(pattern, ...))
end

local function setError(pattern, ...)
	page.notification:error(string.format(pattern, ...))
end

local function save( _sPath )
	-- Create intervening folder
	local sDir = _sPath:sub(1, _sPath:len() - fs.getName(_sPath):len() )
	if not fs.exists( sDir ) then
		fs.makeDir( sDir )
	end

	-- Save
	local file = nil
	local function innerSave()
		file = fs.open( _sPath, "w" )
		if file then
			for _,sLine in ipairs( tLines ) do
				file.write(sLine .. "\n")
			end
		else
			error( "Failed to open ".._sPath )
		end
	end

	local ok, err = pcall( innerSave )
	if file then
		file.close()
	end
	return ok, err
end

local function split(str, pattern)
	pattern = pattern or "(.-)\n"
	local t = {}
	local function helper(line) table.insert(t, line) return "" end
	helper((str:gsub(pattern, helper)))
	return t
end

local tKeywords = {
	["and"] = true,
	["break"] = true,
	["do"] = true,
	["else"] = true,
	["elseif"] = true,
	["end"] = true,
	["false"] = true,
	["for"] = true,
	["function"] = true,
	["if"] = true,
	["in"] = true,
	["local"] = true,
	["nil"] = true,
	["not"] = true,
	["or"] = true,
	["repeat"] = true,
	["return"] = true,
	["then"] = true,
	["true"] = true,
	["until"]= true,
	["while"] = true,
}

local function writeHighlighted(sLine, ny, dy)
	local buffer = {
		fg = '',
		text = '',
	}

	local function tryWrite(line, regex, fgcolor)
		local match = line:match(regex)
		if match then
			local fg = type(fgcolor) == "string" and fgcolor or fgcolor(match)
			buffer.text = buffer.text .. match
			buffer.fg = buffer.fg .. string.rep(fg, #match)
			return line:sub(#match + 1)
		end
		return nil
	end

	while #sLine > 0 do
		sLine =
			tryWrite(sLine, "^%-%-%[%[.-%]%]", color.commentColor ) or
			tryWrite(sLine, "^%-%-.*",         color.commentColor ) or
			tryWrite(sLine, "^\".-[^\\]\"",    color.stringColor  ) or
			tryWrite(sLine, "^\'.-[^\\]\'",    color.stringColor  ) or
			tryWrite(sLine, "^%[%[.-%]%]",     color.stringColor  ) or
			tryWrite(sLine, "^[%w_]+", function(match)
				if tKeywords[match] then
					return color.keywordColor
				end
				return color.textColor
			end) or
			tryWrite(sLine, "^[^%w_]", color.textColor)
	end

	buffer.fg = buffer.fg .. '7'
	buffer.text = buffer.text .. '\183'

	if mark.active and ny >= mark.y and ny <= mark.ey then
		local sx = ny == mark.y and mark.x or 1
		local ex = #buffer.text
		if ny == mark.ey then
			ex = mark.ex
		end
		buffer.bg = string.rep('f', sx - 1) ..
					string.rep('7', ex - sx) ..
					string.rep('f', #buffer.text - ex + 1)

	else
		buffer.bg = string.rep('f', #buffer.text)
	end

	page.editor:blit(1 - scrollX, dy, buffer.text, buffer.bg, buffer.fg)
end

local function redraw()
	if dirty.y > 0 then
		for dy = 1, h do

			local sLine = tLines[dy + scrollY]
			if sLine ~= nil then
				if dy + scrollY >= dirty.y and dy + scrollY <= dirty.ey then
					page.editor:clearLine(dy)
					writeHighlighted(sLine, dy + scrollY, dy)
				end
			else
				page.editor:clearLine(dy)
			end
		end
	end

	local modifiedIndicator = undo.chain[#undo.chain] == lastSave and ' ' or '*'
	page.menuBar.status.value = string.format(' %d:%d%s', y, x, modifiedIndicator)
	page.menuBar.status:draw()

	if page.editor.focused then
		page.editor:setCursorPos(x - scrollX, y - scrollY)
	end

	dirty.y, dirty.ey = 0, 0
end

local function nextWord(line, cx)
	local result = { line:find("(%w+)", cx) }
	if #result > 1 and result[2] > cx then
		return result[2] + 1
	elseif #result > 0 and result[1] == cx then
		result = { line:find("(%w+)", result[2] + 1) }
		if #result > 0 then
			return result[1]
		end
	end
end

actions = {
	undo = function()
		local last = table.remove(undo.chain)
		if last then
			undo.active = true
			actions[last.action](table.unpack(last.args))
			undo.active = false
		else
			setStatus('Already at oldest change')
		end
	end,

	addUndo = function(entry)
		local last = undo.chain[#undo.chain]
		if last and last.action == entry.action then
			if last.action == 'deleteText' then
				if last.args[3] == entry.args[1] and
					 last.args[4] == entry.args[2] then
					last.args = {
						last.args[1], last.args[2], entry.args[3], entry.args[4],
						last.args[5] .. entry.args[5]
					}
				else
					table.insert(undo.chain, entry)
				end
			else
				-- insertText (need to finish)
				table.insert(undo.chain, entry)
			end
		else
			table.insert(undo.chain, entry)
		end
	end,

	autocomplete = function()
		local sLine = tLines[y]:sub(1, x - 1):match("[a-zA-Z0-9_%.]+$")
		local results = sLine and textutils.complete(sLine, _ENV) or { }

		if #results == 0 then
			setError('No completions available')

		elseif #results == 1 then
			actions.insertText(x, y, results[1])

		elseif #results > 1 then
			local prefix = sLine:match('^.+%.(.*)$') or sLine
			for i = 1, #results do
				results[i] = {
					text = prefix .. results[i],
					complete = results[i],
				}
			end
			page.completions:show(results)
		end
	end,

	refresh = function()
		actions.dirty_all()
		mark.continue = mark.active
		setStatus('refreshed')
	end,

	goto_line = function()
		page.gotoLine:show()
	end,

	find = function(pattern, sx)
		local nLines = #tLines
		for i = 1, nLines + 1 do
			local ny = y + i - 1
			if ny > nLines then
				ny = ny - nLines
			end
			local nx = tLines[ny]:lower():find(pattern, sx, true)
			if nx then
				if ny < y or ny == y and nx <= x then
					setStatus('search hit BOTTOM, continuing at TOP')
				end
				actions.go_to(nx, ny)
				actions.mark_to(nx + #pattern, ny)
				actions.go_to(nx, ny)
				return
			end
			sx = 1
		end
		setError('Pattern not found')
	end,

	find_next = function()
		if searchPattern then
			actions.unmark()
			actions.find(searchPattern, x + 1)
		end
	end,

	find_prompt = function()
		page.search:show()
	end,

	file_open = function(force)
		if not force and undo.chain[#undo.chain] ~= lastSave then
			page.unsaved:show('file_open')
		else
			page.file_open:show('file_open')
		end
	end,

	file_new = function(force)
		if not force and undo.chain[#undo.chain] ~= lastSave then
			page.unsaved:show('file_new')
		else
			actions.open('/untitled.txt')
		end
	end,

	open = function(filename)
		if not actions.load(filename) then
			setError('Unable to load file')
		end
	end,

	load = function(path)
		if fs.exists(path) and fs.isDir(path) then
			return false
		end
		fileInfo = getFileInfo(path)

		x, y = 1, 1
		scrollX, scrollY = 0, 0
		lastPos   = { x = 1, y = 1 }
		lastSave = nil
		dirty     = { y = 1, ey = h }
		mark      = { }
		undo      = { chain = { }, pointer = 0 }

		tLines = { }
		if fs.exists(fileInfo.abspath) then
			local file = io.open(fileInfo.abspath, "r")
			local sLine = file:read()
			while sLine do
				table.insert(tLines, sLine)
				sLine = file:read()
			end
			file:close()
		end

		if #tLines == 0 then
			table.insert(tLines, '')
		end

		local name = fileInfo.path
		if fileInfo.isNew then
			if not fileInfo.dirExists then
				setStatus('"%s" [New DIRECTORY]', name)
			else
				setStatus('"%s" [New File]', name)
			end
		elseif fileInfo.isReadOnly then
			setStatus('"%s" [readonly] %dL, %dC',
						name, #tLines, fs.getSize(fileInfo.abspath))
		else
			setStatus('"%s" %dL, %dC',
						name, #tLines, fs.getSize(fileInfo.abspath))
		end

		return true
	end,

	save = function(filename)
		filename = filename or fileInfo.abspath
		if fs.isReadOnly(filename) then
			setError("Access denied")
		else
			local ok = save(filename)
			if ok then
				lastSave = undo.chain[#undo.chain]
				fileInfo = getFileInfo(filename)
				setStatus('"%s" %dL, %dC written',
					 fileInfo.path, #tLines, fs.getSize(fileInfo.abspath))
					 return true
			else
				setError("Error saving to %s", filename)
			end
		end
	end,

	save_as = function()
		page.save_as:show()
	end,

	exit = function(force)
		if not force and undo.chain[#undo.chain] ~= lastSave then
			page.unsaved:show('exit')
		else
			UI:quit()
		end
	end,

	run = function()
		--input:reset()
		local sTempPath = "/.temp"
		local ok = save(sTempPath)
		if ok then
			local nTask = shell.openTab(sTempPath)
			if nTask then
				shell.switchTab(nTask)
			else
				setError("Error starting Task")
			end
			os.sleep(0)
			fs.delete(sTempPath)
		else
			setError("Error saving to %s", sTempPath)
		end
	end,

	status = function()
		local modified = undo.chain[#undo.chain] == lastSave and '' or '[Modified] '
		setStatus('"%s" %s%d lines --%d%%--',
				 fileInfo.abspath, modified, #tLines,
				 math.floor((y - 1) / (#tLines - 1) * 100))
	end,

	dirty_line = function(dy)
		if dirty.y == 0 then
			dirty.y = dy
			dirty.ey = dy
		else
			dirty.y = math.min(dirty.y, dy)
			dirty.ey = math.max(dirty.ey, dy)
		end
	end,

	dirty_range = function(dy, dey)
		actions.dirty_line(dy)
		actions.dirty_line(dey or #tLines)
	end,

	dirty = function()
		actions.dirty_line(y)
	end,

	dirty_all = function()
		actions.dirty_line(1)
		actions.dirty_line(#tLines)
	end,

	mark_begin = function()
		actions.dirty()
		if not mark.active then
			mark.active = true
			mark.anchor = { x = x, y = y }
		end
	end,

	mark_finish = function()
		if y == mark.anchor.y then
			if x == mark.anchor.x then
				mark.active = false
			else
				mark.x = math.min(mark.anchor.x, x)
				mark.y = y
				mark.ex = math.max(mark.anchor.x, x)
				mark.ey = y
			end
		elseif y < mark.anchor.y then
			mark.x = x
			mark.y = y
			mark.ex = mark.anchor.x
			mark.ey = mark.anchor.y
		else
			mark.x = mark.anchor.x
			mark.y = mark.anchor.y
			mark.ex = x
			mark.ey = y
		end
		actions.dirty()
		mark.continue = mark.active
	end,

	unmark = function()
		if mark.active then
			actions.dirty_range(mark.y, mark.ey)
			mark.active = false
		end
	end,

	mark_anchor = function(nx, ny)
		actions.go_to(nx, ny)
		actions.unmark()
		actions.mark_begin()
		actions.mark_finish()
	end,

	mark_to = function(nx, ny)
		actions.mark_begin()
		actions.go_to(nx, ny)
		actions.mark_finish()
	end,

	mark_up = function()
		actions.mark_begin()
		actions.up()
		actions.mark_finish()
	end,

	mark_right = function()
		actions.mark_begin()
		actions.right()
		actions.mark_finish()
	end,

	mark_down = function()
		actions.mark_begin()
		actions.down()
		actions.mark_finish()
	end,

	mark_left = function()
		actions.mark_begin()
		actions.left()
		actions.mark_finish()
	end,

	mark_word = function()
		actions.mark_begin()
		actions.word()
		actions.mark_finish()
	end,

	mark_current_word = function(cx, cy)
		local index = 1
		actions.go_to(cx, cy)
		while true do
			local s, e = tLines[y]:find('%w+', index)
			if not s or s - 1 > x then
				break
			end
			if x >= s and x <= e then
				x = s
				actions.mark_begin()
				x = e + 1
				actions.mark_finish()
				x, y = cx, cy
				break
			end
			index = e + 1
		end
	end,

	mark_backword = function()
		actions.mark_begin()
		actions.backword()
		actions.mark_finish()
	end,

	mark_home = function()
		actions.mark_begin()
		actions.home()
		actions.mark_finish()
	end,

	mark_end = function()
		actions.mark_begin()
		actions.toend()
		actions.mark_finish()
	end,

	mark_all = function()
		mark.anchor = { x = 1, y = 1 }
		mark.active = true
		mark.continue = true
		mark.x = 1
		mark.y = 1
		mark.ey = #tLines
		mark.ex = #tLines[mark.ey] + 1
		actions.dirty_all()
	end,

	setCursor = function()
		lastPos.x = x
		lastPos.y = y

		local screenX = x - scrollX
		local screenY = y - scrollY

		if screenX < 1 then
			scrollX = x - 1
			actions.dirty_all()
		elseif screenX > w then
			scrollX = x - w
			actions.dirty_all()
		end

		if screenY < 1 then
			scrollY = y - 1
			actions.dirty_all()
		elseif screenY > h then
			scrollY = y - h
			actions.dirty_all()
		end
	end,

	top = function()
		actions.go_to(1, 1)
	end,

	bottom = function()
		y = #tLines
		x = #tLines[y] + 1
	end,

	up = function()
		if y > 1 then
			x = math.min(x, #tLines[y - 1] + 1)
			y = y - 1
		end
	end,

	down = function()
		if y < #tLines then
			x = math.min(x, #tLines[y + 1] + 1)
			y = y + 1
		end
	end,

	tab = function()
		if mark.active then
			actions.delete()
		end
		actions.insertText(x, y, '  ')
	end,

	pageUp = function()
		actions.go_to(x, y - h)
	end,

	pageDown = function()
		actions.go_to(x, y + h)
	end,

	home = function()
		x = 1
	end,

	toend = function()
		x = #tLines[y] + 1
	end,

	left = function()
		if x > 1 then
			x = x - 1
		elseif y > 1 then
			x = #tLines[y - 1] + 1
			y = y - 1
		else
			return false
		end
		return true
	end,

	right = function()
		if x < #tLines[y] + 1 then
			x = x + 1
		elseif y < #tLines then
			x = 1
			y = y + 1
		end
	end,

	word = function()
		local nx = nextWord(tLines[y], x)
		if nx then
			x = nx
		elseif x < #tLines[y] + 1 then
			x = #tLines[y] + 1
		elseif y < #tLines then
			x = 1
			y = y + 1
		end
	end,

	backword = function()
		if x == 1 then
			actions.left()
		else
			local sLine = tLines[y]
			local lx = 1
			while true do
				local nx = nextWord(sLine, lx)
				if not nx or nx >= x then
					break
				end
				lx = nx
			end
			if not lx then
				x = 1
			else
				x = lx
			end
		end
	end,

	insertText = function(sx, sy, text)
		x = sx
		y = sy
		local sLine = tLines[y]

		if not text:find('\n') then
			tLines[y] = sLine:sub(1, x - 1) .. text .. sLine:sub(x)
			actions.dirty_line(y)
			x = x + #text
		else
			local lines = split(text)
			local remainder = sLine:sub(x)
			tLines[y] = sLine:sub(1, x - 1) .. lines[1]
			actions.dirty_range(y, #tLines + #lines)
			x = x + #lines[1]
			for k = 2, #lines do
				y = y + 1
				table.insert(tLines, y, lines[k])
				x = #lines[k] + 1
			end
			tLines[y] = tLines[y]:sub(1, x) .. remainder
		end

		if not undo.active then
			actions.addUndo(
				{ action = 'deleteText', args = { sx, sy, x, y, text } })
		end
	end,

	deleteText = function(sx, sy, ex, ey)
		x = sx
		y = sy

		if not undo.active then
			local text = actions.copyText(sx, sy, ex, ey)
			actions.addUndo(
				{ action = 'insertText', args = { sx, sy, text } })
		end

		local front = tLines[sy]:sub(1, sx - 1)
		local back = tLines[ey]:sub(ex, #tLines[ey])
		for _ = 2, ey - sy + 1 do
			table.remove(tLines, y + 1)
		end
		tLines[y] = front .. back
		if sy ~= ey then
			actions.dirty_range(y)
		else
			actions.dirty()
		end
	end,

	copyText = function(csx, csy, cex, cey)
		local count = 0
		local lines = { }

		for cy = csy, cey do
			local line = tLines[cy]
			if line then
				local cx = 1
				local ex = #line
				if cy == csy then
					cx = csx
				end
				if cy == cey then
					ex = cex - 1
				end
				local str = line:sub(cx, ex)
				count = count + #str
				table.insert(lines, str)
			end
		end
		return table.concat(lines, '\n'), count
	end,

	delete = function()
		if mark.active then
			actions.deleteText(mark.x, mark.y, mark.ex, mark.ey)
		else
			local nLimit = #tLines[y] + 1
			if x < nLimit then
				actions.deleteText(x, y, x + 1, y)
			elseif y < #tLines then
				actions.deleteText(x, y, 1, y + 1)
			end
		end
	end,

	backspace = function()
		if mark.active then
			actions.delete()
		elseif actions.left() then
			actions.delete()
		end
	end,

	enter = function()
		local sLine = tLines[y]
		local _,spaces = sLine:find("^[ ]+")
		if not spaces then
			spaces = 0
		end
		spaces = math.min(spaces, x - 1)
		if mark.active then
			actions.delete()
		end
		actions.insertText(x, y, '\n' .. string.rep(' ', spaces))
	end,

	char = function(ch)
		if mark.active then
			actions.delete()
		end
		actions.insertText(x, y, ch)
	end,

	copy_marked = function()
		local text = actions.copyText(mark.x, mark.y, mark.ex, mark.ey)
		os.queueEvent('clipboard_copy', text)
		setStatus('shift-^v to paste')
	end,

	cut = function()
		if mark.active then
			actions.copy_marked()
			actions.delete()
		end
	end,

	copy = function()
		if mark.active then
			actions.copy_marked()
			mark.continue = true
		end
	end,

	paste = function(text)
		if mark.active then
			actions.delete()
		end
		if text then
			actions.insertText(x, y, text)
			setStatus('%d chars added', #text)
		else
			setStatus('Clipboard empty')
		end
	end,

	paste_internal = function()
		os.queueEvent('clipboard_paste')
	end,

	go_to = function(cx, cy)
		y = math.min(math.max(cy, 1), #tLines)
		x = math.min(math.max(cx, 1), #tLines[y] + 1)
	end,

	scroll_up = function()
		if scrollY > 0 then
			scrollY = scrollY - 1
			actions.dirty_all()
		end
		mark.continue = mark.active
	end,

	scroll_down = function()
		local nMaxScroll = #tLines - h
		if scrollY < nMaxScroll then
			scrollY = scrollY + 1
			actions.dirty_all()
		end
		mark.continue = mark.active
	end,

	redraw = function()
		redraw()
	end,

	process = function(action, ...)
		if not actions[action] then
			error('Invaid action: ' .. action)
		end

		local wasMarking = mark.continue
		mark.continue = false

		actions[action](...)

		if x ~= lastPos.x or y ~= lastPos.y then
			actions.setCursor()
		end
		if not mark.continue and wasMarking then
			actions.unmark()
		end

		actions.redraw()
	end,
}

local args = { ... }
if not actions.load(args[1] and args[1] or 'untitled.txt') then
	error('Error opening file')
end

UI:setPage(page)
UI:start()