You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
78 lines
2.6 KiB
78 lines
2.6 KiB
-- Purpose: Minimal log watcher using `docker logs -f` as input.
|
|
-- Requirements: lua 5.4 + luasocket + docker CLI inside the container.
|
|
-- Notes:
|
|
-- - Pattern match is plain substring (fast & simple).
|
|
-- - Optional Wake-on-LAN is native (no external tools).
|
|
-- - Optional port-wait is provided but commented out (mirrors original bash idea).
|
|
|
|
local socket = require("socket")
|
|
|
|
-- Import modules
|
|
local config = require("config")
|
|
local utils = require("utils")
|
|
local network = require("network")
|
|
local ssh = require("ssh")
|
|
local ollama_manager = require("ollama_manager")
|
|
|
|
-- Handle error pattern detection and recovery
|
|
local function handle_error_pattern(config, powered_on)
|
|
utils.log(("Detected EHOSTUNREACH for Ollama (%s:%d)."):format(config.OLLAMA_HOST, config.OLLAMA_PORT))
|
|
|
|
-- Send Wake-on-LAN if configured
|
|
if config.WOL_MAC ~= "" then
|
|
utils.log(("Sending WOL to %s via %s:%d"):format(config.WOL_MAC, config.WOL_BCAST, config.WOL_PORT))
|
|
local ok, err = network.send_wol(config.WOL_MAC, config.WOL_BCAST, config.WOL_PORT)
|
|
if ok then
|
|
powered_on = true
|
|
utils.log(("Successfully sent WOL to %s via %s:%d"):format(config.WOL_MAC, config.WOL_BCAST, config.WOL_PORT))
|
|
else
|
|
utils.log("WOL failed: " .. tostring(err))
|
|
end
|
|
end
|
|
|
|
-- Wait for SSH and start service
|
|
utils.log("Waiting for SSH to become reachable...")
|
|
if network.port_is_up(config.OLLAMA_HOST, config.SSH_PORT, 60) then
|
|
ollama_manager.start_service(config)
|
|
end
|
|
end
|
|
|
|
-- Handle finish pattern detection and shutdown
|
|
local function handle_finish_pattern(config)
|
|
utils.log(("Detected finish pattern: %q"):format(config.FINISH_PATTERN))
|
|
ollama_manager.stop_service_and_shutdown(config)
|
|
end
|
|
|
|
-- Main application logic
|
|
local function main()
|
|
utils.log(("Watching container='%s' since='%s'"):format(config.CONTAINER_NAME, config.SINCE))
|
|
utils.log(("Looking for pattern: %q"):format(config.ERROR_PATTERN))
|
|
|
|
local cmd = ("docker logs -f --since %q %q 2>&1"):format(config.SINCE, config.CONTAINER_NAME)
|
|
local powered_on = false
|
|
|
|
while true do
|
|
local fh = assert(io.popen(cmd, "r"))
|
|
|
|
for line in fh:lines() do
|
|
-- Handle error pattern detection
|
|
if line:find(config.ERROR_PATTERN, 1, true) ~= nil then
|
|
handle_error_pattern(config, powered_on)
|
|
powered_on = true
|
|
end
|
|
|
|
-- Handle finish pattern detection
|
|
if line:find(config.FINISH_PATTERN, 1, true) ~= nil and powered_on == true then
|
|
handle_finish_pattern(config)
|
|
powered_on = false
|
|
break
|
|
end
|
|
end
|
|
|
|
fh:close()
|
|
utils.log("Restarting log watch loop...")
|
|
end
|
|
end
|
|
|
|
-- Run the application
|
|
main()
|
|
|