local Luan = require "luan:Luan.luan"
local error = Luan.error
local ipairs = Luan.ipairs or error()
local type = Luan.type or error()
local String = require "luan:String.luan"
local starts_with = String.starts_with or error()
local Html = require "luan:Html.luan"
local html_encode = Html.encode or error()
local Parsers = require "luan:Parsers.luan"
local json_parse = Parsers.json_parse or error()
local json_string = Parsers.json_string or error()
local Claude = require "site:/lib/ai/claude/Claude.luan"
local claude_chat = Claude.chat or error()
local Logging = require "luan:logging/Logging.luan"
local logger = Logging.logger "claude/Chat"
local Thread = require "luan:Thread.luan"


local Chat = {}

function Chat.output_messages_html(thread)
	local messages = thread and json_parse(thread) or {nil}
	for _, message in ipairs(messages) do
		local role = message.role or error()
		local who
		if role=="assistant" then
			who = "Claude"
		elseif role=="user" then
			who = "You"
		else
			error(role)
		end
		local function output(text)
			if not starts_with( text, "[INTERNAL_UPDATE]" ) then
%>
				<h3><%=who%></h3>
				<div role="<%=role%>"><%=html_encode(text)%></div>
<%
			end
		end
		local content = message.content or error()
		if type(content) == "string" then
			output(content)
		else
			for _, part in ipairs(content) do
				if part.type=="text" then
					local text = part.text or error()
					output(text)
				end
			end
		end
	end_for
end

function Chat.ask(thread,input)
	local messages = thread and json_parse(thread) or {nil}
	messages[#messages+1] = {
		role = "user"
		content = input
	}
--[=[
	messages[#messages+1] = {
		role = "assistant"
		content = [[
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
]]
	Thread.sleep(5000)
	}
	if true then
		return json_string(messages)
	end
--]=]
	local resultJson = claude_chat{
		messages = messages
	}
	local result = json_parse(resultJson)
	-- logger.info(json_string(result))
	result.type == "message" or error()
	result.role == "assistant" or error()
	result.stop_reason == "end_turn" or result.stop_reason == "tool_use" or error()
	local content = result.content or error()
	messages[#messages+1] = {
		role = "assistant"
		content = content
	}
	return json_string(messages)
end

return Chat