git.iko.soy/opencode-android
on 0a26f8d help sign in
opencode-android/app/src/main/java/soy/iko/opencode/data/model/Todo.kt
84 lines 3.3 kB history raw
1package soy.iko.opencode.data.model
3import androidx.compose.runtime.Immutable
4import kotlinx.serialization.Serializable
5import kotlinx.serialization.builtins.ListSerializer
6import kotlinx.serialization.json.Json
7import kotlinx.serialization.json.JsonElement
9/** The tool name opencode uses to write the agent's task plan. */
10const val TODO_WRITE_TOOL = "todowrite"
12/**
13 * A single entry in an agent's task plan, as written by opencode's `todowrite` tool.
14 * Every field is defaulted and decoding ignores unknown keys (see [parseTodos]) so a
15 * change to the tool's payload shape degrades to a best-effort render instead of
16 * dropping the whole plan.
17 */
18@Immutable
19@Serializable
20data class TodoItem(
21 val id: String = "",
22 val content: String = "",
23 val status: String = "pending",
24 val priority: String? = null,
27/** Normalized lifecycle of a [TodoItem]; [UNKNOWN] covers any status the client doesn't model. */
28enum class TodoStatus { PENDING, IN_PROGRESS, COMPLETED, CANCELLED, UNKNOWN }
30fun TodoItem.statusEnum(): TodoStatus = when (status.trim().lowercase()) {
31 "pending", "todo", "" -> TodoStatus.PENDING
32 "in_progress", "in-progress", "inprogress", "running", "active" -> TodoStatus.IN_PROGRESS
33 "completed", "complete", "done" -> TodoStatus.COMPLETED
34 "cancelled", "canceled", "skipped" -> TodoStatus.CANCELLED
35 else -> TodoStatus.UNKNOWN
38@Serializable
39private data class TodoWriteInput(val todos: List<TodoItem> = emptyList())
41// A lenient decoder mirroring OpencodeJson's resilience (unknown keys ignored, nulls
42// coerced to defaults) without pulling the network layer into data/model.
43private val todoJson = Json {
44 ignoreUnknownKeys = true
45 isLenient = true
46 coerceInputValues = true
47 explicitNulls = false
50/**
51 * Extract the todo list from a `todowrite` tool call's input JSON. The tool sends the
52 * full list on every write, so a single call's input is a complete plan snapshot.
53 * Accepts either the documented `{ "todos": [...] }` object or a bare array, and returns
54 * an empty list on anything unparseable.
55 */
56fun parseTodos(input: JsonElement?): List<TodoItem> {
57 if (input == null) return emptyList()
58 return runCatching {
59 todoJson.decodeFromJsonElement(TodoWriteInput.serializer(), input).todos
60 }.recoverCatching {
61 todoJson.decodeFromJsonElement(ListSerializer(TodoItem.serializer()), input)
62 }.getOrDefault(emptyList())
65/**
66 * The current task plan for a conversation: the todos of the most recent `todowrite`
67 * tool call that carries a non-empty, parseable list. opencode rewrites the whole list
68 * on every write, so a single call is a complete snapshot; scanning newest-first and
69 * skipping empties avoids a mid-stream write (input not yet arrived) briefly blanking a
70 * plan that's still valid.
71 */
72fun currentTodoPlan(messages: List<MessageWithParts>): List<TodoItem> {
73 for (mi in messages.indices.reversed()) {
74 val parts = messages[mi].parts
75 for (pi in parts.indices.reversed()) {
76 val part = parts[pi]
77 if (part is ToolPart && part.tool.equals(TODO_WRITE_TOOL, ignoreCase = true)) {
78 val todos = parseTodos(part.state.inputElement())
79 if (todos.isNotEmpty()) return todos
80 }
81 }
82 }
83 return emptyList()