koreader/commands.lua

require "keys"
require "battery"

Keydef = {
	keycode = nil,
	modifier = nil,
	descr = nil
}

function Keydef:_new(obj)
	-- obj definition
	obj = obj or {}
	setmetatable(obj, self)
	self.__index = self
	self.__tostring=Keydef.tostring
	return obj
end

function Keydef:new(keycode,modifier,descr)
	obj = Keydef:_new()
	obj.keycode = keycode
	obj.modifier = modifier
	obj.descr = descr
	return obj
end

function Keydef:display()
	return (self.modifier or "")..(self.descr or "")
end

function Keydef:tostring()
	return ((self.modifier and self.modifier.."+") or "").."["..(self.keycode or "").."]"..(self.descr or "")
end


Command = {
	keydef = nil,
	keygroup = nil,
	func = nil,
	help = nil,
	order = nil
}

function Command:_new(obj)
	-- obj definition
	obj = obj or {}
	setmetatable(obj, self)
	self.__index = self
	self.__tostring=Command.tostring
	return obj
end

function Command:new(keydef, func, help, keygroup, order)
	obj = Command:_new()
	obj.keydef = keydef
	obj.func = func
	obj.help = help
	obj.keygroup = keygroup
	obj.order = order
	--Debug("creating command: ["..tostring(keydef).."] keygroup:["..(keygroup or "").."] help:"..help)
	return obj
end

function Command:tostring()
	return tostring(self.keydef)..": "..(self.help or "<no help defined>")
end


Commands = {
	map = {},
	size = 0
}

function Commands:add(keycode,modifier,keydescr,help,func)
	if type(keycode) == "table" then
		for i=1,#keycode,1 do
			local keydef = Keydef:new(keycode[i],modifier,keydescr)
			self:_addImpl(keydef,help,func)
		end
	else
		local keydef = Keydef:new(keycode,modifier,keydescr)
		self:_addImpl(keydef,help,func)
	end
end

function Commands:addGroup(keygroup,keys,help,func)
	for _k,keydef in pairs(keys) do
		self:_addImpl(keydef,help,func,keygroup)
	end
end

--@TODO handle MOD_ANY  06.04 2012 (houqp)
function Commands:del(keycode, modifier, keydescr)
	local keydef = nil

	if not keydescr then
		for k,v in pairs(self.map) do
			if v.keydef.keycode == keycode 
			and v.keydef.modifier == modifier then
				keydef = k
				break
			end
		end -- EOF for
	else
		keydef = Keydef:new(keycode, modifier, keydescr)
	end -- EOF if

	self.map[keydef] = nil
end

function Commands:delGroup(keygroup)
	if keygroup then
		for k,v in pairs(self.map) do
			if v.keygroup == keygroup then
				self.map[k] = nil
			end
		end -- EOF for
	end
end

function Commands:_addImpl(keydef,help,func,keygroup)
	if keydef.modifier==MOD_ANY then
		self:addGroup(keygroup or keydef.descr,{Keydef:new(keydef.keycode,nil), Keydef:new(keydef.keycode,MOD_SHIFT), Keydef:new(keydef.keycode,MOD_ALT)},help,func)
	elseif keydef.modifier==MOD_SHIFT_OR_ALT then
		self:addGroup(keygroup or (MOD_SHIFT..MOD_ALT..(keydef.descr or "")),{Keydef:new(keydef.keycode,MOD_SHIFT), Keydef:new(keydef.keycode,MOD_ALT)},help,func)
	else
		local command = self.map[keydef]
		if command == nil then
			self.size = self.size + 1
			command = Command:new(keydef,func,help,keygroup,self.size)
			self.map[keydef] = command
		else
			command.func = func
			command.help = help
			command.keygroup = keygroup
		end
	end
end

function Commands:get(keycode,modifier)
	return self.map[Keydef:new(keycode, modifier)]
end

function Commands:getByKeydef(keydef)
	return self.map[keydef]
end

function Commands:new(obj)
	-- obj definition
	obj = obj or {}
	obj.map = {}
	obj.size = 0
	setmetatable(obj, self)
	self.__index = self

	-- payload
	local mt = {}
	mt.__index = function(table, key)
		return rawget(table,(key.modifier or "").."@#@"..(key.keycode or ""))
	end
	mt.__newindex = function(table, key, value)
		return rawset(table,(key.modifier or "").."@#@"..(key.keycode or ""),value)
	end
	setmetatable(obj.map, mt)

	obj:add(KEY_INTO_SCREEN_SAVER, nil, "Slider",
		"toggle screen saver",
		function()
			--os.execute("echo 'screensaver in' >> /mnt/us/event_test.txt")
			if G_charging_mode == false and G_screen_saver_mode == false then
				if G_battery_logging then logBatteryLevel("SLEEP") end
				Screen:saveCurrentBB()
				InfoMessage:inform("Going into screensaver... ", DINFO_TIMEOUT_FAST, 0, MSG_AUX)
				Screen.kpv_rotation_mode = Screen.cur_rotation_mode
				fb:setOrientation(Screen.native_rotation_mode)
				util.sleep(1)
				os.execute("killall -cont cvm")
				G_screen_saver_mode = true
			end
		end
	)
	obj:add(KEY_OUTOF_SCREEN_SAVER, nil, "Slider",
		"toggle screen saver",
		function()
			--os.execute("echo 'screensaver out' >> /mnt/us/event_test.txt")
			if G_screen_saver_mode == true and G_charging_mode == false then
				if G_battery_logging then logBatteryLevel("WAKEUP") end
				util.usleep(1500000)
				os.execute("killall -stop cvm")
				fb:setOrientation(Screen.kpv_rotation_mode)
				Screen:restoreFromSavedBB()
				fb:refresh(0)
			end
			G_screen_saver_mode = false
		end
	)
	obj:add(KEY_CHARGING, nil, "plugin/out usb",
		"toggle usb drive mode",
		function()
			--os.execute("echo 'usb in' >> /mnt/us/event_test.txt")
			if G_charging_mode == false and G_screen_saver_mode == false then
				if G_battery_logging then logBatteryLevel("USB PLUG") end
				Screen:saveCurrentBB()
				Screen.kpv_rotation_mode = Screen.cur_rotation_mode
				fb:setOrientation(Screen.native_rotation_mode)
				InfoMessage:inform("Going into USB mode... ", DINFO_TIMEOUT_FAST, 0, MSG_AUX)
				util.sleep(1)
				os.execute("killall -cont cvm")
			end
			G_charging_mode = true
		end
	)
	obj:add(KEY_NOT_CHARGING, nil, "plugin/out usb",
		"toggle usb drive mode",
		function()
			--os.execute("echo 'usb out' >> /mnt/us/event_test.txt")
			if G_charging_mode == true and G_screen_saver_mode == false then
				util.usleep(1500000)
				os.execute("killall -stop cvm")
				fb:setOrientation(Screen.kpv_rotation_mode)
				Screen:restoreFromSavedBB()
				fb:refresh(0)
				if G_battery_logging then logBatteryLevel("USB UNPLUG") end
			end
			FileChooser:setPath(FileChooser.path)
			FileChooser.pagedirty = true
			G_charging_mode = false
		end
	)
	-- Shift+P would be overwritten in inputbox by entering char 'P'
	-- I suggest one should probably change the hotkey to, say, Alt+Space
	obj:add(KEY_P, MOD_SHIFT, "P", "make screenshot",
		function()
			if G_battery_logging then logBatteryLevel("SCREENSHOT") end
			Screen:screenshot()
		end
	)
	return obj
end