Update storageGen to a more user-friendly UI

- Able to merge into current Milo configs - Moved it to the Milo package
2020-05-04 02:53:17 -04:00
parent 38b5c4a5ed
commit 39fb43d6a3
2 changed files with 143 additions and 29 deletions
--- a/milo/apps/storageGen.lua
+++ b/milo/apps/storageGen.lua
@@ -0,0 +1,143 @@
+--[[
+	For initially setting up large amounts of storage chests.
+]]
+
+local UI         = require('opus.ui')
+local Util       = require('opus.util')
+local Peripheral = require('opus.peripheral')
+
+local defaultStoragePath = "/usr/config/storage"
+
+local page = UI.Page {
+	notification = UI.Notification {},
+
+	infoText = UI.TextArea {
+		x = 2, y = 2,
+		height = 2,
+		textColor = colors.yellow,
+		value = "Select storage types to merge into your Milo storage config.",
+	},
+
+	typeGrid = UI.CheckboxGrid {
+		x = 2, y = 4,
+		ex = -2, ey = -4,
+		sortColumn = "amount",
+		inverseSort = true,
+		columns = {
+			{heading = 'Type', key = 'type'},
+			{heading = 'Amount', key = 'amount', align = 'right', width = 6},
+		},
+	},
+
+	rescanButton = UI.Button {
+		x = 2, y = -2,
+		text = 'Rescan',
+		event = 'rescan',
+	},
+
+	doneButton = UI.Button {
+		x = -7, y = -2,
+		text = 'Save',
+		event = 'save',
+	},
+
+	confirm = UI.Question {
+		x = -38, y = -2,
+		label = 'Overwrite Milo settings?',
+	},
+
+	fileSelect = UI.FileSelect {
+		modal = true,
+		enable = function() end,
+		transitionHint = 'expandUp',
+		show = function(self)
+			UI.FileSelect.enable(self)
+			self:focusFirst()
+			self:draw()
+		end,
+		disable = function(self)
+			UI.FileSelect.disable(self)
+			self.parent:focusFirst()
+			self.parent:capture(self.parent)
+		end,
+		eventHandler = function(self, event)
+			if event.type == 'select_cancel' then
+				self:disable()
+			elseif event.type == 'select_file' then
+				self:disable()
+			end
+			return UI.FileSelect.eventHandler(self, event)
+		end,
+	},
+}
+
+function page:scan()
+	_syslog("Scanned")
+
+	self.storages = Util.filter(Peripheral.getList(), function(dev)
+		return not not dev.pushItems
+	end)
+
+	local types = {}
+	Util.each(self.storages, function(dev, name)
+		if not types[dev.type] then types[dev.type] = {amount = 0, type = dev.type} end
+		types[dev.type].amount = types[dev.type].amount + 1
+	end)
+
+	self.typeGrid:setValues(types)
+	self:draw()
+	self:sync()
+end
+
+function page:saveConfig(path)
+	_syslog("Saving to "..path)
+
+	local config = Util.readTable(path) or {}
+	Util.each(self.storages, function(dev, name)
+		if self.typeGrid.values[dev.type] and self.typeGrid.values[dev.type].checked then
+			config[name] = {
+				name = name,
+				category = 'storage',
+				mtype = 'storage',
+			}
+		end
+	end)
+	Util.writeTable(path, config)
+	self.notification:success("Config saved to "..path)
+end
+
+function page:enable()
+	self:scan()
+	UI.Page.enable(self)
+end
+
+function page.typeGrid:getRowTextColor(row, selected)
+	return row.checked and colors.yellow or UI.Grid.getRowTextColor(self, row, selected)
+end
+
+function page:eventHandler(event)
+	_syslog(event)
+	if event.type == "rescan" then
+		self:scan()
+
+	elseif event.type == "save" then
+		self.confirm:show()
+
+	elseif event.type == "question_yes" then
+		self:saveConfig(defaultStoragePath)
+		self.confirm:hide()
+
+	elseif event.type == "question_no" then
+		self.confirm:hide()
+		self.fileSelect:show()
+
+	elseif event.type == "select_file" then
+		self:saveConfig(event.file)
+
+	else return UI.Page.eventHandler(self, event)
+	end
+	return true
+end
+
+UI:setPage(page)
+UI:start()