
    epjR                        d Z ddlZddlZddlZddlZddlZddlZddlmZm	Z	m
Z
mZmZmZmZ ddlmZ ddlmZmZmZmZmZmZmZ  ej        d          Z G d d          ZdS )	u  Full-text / trigram / CJK message search and FTS maintenance for SessionDB.

Mixin contract: this is a plain mixin class consumed by
``hermes_state.SessionDB``. It defines no ``__init__`` and no state of its
own; methods access the host's attributes (``self._conn``, ``self.db_path``,
``self._execute_write`` and other SessionDB methods) established by
``SessionDB.__init__``. It must never import hermes_state (cycle) — shared
module-level constants live in hermes_state_common.
    N)AnyCallable
CollectionDictListOptionalTuple)describe_skill_invocation)FTS_CJK_STALE_KEYFTS_SQLFTS_STORAGE_VERSIONFTS_TRIGRAM_SQLMAX_FTS5_QUERY_CHARSSCHEMA_VERSION_FTS_CJK_TRIGGERShermes_statec                      e Zd ZdZdZedeee                  dee	edf                  fd            Z
dTdZdeeeef                  fd	ZdTd
ZdefdZdefdZdeeeef                  fdZdefdZdTdZdTdZdefdZdefdZdTdZdddedefdZdTdZdefdZdefdZddddeeeeef         gdf                  dedeeef         fdZ 	 	 	 dUd#ed$ed%ed&ed'ee	edf                  deeef         fd(Z!	 	 dVd#ed*ed+ede"eeef                  fd,Z#e$d-edefd.            Z%e$d/edefd0            Z&e$d1edefd2            Z'ed1edefd3            Z(ed-edefd4            Z)e$d-edefd5            Z*d6dddd)d7d8d9ed:ed;ed+ed<e"e         d=e"e         d>e"e         d*ed?edee"eeef                           fd@Z+	 	 	 	 	 	 	 	 dWd-ed<e"e         d=e"e         d>e"e         d*ed?edAed+edeee                  de"eeef                  fdBZ,d-edefdCZ-	 	 	 	 	 	 	 	 dWd-ed<e"e         d=e"e         d>e"e         d*ed?edAed+edeee                  de"eeef                  fdDZ.dddddEdFed*ed+ed<ee"e                  d=ee"e                  d>ee"e                  de"eeef                  fdGZ/	 	 	 	 	 dXd-ed*edHedIedJe"e         d=e"e         de"eeef                  fdKZ0dLedefdMZ1defdNZ2defdOZ3ddPdQedRee         defdSZ4dS )YSessionSearchMixinu>   See module docstring — mixin for SessionDB (Search cluster).)
id
session_idrolesnippet	timestamp	tool_namesourcemodelsession_startedcontextfieldsreturn.c                 N   |dS t          |t                    rt          d          t          |                              | j                  }|r2t          dd                    t          |                               t          fd| j        D                       S )z=Validate and canonically order an optional result projection.Nz?search fields must be a collection of field names, not a stringz unknown search result field(s): , c              3   $   K   | ]
}|v |V  d S N ).0field	requesteds     9/home/thesage/.hermes/hermes-agent/hermes_state_search.py	<genexpr>z<SessionSearchMixin._search_message_fields.<locals>.<genexpr>@   s6       
 
EYDVDVEDVDVDVDV
 
    )

isinstancestr	TypeErrorset
difference_SEARCH_MESSAGE_RESULT_FIELDS
ValueErrorjoinsortedtuple)clsr   unknownr(   s      @r)   _search_message_fieldsz)SessionSearchMixin._search_message_fields3   s    
 >4fc"" 	_]^^^KK	&&s'HII 	^\		&QX//@Z@Z\\]]] 
 
 
 
"@
 
 
 
 
 	
r+   Nc                     | j         sdS 	 |                     | j                   dS # t          j        $ r&}t
                              d|           Y d}~dS d}~ww xY w)zDRun one bounded FTS5 merge pass without failing the completed write.N)	max_pagesz FTS incremental merge failed: %s)_fts_enabled_merge_fts_incrementally_FTS_MERGE_MAX_PAGES_PER_INDEXsqlite3Errorloggerwarning)selfexcs     r)   _try_incremental_merge_ftsz-SessionSearchMixin._try_incremental_merge_ftsD   s      	F	D))= *      } 	D 	D 	D NN=sCCCCCCCCC		Ds   ( AAAc                    |                                  5 }|                    dd                                          }ddd           n# 1 swxY w Y   d |D             }|                    d          }|dS t	          |                    d          pd          }t	          |          }|dk    rdS t          dt	          d|z  |z                      }d	|||d
S )a!  Return deferred-rebuild progress, or None when no rebuild pending.

        Shape: {"pending": True, "total": <rows at drop time>,
        "indexed": <rows backfilled>, "percent": <0-100 int>}.
        Consumed by search_messages() notes and by status surfaces
        (dashboard/desktop can poll this to render a progress indicator).

        Reads state_meta directly via _read_ctx instead of calling
        get_meta() (which takes self._lock) so search_messages doesn't
        block on the writer lock when checking rebuild status.
        5SELECT key, value FROM state_meta WHERE key IN (?, ?))fts_rebuild_high_waterfts_rebuild_progressNc                 ,    i | ]}|d          |d         S keyvaluer%   r&   rs     r)   
<dictcomp>z9SessionSearchMixin.fts_rebuild_status.<locals>.<dictcomp>c   "    222%!G*222r+   rG   rH   r   d   Tpendingtotalindexedpercent	_read_ctxexecutefetchallgetintminrB   connrowmeta
high_waterprogressrT   pcts           r)   fts_rebuild_statusz%SessionSearchMixin.fts_rebuild_statusR   s*    ^^ 	,,GB  hjj 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	
 32c222XX677
4txx 677<1==JA::4#s3>E12233%HQTUUU   )A

AAc                 ~    | j         fd}|                     |           t                              d           dS )u4  Finalize the deferred rebuild: boundary sweep + clear markers.

        The sweep is cheap insurance against any write that slipped through
        the migration-boundary instant (between high_water capture and
        trigger activation): re-index any row near the boundary that the
        index is missing. docsize has one row per indexed doc, so the
        anti-join is exact and runs on a narrow id range.

        The trigram half of the sweep is gated on ``self._trigram_available``
        for the same reason ``fts_rebuild_step()`` gates its backfill INSERT:
        when the SQLite build has no trigram tokenizer (or the table was
        never created), an unconditional INSERT raises ``no such table``
        and aborts the whole rebuild — taking ``optimize_fts_storage()``
        down with it.
        c                 &   |                      d                                          }|Qt          |d                   }|dz
  |dz   }}|                      d||f           r|                      d||f           |                      d           d S )NASELECT value FROM state_meta WHERE key = 'fts_rebuild_high_water'r     zINSERT INTO messages_fts(rowid, content, tool_name, tool_calls) SELECT m.id, m.content, m.tool_name, m.tool_calls FROM messages m WHERE m.id > ? AND m.id <= ? AND NOT EXISTS (SELECT 1 FROM messages_fts_docsize d WHERE d.id = m.id)a  INSERT INTO messages_fts_trigram(rowid, content, tool_name, tool_calls) SELECT m.id, m.content, m.tool_name, m.tool_calls FROM messages m WHERE m.id > ? AND m.id <= ? AND m.role <> 'tool' AND NOT EXISTS (SELECT 1 FROM messages_fts_trigram_docsize d WHERE d.id = m.id)zVDELETE FROM state_meta WHERE key IN ('fts_rebuild_high_water', 'fts_rebuild_progress')rY   fetchoner\   )r_   hw_rowhwlohiinclude_trigrams        r)   _doz3SessionSearchMixin._fts_rebuild_finish.<locals>._do   s    \\S hjj  !^^dBIB^
 H   # LLj
 R   LLE    r+   u7   Deferred FTS rebuild complete — all messages indexed.N)_trigram_available_execute_writer@   info)rB   rr   rq   s     @r)   _fts_rebuild_finishz&SessionSearchMixin._fts_rebuild_finishn   sU      1	 	 	 	 	: 	C   MNNNNNr+   c           	           j         5  d  j                            d j                            dd          dz   f                                          D             }ddd           n# 1 swxY w Y   |sdS |d          fd	}	 t                               |                    S # t          j	        $ r&}t                              d
|           Y d}~dS d}~ww xY w)a"  Tear down one chunk of a demoted v22 FTS shadow table.

        The trash tables are PLAIN tables (their vtable parent was demoted
        away during the migration), so chunked DELETE + final DROP involve
        no FTS5 machinery at all. Returns True while teardown work remains.
        c                     g | ]
}|d          S r   r%   rM   s     r)   
<listcomp>z?SessionSearchMixin._fts_teardown_trash_step.<locals>.<listcomp>   s)       !  r+   zNSELECT name FROM sqlite_master WHERE type = 'table' AND name LIKE ? ESCAPE '\'_\_%NFr   c                 P   d |                      d d          D             }|rd                    |          nd}|                      d d| d| d	 d
j         d          }|j        dk    r3|                      d            t                              d           dS )Nc                 6    g | ]}|d          dk    |d         S )   r      r%   rM   s     r)   rz   zLSessionSearchMixin._fts_teardown_trash_step.<locals>._do.<locals>.<listcomp>   s1       Q4!88 !88r+   zPRAGMA table_info()r"   rowidzDELETE FROM z WHERE (z) IN (SELECT z FROM z LIMIT r   zDROP TABLE IF EXISTS z"Old FTS shadow table %s torn down.T)rY   r3   _FTS_REBUILD_CHUNK_ROWSrowcountr@   ru   )r_   pk_colsrK   currB   tbls       r)   rr   z8SessionSearchMixin._fts_teardown_trash_step.<locals>._do   s    "ll+F+F+F+FGG  G )0<$))G$$$WC,,Rs R RC R RR R&)R R262NR R R C |q  :S::;;;@#FFF4r+   z0FTS trash teardown chunk failed (will retry): %sT)_lock_connrY   _FTS_TRASH_PREFIXreplacerZ   boolrt   r>   OperationalErrorr@   debug)rB   trashrr   rC   r   s   `   @r)   _fts_teardown_trash_stepz+SessionSearchMixin._fts_teardown_trash_step   s\    Z 	 	 "j002+33C??#EG  (**  E	 	 	 	 	 	 	 	 	 	 	 	 	 	 	  	5Ah	 	 	 	 	 	 	++C00111' 	 	 	LLKSQQQ44444	s*   AA++A/2A/	!B+ +C :CC c                    | j         sdS |                     d          }|dS t          |          | j        | j        fd}	 |                     |          }n8# t          j        $ r&}t          	                    d|           Y d}~dS d}~ww xY w|du r>| 
                                }|&|d         |d         k    r|                                  dS t          |          S )	a\  Backfill one chunk of the deferred FTS rebuild.

        Returns True when more work remains, False when the rebuild is
        complete (or none is pending). Safe to call from any process at any
        time; chunks are claimed atomically inside the write transaction, so
        concurrent callers interleave instead of duplicating rows.
        FrG   Nc                 r   |                      d                                          }|dS t          |d                   }|k    rdS t          |z             }|                      d||f           r|                      d||f           |                      dt	          |          f           |k     S )N?SELECT value FROM state_meta WHERE key = 'fts_rebuild_progress'Fr   zINSERT INTO messages_fts(rowid, content, tool_name, tool_calls) SELECT id, content, tool_name, tool_calls FROM messages WHERE id > ? AND id <= ?zINSERT INTO messages_fts_trigram(rowid, content, tool_name, tool_calls) SELECT id, content, tool_name, tool_calls FROM messages WHERE id > ? AND id <= ? AND role <> 'tool'zBUPDATE state_meta SET value = ? WHERE key = 'fts_rebuild_progress'rY   rl   r\   r]   r-   )r_   r`   rc   upperchunkrb   rq   s       r)   rr   z0SessionSearchMixin.fts_rebuild_step.<locals>._do   s     ,,Q hjj  {u3q6{{H:%%u 5(*55ELL+ 5!	    B u%   LL5U  
 :%%r+   z)FTS rebuild chunk failed (will retry): %sTrU   rT   )r;   get_metar\   rs   r   rt   r>   r   r@   r   re   rv   r   )	rB   high_water_rawrr   morerC   statusr   rb   rq   s	         @@@r)   fts_rebuild_stepz#SessionSearchMixin.fts_rebuild_step   s+      	5'?@@!5((
1,%	& %	& %	& %	& %	& %	& %	&N	&&s++DD' 	 	 	LLDcJJJ44444	 5==,,..F!fY&76'?&J&J((***5Dzzs   A! !B0BBc                    |                                  5 }|                    dd                                          }ddd           n# 1 swxY w Y   d |D             }|                    d          }|dS t	          |                    d          pd          }t	          |          }|dk    rdS t          dt	          d|z  |z                      }d	|||d
S )z:CJK-index backfill progress, or None when none is pending.rF   )fts_cjk_rebuild_high_waterfts_cjk_rebuild_progressNc                 ,    i | ]}|d          |d         S rJ   r%   rM   s     r)   rO   z=SessionSearchMixin.fts_cjk_rebuild_status.<locals>.<dictcomp>  rP   r+   r   r   r   rQ   TrR   rW   r^   s           r)   fts_cjk_rebuild_statusz)SessionSearchMixin.fts_cjk_rebuild_status  s(   ^^ 	,,GJ  hjj 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	
 32c222XX:;;
4txx :;;@qAAJA::4#s3>E12233%HQTUUUrf   c                    | j         r| j        sdS |                     d          }|dS t          |          | j        fd}	 |                     |          }n8# t          j        $ r&}t          	                    d|           Y d}~dS d}~ww xY w|du r>| 
                                }|&|d         |d         k    r|                                  dS t          |          S )	z=Backfill one chunk of the CJK index. True while work remains.Fr   Nc                 >   |                      d                                          }|dS t          |d                   }|k    rdS t          |z             }|                      d||f           |                      dt	          |          f           |k     S )NzCSELECT value FROM state_meta WHERE key = 'fts_cjk_rebuild_progress'Fr   zINSERT INTO messages_fts_cjk(rowid, content, tool_name, tool_calls) SELECT id, content, tool_name, tool_calls FROM messages WHERE id > ? AND id <= ? AND role <> 'tool'zFUPDATE state_meta SET value = ? WHERE key = 'fts_cjk_rebuild_progress'r   )r_   r`   rc   r   r   rb   s       r)   rr   z4SessionSearchMixin.fts_cjk_rebuild_step.<locals>._do*  s    ,,9  hjj  {u3q6{{H:%%u5(*55ELL> 5!	   LL9U  
 :%%r+   z-CJK FTS rebuild chunk failed (will retry): %sTrU   rT   )r;   _fts_cjk_loadedr   r\   r   rt   r>   r   r@   r   r   _fts_cjk_rebuild_finishr   )rB   r   rr   r   rC   r   r   rb   s         @@r)   fts_cjk_rebuild_stepz'SessionSearchMixin.fts_cjk_rebuild_step   s$     	(< 	5'CDD!5((
,	& 	& 	& 	& 	& 	&0	&&s++DD' 	 	 	LLH#NNN44444	 5==0022F!fY&76'?&J&J,,...5Dzzs   	A B.BBc                 x    d }|                      |           d| _        t                              d           dS )z?Boundary sweep + clear the cjk markers; index becomes servable.c                     |                      d                                          }|7t          |d                   }|dz
  |dz   }}|                      d||f           |                      d           d S )NzESELECT value FROM state_meta WHERE key = 'fts_cjk_rebuild_high_water'r   rj   a  INSERT INTO messages_fts_cjk(rowid, content, tool_name, tool_calls) SELECT m.id, m.content, m.tool_name, m.tool_calls FROM messages m WHERE m.id > ? AND m.id <= ? AND m.role <> 'tool' AND NOT EXISTS (SELECT 1 FROM messages_fts_cjk_docsize d WHERE d.id = m.id)z^DELETE FROM state_meta WHERE key IN ('fts_cjk_rebuild_high_water', 'fts_cjk_rebuild_progress')rk   )r_   rm   rn   ro   rp   s        r)   rr   z7SessionSearchMixin._fts_cjk_rebuild_finish.<locals>._doP  s    \\;  hjj  !^^dBIBb
 H   LLM    r+   Tu7   CJK FTS index backfill complete — serving CJK search.N)rt   _fts_cjk_availabler@   ru   rB   rr   s     r)   r   z*SessionSearchMixin._fts_cjk_rebuild_finishN  sH    	 	 	( 	C   "&MNNNNNr+   c                     | j         sdS d }|                     |          }|rU| j        5  |                     | j                   | j                                         ddd           dS # 1 swxY w Y   dS dS )a  Rebuild path for a stale cjk index (triggers were dropped).

        The gap's extent is unknown, so the only safe recovery is a from-
        scratch rebuild: drop the table + triggers, clear the breadcrumb,
        recreate via ``_ensure_fts_cjk_schema`` (which sets fresh backfill
        markers on a populated DB). Called from ``optimize_fts_storage`` on
        a tokenizer-capable host; no-op when not stale.
        Nc                 >   |                      dt          f                                          }|sdS t          D ]}|                      d|            |                      d           |                      d           |                      dt           d           dS )	Nz&SELECT 1 FROM state_meta WHERE key = ?FzDROP TRIGGER IF EXISTS z%DROP TABLE IF EXISTS messages_fts_cjkz(DROP VIEW IF EXISTS messages_fts_cjk_srcz&DELETE FROM state_meta WHERE key IN ('z<', 'fts_cjk_rebuild_high_water', 'fts_cjk_rebuild_progress')T)rY   r   rl   r   )r_   staletrigs      r)   rr   z7SessionSearchMixin._fts_cjk_reset_if_stale.<locals>._dot  s    LL8"$  hjj   u) ? ?=t==>>>>LL@AAALLCDDDLL.&. . .  
 4r+   )r   rt   r   _ensure_fts_cjk_schemar   commit)rB   rr   	was_stales      r)   _fts_cjk_reset_if_stalez*SessionSearchMixin._fts_cjk_reset_if_staleh  s     # 	F	 	 	" '',,	 	$
  $ $++DJ777
!!###$ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $	$ 	$s   4A,,A03A0c                     	 |                     d                                          d         }|sdS |                     d                                          d         }| S # t          j        $ r Y dS w xY w)a  True when the base FTS table exists but indexes nothing while
        ``messages`` has rows. Caller must hold ``self._lock``.

        This is the post-demote empty-index shape: external-content FTS with
        zero ``messages_fts_docsize`` rows against a non-empty messages table.
        Healthy installs (and mid-backfill installs that still hold markers)
        never match.
        z%SELECT EXISTS(SELECT 1 FROM messages)r   Fz1SELECT EXISTS(SELECT 1 FROM messages_fts_docsize))rY   rl   r>   r   )rB   r_   has_msghas_ftss       r)   '_fts_external_index_empty_with_messagesz:SessionSearchMixin._fts_external_index_empty_with_messages  s    	ll7 hjjG  u llC hjjG ;' 	 	 	55	s   /A# /A# #A65A6c                     	 |                     d                                          d         }t          |          dk    S # t          j        $ r Y dS w xY w)zTrue when the base external-content index holds no rows.

        A missing table counts as empty: the schema ensure that follows
        creates it fresh.
        z)SELECT COUNT(*) FROM messages_fts_docsizer   T)rY   rl   r\   r>   r   )rB   r_   ns      r)   _fts_index_known_emptyz)SessionSearchMixin._fts_index_known_empty  sd    	; hjjA q66Q;' 	 	 	44	s   ?A AAc                 t    dD ]4}	 |                     d| d| d            # t          j        $ r Y 1w xY wdS )u&  Delete every indexed row from the v23 external-content tables.

        Uses the FTS5 ``'delete-all'`` special command — the documented O(1)
        truncate for external-content tables. A plain no-WHERE ``DELETE`` is
        O(rows) on external-content FTS5 (each row's delete tokens are
        regenerated from the content table; measured ~12µs/row, minutes on a
        large index, while holding the write lock) and corrupts the index if
        indexed rows have diverged from ``messages`` — precisely the broken-
        bookkeeping shape this repair path handles. The backfill chunk worker
        replays its whole selected id range with no anti-join, so a replay
        from zero is only safe once the index is known empty — this is how a
        partially indexed DB gets there.
        )messages_ftsmessages_fts_trigramINSERT INTO (z) VALUES('delete-all')N)rY   r>   r   )rB   r_   r   s      r)   _reset_fts_index_to_emptyz,SessionSearchMixin._reset_fts_index_to_empty  sn     < 	 	CMCMM#MMMNNNN+   	 	s   #55Fforcer   c                0   |                     d                                          }||st          |d                   }|                     d                                          }|?|                     |          s|                     |           |                     d           |S |                     d                                          d         }dt          |          fdfD ]\  }}|                     d	||f           t          |          S )
ah  Write ``fts_rebuild_high_water`` / ``fts_rebuild_progress`` for a
        full backfill. Returns the high-water id.

        When ``force`` is False and high_water is already set, only repairs a
        missing progress key (stuck no-op when high_water exists alone), and
        only after the index is known empty: the chunk worker replays its
        whole selected id range without an anti-join, so a partially indexed
        DB is first reset to a known-empty surface rather than rebuilt from
        zero on top of surviving rows. Caller must hold the write
        transaction / lock as appropriate.
        ri   Nr   r   z~INSERT INTO state_meta (key, value) VALUES ('fts_rebuild_progress', '0') ON CONFLICT(key) DO UPDATE SET value = excluded.valuez)SELECT COALESCE(MAX(id), 0) FROM messagesrG   )rH   0zgINSERT INTO state_meta (key, value) VALUES (?, ?) ON CONFLICT(key) DO UPDATE SET value = excluded.value)rY   rl   r\   r   r   r-   )rB   r_   r   existing_hwrn   rc   kvs           r)   _seed_fts_rebuild_markersz,SessionSearchMixin._seed_fts_rebuild_markers  sB    llO
 

(** 	 "5"[^$$B||Q hjj   22488 9224888L  
 I\\7
 

(**Q &s2ww/)
 	 	DAq LLHA   
 2wwr+   c                 <      fd}                      |           dS )u  Heal interrupted demote/backfill bookkeeping before optimize runs.

        Covers two post-#65798 failure classes:

        1. Empty external-content index with messages present and no rebuild
           markers (demote crash window after empty v23 tables landed but
           before markers, or settle that stamped without backfill). Seed a
           full backfill.
        2. ``fts_rebuild_high_water`` present without ``fts_rebuild_progress``
           (partial meta) — seed progress so the chunk loop is not a no-op,
           resetting a partially populated index to a known-empty surface
           first so the anti-join-free chunk replay cannot duplicate rows.

        Must not invent markers on a still-legacy inline DB: that would make
        ``optimize_fts_storage`` skip demote (``legacy and not pending``) and
        attempt v23-shaped INSERTs against the inline table forever.
        c                    |                      d                                          }|j|                      d                                          }|?                    |           s                    |            |                      d           d S                     |           rd S                     |           r.|                      d                               | d           d S d S )Nri   z;SELECT 1 FROM state_meta WHERE key = 'fts_rebuild_progress'zsINSERT INTO state_meta (key, value) VALUES ('fts_rebuild_progress', '0') ON CONFLICT(key) DO UPDATE SET value = '0'z8DELETE FROM state_meta WHERE key = 'fts_storage_version'Tr   )rY   rl   r   r   _db_has_legacy_inline_ftsr   r   )r_   r   rc   rB   s      r)   rr   z<SessionSearchMixin._repair_optimize_bookkeeping.<locals>._do  s*   ,,7  hjj 
 &  <<9  (**  #66t<< =66t<<<LLE  
  --d33  ;;DAA AN   ..t4.@@@@@	A Ar+   N)rt   r   s   ` r)   _repair_optimize_bookkeepingz/SessionSearchMixin._repair_optimize_bookkeeping  s>    $!	A !	A !	A !	A !	AD 	C     r+   c                 @   | j         r| j        rdS | j        5  |                     | j                  r	 ddd           dS | j                            d                                          r	 ddd           dS | j        rC| j                            dt           d                                          r	 ddd           dS | 	                    | j                  r	 ddd           dS | 
                    | j                  cddd           S # 1 swxY w Y   dS )az  True when `optimize_fts_storage()` has work to do: either this DB
        is a legacy inline-FTS install that can be optimized to the v23
        external-content schema, or a previous optimize run was interrupted
        (legacy vtables already demoted, but backfill markers and/or trash
        tables remain) and re-running would resume it, or the CJK-bigram
        index needs a backfill/rebuild on this tokenizer-capable host, or
        a prior demote left an empty external-content index without markers
        (healable on re-run).
        False for fresh and fully-optimized installs (and when FTS5 is
        unavailable).FNTESELECT 1 FROM state_meta WHERE key = 'fts_rebuild_high_water' LIMIT 1zFSELECT 1 FROM state_meta WHERE key IN ('fts_cjk_rebuild_high_water', 'z
') LIMIT 1)r;   	read_onlyr   r   r   rY   rl   r   r   _has_fts_trashr   )rB   s    r)   fts_optimize_availablez)SessionSearchMixin.fts_optimize_available3  s,      	DN 	5Z 	L 	L--dj99 	L 	L 	L 	L 	L 	L 	L 	L z!!?  hjj 	L 	L 	L 	L 	L 	L 	L 	L  # 
(:(:Q3DQ Q Q) ) hjj )	L 	L 	L 	L 	L 	L 	L 	L* ""4:.. -	L 	L 	L 	L 	L 	L 	L 	L4 ??
KK5	L 	L 	L 	L 	L 	L 	L 	L 	L 	L 	L 	L 	L 	L 	L 	L 	L 	Ls)   D-D;=DD-DDDc                      fd}t                               |                    } j        5                        j        dt
                    }                      j        dt                    }t          |           _        |st          j
        d           j                                         ddd           n# 1 swxY w Y   |S )u  Demote the legacy inline FTS vtables and stage their shadow tables
        for chunked teardown. Returns MAX(messages.id) as the rebuild high
        water. O(1) schema surgery — the heavy delete is deferred to the
        chunked teardown, exactly as the validated auto path did.

        Markers are written in the same BEGIN IMMEDIATE as the demote, *before*
        the empty v23 schema is created. Schema creation uses
        ``executescript`` and therefore cannot run inside that transaction
        (it issues an implicit COMMIT — see the CJK recreate path). Creating
        the empty schema only after markers are durable closes the crash
        window where trash + empty v23 tables exist with no backfill claim.
        c                 @                        |            |                     d           t          |                     d                                                    }|r|                     d           |                     d           |                     d           d |                     d                                          D             }|D ]}|                     d| d	|                                | d
          }|                     d           |S )Nz,DROP VIEW IF EXISTS messages_fts_trigram_srczSELECT 1 FROM sqlite_master WHERE type = 'table' AND name IN ('messages_fts', 'messages_fts_trigram') AND sql LIKE 'CREATE VIRTUAL TABLE%' LIMIT 1zPRAGMA writable_schema=ONzDELETE FROM sqlite_master WHERE type = 'table' AND name IN ('messages_fts', 'messages_fts_trigram') AND sql LIKE 'CREATE VIRTUAL TABLE%'zPRAGMA writable_schema=RESETc                     g | ]
}|d          S ry   r%   rM   s     r)   rz   zRSessionSearchMixin._demote_legacy_fts_to_trash.<locals>._stage.<locals>.<listcomp>y  s)       AaD  r+   zSELECT name FROM sqlite_master WHERE type = 'table' AND (name LIKE 'messages_fts_%' ESCAPE '\' OR name LIKE 'messages_fts_trigram_%' ESCAPE '\')zALTER TABLE z RENAME TO fts_v22_trash_Tr   ;DELETE FROM state_meta WHERE key = 'fts_optimize_available')_drop_fts_triggersrY   r   rl   rZ   r   )r_   hadshadowsshrn   rB   s        r)   _stagez>SessionSearchMixin._demote_legacy_fts_to_trash.<locals>._stagei  sW   ##D)))LLGHHHt||?  hjj	 C
  S8999;  
 ;<<< "&,,M# # hjj   " S SBLL!Q!Q!QR!Q!QRRRR
 //D/AABLLM   Ir+   r   r   z@failed to create v23 messages_fts during optimize-storage demoteN)r\   rt   r   _ensure_fts_schemar   r   r   r   rs   r>   r   r   )rB   r   rn   base_ok
trigram_oks   `    r)   _demote_legacy_fts_to_trashz.SessionSearchMixin._demote_legacy_fts_to_trash\  s    !	 !	 !	 !	 !	F $$V,,-- Z 
	  
	 --dj.'RRG00
2O J '+:&6&6D# .V   J
	  
	  
	  
	  
	  
	  
	  
	  
	  
	  
	  
	  
	  
	  
	  	s   BCC	CT)progress_cbvacuumr   r   c                	     j         sdddS  j        rdddS                                    j        5                        j                  }ddd           n# 1 swxY w Y                        d          du}|r|s                                  n|r|s j        5                        j        dt                    }                      j        dt                    }t          |           _        |st          j        d	           j                                         ddd           n# 1 swxY w Y                                      j        rR j        5                        j                    j                                         ddd           n# 1 swxY w Y   d
t&          ddf fd}dt(          ddf fd} |d           	 t+          j                    }	                                 sn+ |d            |t+          j                    |	z
             S |d           	 t+          j                    }	                                 sn+ |d            |t+          j                    |	z
             S |d           	 t+          j                    }	                                 sn+ |d            |t+          j                    |	z
             S j        5   j                            d                                          du}
                      j                  }                      j                  }ddd           n# 1 swxY w Y   |
s|s|r,|
s|rdnd}t<                              d||
||           d|ddS d}|r |d           	  j        5   j                            d           ddd           n# 1 swxY w Y   d}n9# t          j        $ r'}t<                              d|           d}Y d}~nd}~ww xY w	  j        5   j                            d           ddd           n# 1 swxY w Y   n2# t@          $ r%}t<          !                    d|           Y d}~nd}~ww xY w fd} "                    |          }|!t<                              d|           d||dS  |d           t<          #                    dtH                     d|d S )!u  Migrate a legacy v22 inline-FTS DB to the v23 external-content
        schema, foreground and to completion. Safe to re-run: if a previous
        attempt was interrupted it resumes from the progress marker.

        ``progress_cb`` receives {"phase", "percent", "indexed", "total"}
        dicts for a CLI progress bar. Returns a summary dict.

        The trigram tokenizer being unavailable is not fatal — the base index
        is still rebuilt (CJK falls back to LIKE), mirroring normal startup.
        Ffts5_unavailable)okreasonr   NrG   r   r   z?failed to re-create v23 messages_fts on optimize-storage resumephaser    c                     d S                                  }|                                } | |r|d         nd|r|d         nd|r|d         ndd           d S )NrV   rQ   rU   r   rT   )r   rV   rU   rT   )re   r   )r   str   rB   s     r)   _emitz6SessionSearchMixin.optimize_fts_storage.<locals>._emit  s    "((**Bz0022K,.72i==C,.52i==A(*1G	      r+   chunk_secondsc                 f    t          j        t          j        | j        z                       dS )a  Inter-chunk throttle (see the chunk-engine note above).

            The chunk methods themselves never sleep, so this loop is the
            single place the duty cycle is enforced: without it, back-to-back
            BEGIN IMMEDIATE chunks starve any live gateway/CLI process
            sharing the DB out of its lock retries (the measured ~85%
            write-lock ownership that froze concurrent sessions).
            N)timesleepmax_FTS_REBUILD_MIN_PAUSE_FTS_REBUILD_DUTY_FACTOR)r   rB   s    r)   _pausez7SessionSearchMixin.optimize_fts_storage.<locals>._pause  s@     Js+ ==      r+   backfillTteardownr   backfill_incompleteteardown_incompletezPFTS storage optimization did not settle (%s): pending=%s trash=%s empty_index=%s)r   r   vacuumedr   VACUUMz$VACUUM after FTS optimize failed: %szPRAGMA wal_checkpoint(TRUNCATE)z:WAL checkpoint (TRUNCATE) after optimize VACUUM failed: %sc                 v   |                      d                                          dS                     |           rdS                     |           rdS |                      dt	          t
                    f           |                      d           |                      dt          t          f           d S )Nr   r   r   z{INSERT INTO state_meta (key, value) VALUES ('fts_storage_version', ?) ON CONFLICT(key) DO UPDATE SET value = excluded.valuer   z7UPDATE schema_version SET version = ? WHERE version < ?)rY   rl   r   r   r-   r   r   )r_   rB   s    r)   _settlez8SessionSearchMixin.optimize_fts_storage.<locals>._settleW  s     ||?  hjj% -,""4(( -,,;;DAA -,,LLH())+  
 LLVWWWLLI0   4r+   z,FTS storage optimization settle refused (%s)donez/FTS storage optimization complete (layout v%d).)r   r   )%r;   r   r   r   r   r   r   r   r   r   r   r   rs   r>   r   r   r   r   r   r-   floatr   	monotonicr   r   r   rY   rl   r   r   r@   rA   	Exceptionr   rt   ru   r   )rB   r   r   legacyrS   r   r   r   r   _t0still_pendingstill_trashempty_indexr   	vacuum_okrC   r   refusals   ``                r)   optimize_fts_storagez'SessionSearchMixin.optimize_fts_storage  s       	?+=>>>> 	8;777 	))+++ Z 	@ 	@33DJ??F	@ 	@ 	@ 	@ 	@ 	@ 	@ 	@ 	@ 	@ 	@ 	@ 	@ 	@ 	@-- 899E 	$' 	$,,.... 	$V 	$  $ $11J  "44J 6 
 +/z*:*:'  "25   
!!###$ $ $ $ $ $ $ $ $ $ $ $ $ $ $( 	$$&&&  	$ $ $++DJ777
!!###$ $ $ $ $ $ $ $ $ $ $ $ $ $ $	 	 	 	 	 	 	 	 		% 	D 	 	 	 	 	 	  	j	+.""C((** E*F4>##c)***	+ 	j	+.""C,,.. E*F4>##c)***	+ 	j	+.""C0022 E*F4>##c)***	+ Z 	S 	S J..?  hjj%M --dj99KFFtzRRK	S 	S 	S 	S 	S 	S 	S 	S 	S 	S 	S 	S 	S 	S 	S  
	EK 
	E; 
	E)6 ++ +%%*  NN5{K  
  6tDDD 	 	E(OOO	"Z 1 1J&&x0001 1 1 1 1 1 1 1 1 1 1 1 1 1 1 		+ " " " EsKKK!						"Z J JJ&&'HIIIJ J J J J J J J J J J J J J J   P       	 	 	 	 	0 %%g..
 NN>    7	JJJf=?R	
 	
 	
 	222s   AA!$A!#BD55D9<D9"4F""F&)F&0A#MM#&M#,O( 3OO( OO( !O"O( (P7PP"Q )QQ QQ QQ 
R&RRr      user	assistantr   around_message_idwindowbookend
keep_rolesc                 f    |dk     rd}                      ||          }|d         }|sg ddg g dS |t          |          fd|D             }n|}|d         d         }	|d         d         }
g }g }|dk    r                                 5 }d	}g }|4d
                    d |D                       }d| d}t	          |          }|                    d| d||	g||R                                           }|                    d| d||
g||R                                           }t	          t          |                    }ddd           n# 1 swxY w Y   dt          t          t          f         f fd||d         |d         fd|D             fd|D             dS )u)  Return an anchored window plus session bookends.

        Built on top of ``get_messages_around``. Three slices:

          - ``window``: messages immediately surrounding the anchor. Filtered
            to ``keep_roles`` (tool-response noise dropped by default), EXCEPT
            the anchor itself is always preserved regardless of role.
          - ``bookend_start``: first ``bookend`` user/assistant messages of the
            session — but only those whose id is strictly before the window's
            first message id. Empty when the window already overlaps the
            session head. Empty-content messages (tool-call-only assistant
            turns) are skipped so they don't crowd out actual prose openings.
          - ``bookend_end``: last ``bookend`` user/assistant messages of the
            session, same non-overlap rule at the tail.

        Bookends let an FTS5 hit anywhere in a long session yield the goal
        (opening) and the resolution (closing) on a single call — without
        loading the whole transcript.

        Returns ``{"window": [], "messages_before": 0, "messages_after": 0,
        "bookend_start": [], "bookend_end": []}`` when the anchor isn't in
        the session.

        ``keep_roles=None`` disables role filtering (raw window + raw
        bookends).
        r   )r	  r	  )r	  messages_beforemessages_afterbookend_startbookend_endNc                 t    g | ]4}|                     d           k    s|                     d          v 2|5S )r   r   )r[   )r&   mr  keep_sets     r)   rz   z8SessionSearchMixin.get_anchored_view.<locals>.<listcomp>  sM       55;;"333quuV}}7P7P 7P7P7Pr+   r    ,c              3      K   | ]}d V  dS ?Nr%   r&   r{   s     r)   r*   z7SessionSearchMixin.get_anchored_view.<locals>.<genexpr>  s"      0I0I0I0I0I0I0I0Ir+   z AND role IN (r   z6SELECT * FROM messages WHERE session_id = ? AND id < ?z0 AND length(content) > 0 ORDER BY id ASC LIMIT ?z6SELECT * FROM messages WHERE session_id = ? AND id > ?z1 AND length(content) > 0 ORDER BY id DESC LIMIT ?r    c                    t          |           }d|v r                    |d                   |d<   |                    d          rZ	 t          j        |d                   |d<   n;# t          j        t          f$ r" t                              d           g |d<   Y nw xY w|                    d          	                    |d                   |d<   |S )Ncontent
tool_callszIFailed to deserialize tool_calls in get_anchored_view, falling back to []display_metadata)
dict_decode_contentr[   jsonloadsJSONDecodeErrorr.   r@   rA   _decode_display_metadata)r`   msgrB   s     r)   _hydratez6SessionSearchMixin.get_anchored_view.<locals>._hydrate  s    s))CC!%!5!5c)n!E!EIww|$$ ++(,
3|3D(E(EC%%,i8 + + +NNc   )+C%%%	+
 ww)**6*.*G*GL^H_*`*`&'Js   	A' '5BBr  r  c                 &    g | ]} |          S r%   r%   r&   rN   r&  s     r)   rz   z8SessionSearchMixin.get_anchored_view.<locals>.<listcomp>  s!    FFFahhqkkFFFr+   c                 &    g | ]} |          S r%   r%   r(  s     r)   rz   z8SessionSearchMixin.get_anchored_view.<locals>.<listcomp>  s!    BBBAHHQKKBBBr+   )get_messages_aroundr/   rX   r3   listrY   rZ   reversedr   r-   r   )rB   r   r  r	  r
  r  	primitivewindow_rowsfiltered_windowwindow_min_idwindow_max_idbookend_start_rowsbookend_end_rowsr_   role_clauserole_paramsrole_placeholdersr&  r  s   ` `              @@r)   get_anchored_viewz$SessionSearchMixin.get_anchored_view  s   D Q;;G ,,)& - 
 
	  ) 	#$"#!#!   !:H    &  OO
 *O#At,#B- )+&(Q;;!! DT $&)(+0I0Ij0I0I0I(I(I%"G3D"G"G"GK"&z"2"2K%)\\/6A/ / /  FFgFF& & (** # $(<<06A0 0 0  FFgFF$ $ (** ! $(1A(B(B#C#C 1D D D D D D D D D D D D D D D4	T#s(^ 	 	 	 	 	 	" &():;'(89FFFF3EFFFBBBB1ABBB
 
 	
s   B=EE!$E!   limitinclude_inactivec                    |rdnd}d}| j         5  | j                            d| | d|t          |          f          }|                                }ddd           n# 1 swxY w Y   g }|D ]
}	|                     |	d                   }
t          |
t                    rBd |
D             }d	                    d
 |D                       	                                }|sd}n)t          |
t                    rt          |
          p|
}nd}d	                    |                                          }t          |          dk    r|dd         dz   }|                    |	d         |	d         |d           |S )u  Return the *limit* most-recent user messages, newest first.

        Each entry is a dict with keys ``id``, ``timestamp``, ``preview``.
        ``preview`` is the first 80 characters of the message content
        (with line breaks collapsed to spaces). Used by the /rewind
        slash command picker, CLI/TUI/gateway ``/undo [N]``, and any other
        caller that needs real user-turn targets.

        Bookkeeping timeline rows (``display_kind`` set — e.g. model_switch,
        async_delegation_complete, auto_continue, hidden) are excluded. They
        are durable ``role='user'`` rows for the API transcript, but no client
        counts them as user turns (desktop demotes them to system / drops them;
        the CLI already uses ``not m.get("display_kind")``). Including them here
        made ``/undo`` soft-delete from a marker instead of the last real turn —
        same class of index skew as the prompt.submit ordinal bug.

        By default only active messages are returned.
        r  z AND active = 1z0 AND (display_kind IS NULL OR display_kind = '')zRSELECT id, timestamp, content FROM messages WHERE session_id = ? AND role = 'user'z ORDER BY id DESC LIMIT ?Nr  c                     g | ]F}t          |t                    r/|                    d           dk    0|                    dd          GS typetextr  r,   r  r[   r&   ps     r)   rz   z@SessionSearchMixin.list_recent_user_messages.<locals>.<listcomp>#  sW       *+!!T**/0uuV}}/F/F EE&"%%/F/F/Fr+    c              3      K   | ]}||V  	d S r$   r%   r&   ts     r)   r*   z?SessionSearchMixin.list_recent_user_messages.<locals>.<genexpr>'  s'      ">">A">1">">">">">">r+   [multimodal content]P   M   z...r   r   )r   r   preview)r   r   rY   r\   rZ   r   r,   r+  r3   stripr-   r
   splitlenappend)rB   r   r9  r:  active_clausedisplay_clausecursorrowsresultr`   decoded
text_partsrJ  s                r)   list_recent_user_messagesz,SessionSearchMixin.list_recent_user_messages  s   0 /E4EKZ 	% 	%Z''+ +"0+ + + SZZ( F ??$$D	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% (* 	 	C**3y>::G'4((  /6  
 ((">">j">">">>>DDFF 54GGS))  4G<<Ghhw}}//G7||b  !#2#,.MMd)!$[!1&     s   AA!!A%(A%queryc                    | dt                    } g }g }d}|t          |           k     r| |         }|dk    r|                    |           |dz  }<|                     d|dz             }|dk    r|                    d           |dz  }v|                    | ||dz                       |                    dt          |          dz
   d           |dz   }|t          |           k     d	                    |          }t          j        d
d|          }t          j        dd|          }t          j        dd|          }t          j        dd	|                                          }t          j        dd	|                                          }t          j        dd|          }t          |          D ]\  }}|	                    d| d|          } |                                S )a  Sanitize user input for safe use in FTS5 MATCH queries.

        FTS5 has its own query syntax where characters like ``"``, ``(``, ``)``,
        ``+``, ``*``, ``{``, ``}``, the column-filter operator ``:`` and bare
        boolean operators (``AND``, ``OR``, ``NOT``) have special meaning.
        Passing raw user input directly to MATCH can cause
        ``sqlite3.OperationalError``.

        Strategy:
        - Preserve properly paired quoted phrases (``"exact phrase"``)
        - Strip unmatched FTS5-special characters that would cause errors
        - Wrap unquoted hyphenated and dotted terms in quotes so FTS5
          matches them as exact phrases instead of splitting on the
          hyphen/dot (e.g. ``chat-send``, ``P2.2``, ``my-app.config.ts``)
        Nr   "r   r  rC  z Q r  z[+{}():\"^]z\*+*z(^|\s)\*z\1z(?i)^(AND|OR|NOT)\b\s*z(?i)\s+(AND|OR|NOT)\s*$z\b(\w+(?:[._-]\w+)+)\bz"\1")
r   rM  rN  findr3   resubrK  	enumerater   )rW  _quoted_partspiecesichend	sanitizedquoteds           r)   _sanitize_fts5_queryz'SessionSearchMixin._sanitize_fts5_query<  s   ( +++, !#e**nnqBSyyb!!!Q**S!a%((Cbyy c"""Q  qqy!1222MM>#m"4"4q"8>>>???aA #e**nn" GGFOO	 F>3	::	 F63	22	F;y99	 F4b)//:K:KLL	F5r9??;L;LMM	 F4gyII	 #=11 	C 	CIAv!))/!///6BBII   r+   cpc                     d| cxk    odk    nc p_d| cxk    odk    nc pOd| cxk    odk    nc p?d| cxk    odk    nc p/d	| cxk    od
k    nc pd| cxk    odk    nc pd| cxk    odk    nc S )N N     4  M     ߦ  0  ?0  @0  0  0  0       r%   )rh  s    r)   _is_cjk_codepointz$SessionSearchMixin._is_cjk_codepoint  s   "&&&&&&&& '"&&&&&&&&'2((((((((' "&&&&&&&&' "&&&&&&&&	'
 "&&&&&&&&' "&&&&&&&&	(r+   r?  c                    | D ]~}t          |          }d|cxk    rdk    s]n d|cxk    rdk    sNn d|cxk    rdk    s?n d|cxk    rdk    s0n d	|cxk    rd
k    s!n d|cxk    rdk    sn d|cxk    rdk    rn { dS dS )zBCheck if text contains CJK (Chinese, Japanese, Korean) characters.rj  rk  rl  rm  rn  ro  rp  rq  rr  rs  rt  ru  rv  rw  TF)ord)r?  rc  rh  s      r)   _contains_cjkz SessionSearchMixin._contains_cjk  s     		 		BRB"&&&&&&&&"&&&&&&&&2(((((((("&&&&&&&&"&&&&&&&&"&&&&&&&&"&&&&&&&&&tt 'ur+   c                 :     t           fd|D                       S )zCount CJK characters in text.c              3   `   K   | ](}                     t          |                    $d V  )dS )r   Nrx  rz  )r&   rc  r6   s     r)   r*   z0SessionSearchMixin._count_cjk.<locals>.<genexpr>  s<      FFs'<'<SWW'E'EF1FFFFFFr+   )sum)r6   r?  s   ` r)   
_count_cjkzSessionSearchMixin._count_cjk  s(     FFFFtFFFFFFr+   c                     d}|D ]5}|                      t          |                    r|dz  }*|dk    r dS d}6|dk    S )u&  True when any maximal CJK run in the query is a single char.

        The cjk-bigram index stores bigrams for runs >=2 chars and unigrams
        only for isolated chars, so a 1-char CJK term can't match inside
        longer runs there — those queries keep the LIKE substring route.
        r   r   Tr~  )r6   rW  runrc  s       r)   _has_lone_cjk_runz$SessionSearchMixin._has_lone_cjk_run  s_      	 	B$$SWW-- q!8844axr+   c                     d |                      d                                                                           D             }t          |          ot          d |D                       S )a  True when every non-operator token is long enough for the trigram
        tokenizer to match (>=3 chars).

        The trigram tokenizer indexes overlapping 3-character sequences, so a
        token shorter than 3 chars produces no trigrams and can never match.
        With FTS5's implicit-AND between tokens, a single short token makes the
        whole MATCH return nothing, so the trigram path is only worth taking
        when every searchable token qualifies.
        c                 >    g | ]}|                                 d v|S >   ORANDNOTr   rE  s     r)   rz   z?SessionSearchMixin._trigram_eligible_tokens.<locals>.<listcomp>  s6     
 
 
wwyy 444 444r+   rY  c              3   <   K   | ]}t          |          d k    V  dS r  N)rM  rE  s     r)   r*   z>SessionSearchMixin._trigram_eligible_tokens.<locals>.<genexpr>  s,      #@#@ACFFaK#@#@#@#@#@#@r+   )rK  rL  r   all)rW  tokenss     r)   _trigram_eligible_tokensz+SessionSearchMixin._trigram_eligible_tokens  sn    
 
{{3''--//5577
 
 
 F||@#@#@#@#@#@ @ @@r+   r   r   )tablesource_filterexclude_sourcesrole_filterr9  offset	raw_queryr  order_by_sqlr  r  r  r  c                   |                                 }
g }|
D ]]}|                                dv r|                    |           .|                    d|                    dd          z   dz              ^d                    |          }| dg}|g}|s|                    d           |K|                    dd	                    d
 |D                        d           |                    |           |K|                    dd	                    d |D                        d           |                    |           |rK|                    dd	                    d |D                        d           |                    |           d| d| d| dd                    |           d| d}|                    ||	g           |                                 5 }	 |                    ||          }n!# t          j	        $ r Y ddd           dS w xY wd |
                                D             cddd           S # 1 swxY w Y   dS )u  Run a search against a substring-capable FTS index.

        ``table`` is ``messages_fts_trigram`` (default) or
        ``messages_fts_cjk``. The trigram tokenizer indexes overlapping
        3-byte sequences, so it matches substrings regardless of word
        boundaries — both CJK phrases the unicode61 tokenizer splits into
        single characters and Latin runs the unicode61 tokenizer fuses onto
        adjacent CJK (e.g. ``修改youer服务端``). The cjk-bigram tokenizer
        splits Latin runs off adjacent CJK, giving the same recovery as an
        exact ranked token match. Each non-operator token is quoted to
        neutralise FTS5 special characters while boolean operators
        (AND/OR/NOT) are preserved.

        Returns the matching rows, or ``None`` when the query cannot be
        executed (e.g. the tokenizer is unavailable at runtime) so the
        caller can fall back to another strategy.
        >   r  r  r  rY  ""rC  z MATCH ?!(m.active = 1 OR m.compacted = 1)Ns.source IN (r  c              3      K   | ]}d V  dS r  r%   r  s     r)   r*   z9SessionSearchMixin._run_trigram_search.<locals>.<genexpr>  s"      5Q5Qac5Q5Q5Q5Q5Q5Qr+   r   s.source NOT IN (c              3      K   | ]}d V  dS r  r%   r  s     r)   r*   z9SessionSearchMixin._run_trigram_search.<locals>.<genexpr>  s"      9W9W!#9W9W9W9W9W9Wr+   m.role IN (c              3      K   | ]}d V  dS r  r%   r  s     r)   r*   z9SessionSearchMixin._run_trigram_search.<locals>.<genexpr>  s"      3M3MAC3M3M3M3M3M3Mr+   zx
            SELECT
                m.id,
                m.session_id,
                m.role,
                snippet(z, -1, '>>>', '<<<', '...', 40) AS snippet,
                m.content,
                m.timestamp,
                m.tool_name,
                s.source,
                s.model,
                s.started_at AS session_started
            FROM z'
            JOIN messages m ON m.id = zL.rowid
            JOIN sessions s ON s.id = m.session_id
            WHERE  AND 
            &
            LIMIT ? OFFSET ?
        c                 ,    g | ]}t          |          S r%   r  r&   r`   s     r)   rz   z:SessionSearchMixin._run_trigram_search.<locals>.<listcomp>  s    ???#DII???r+   )rL  r   rN  r   r3   extendrX   rY   r>   r   rZ   )rB   r  r  r  r:  r  r  r  r9  r  r  partstoktrigram_query	tri_where
tri_paramstri_sqlr_   
tri_cursors                      r)   _run_trigram_searchz&SessionSearchMixin._run_trigram_search  sQ   < "" 	A 	ACyy{{222S!!!!S3;;sD#9#99C?@@@@'''(	)?
 	B@AAA$TSXX5Q5Q=5Q5Q5Q-Q-QTTTUUUm,,,&Z9W9W9W9W9W1W1WZZZ[[[o... 	+P3883M3M3M3M3M+M+MPPPQQQk***
     (-  <<	**   !  & 	5&/***^^ 	@!\\':>>

+   	@ 	@ 	@ 	@ 	@ 	@ 	@ 	@ @?)<)<)>)>???	@ 	@ 	@ 	@ 	@ 	@ 	@ 	@ 	@ 	@ 	@ 	@ 	@ 	@ 	@ 	@ 	@ 	@s6   #I%G<;I<HIH IIIsortc
                    t          j                     }
d}	 |                     |||||||||		  	        }|	 t          t          j        dd                    }n# t
          t          f$ r d}Y nw xY wt          j                     |
z
  dz  }||k    rKt                              d| 	                    |          ||t          |          nd|dd                    S S # 	 t          t          j        dd                    }n# t
          t          f$ r d}Y nw xY wt          j                     |
z
  dz  }||k    rKt                              d| 	                    |          ||t          |          nd|dd                    w w xY w)	a  Instrumented wrapper around :meth:`_search_messages_impl`.

        Logs one line per slow search with the routing path taken, so
        production latency stays attributable per query shape (the 2026-07
        session_search investigation needed trace archaeology to discover
        the LIKE full scans; this makes the next regression a grep).
        Threshold: HERMES_SEARCH_SLOW_MS (default 1000; 0 logs every call).
        N)r  r  r  r9  r  r  r:  r   HERMES_SEARCH_SLOW_MS1000g     @@z<slow session search: path=%s elapsed=%.0fms rows=%s query=%rerr   )r   _search_messages_implr   osgetenvr.   r2   r@   ru   _describe_search_pathrM  )rB   rW  r  r  r  r9  r  r  r:  r   startedrR  	threshold
elapsed_mss                 r)   search_messagesz"SessionSearchMixin.search_messages  s   ( )++	--+ /'!1 . 
 
D #!"),CV"L"LMM		z* # # #"			#)++/69JY&&R..u55!%!1CIIIu$3$K    '#!"),CV"L"LMM		z* # # #"			#)++/69JY&&R..u55!%!1CIIIu$3$K    'sA   C "AA0/A0F "DFDFDA.Fc                     	                       |pd          }|sdS                      |          sdS |                    d                                          } j        r                     |          sdS  fd|                                D             }t           fd|D                       }                     |          dk    r|s	 j        rd	S d
S # t          $ r Y dS w xY w)z>Best-effort name of the routing path a query takes (log-only).r  emptyfts5rY  fts_cjkc                 j    g | ]/}|                                 d v                    |          -|0S r  r   r{  r&   rF  rB   s     r)   rz   z<SessionSearchMixin._describe_search_path.<locals>.<listcomp>X  sK       7799$888T=O=OPQ=R=R8 888r+   c              3   J   K   | ]}                     |          d k     V  dS r  r  r  s     r)   r*   z;SessionSearchMixin._describe_search_path.<locals>.<genexpr>\  s4      ??1**Q.??????r+   r  trigram	like_scanr7   )
rg  r{  rK  r   r  rL  anyr  rs   r   )rB   rW  re  rawr  shorts   `     r)   r  z(SessionSearchMixin._describe_search_pathM  s-   	11%+2>>I w%%i00 v//#&&,,..C& !t/E/Ec/J/J ! y   99;;  F ?????????Es##q(((4;R( y; 	 	 	99	s#   C C AC :AC 
C)(C)c
                 p   AB                       |	          A j        sg S |r|                                sg S                      |          }|sg S t	          |t
                    r-|                                                                }
|
dvrd}
nd}
|
dk    rd}n|
dk    rd}nd}dg}|g}|s|                    d	           |Md
                    d |D                       }|                    d| d           |	                    |           |Md
                    d |D                       }|                    d| d           |	                    |           |rMd
                    d |D                       }|                    d| d           |	                    |           d                    |          }|	                    ||g           d| d| d}g } 
                    |          }|	rV|                    d                                          }                     |          } fd|                                D             }t           fd|D                       }d}t          |          od|v } j        rK|sH                     |          s2|                                }g }|D ]]}|                                dv r|                    |           .|                    d|                    dd          z   dz              ^d                    |          }dg}|g} |s|                    d	           |K|                    dd
                    d |D                        d           | 	                    |           |K|                    dd
                    d  |D                        d           | 	                    |           |rK|                    dd
                    d! |D                        d           | 	                    |           d"d                    |           d#| d$}!| 	                    ||g           	                                  5 }"|"                    |!|           }#d% |#                                D             }d&}ddd           n# 1 swxY w Y   n# t,          j        $ r t0                              d'd&(           Y nt,          j        $ r}$                     |$          r	                                  5 }"|"                    |!|           }#d) |#                                D             }d&}ddd           n# 1 swxY w Y   nJ# t,          j        $ r t0                              d*           Y nw xY wt0                              d+|$           Y d}$~$nd}$~$ww xY w|s*|d,k    r#|s  j        r|s|                                }g }|D ]]}|                                dv r|                    |           .|                    d|                    dd          z   dz              ^d                    |          }%d-g}&|%g}'|s|&                    d	           |K|&                    dd
                    d. |D                        d           |'	                    |           |K|&                    dd
                    d/ |D                        d           |'	                    |           |rK|&                    dd
                    d0 |D                        d           |'	                    |           d1d                    |&           d#| d$}(|'	                    ||g           	                                  5 }"|"                    |(|'          })d2 |)                                D             }d&}ddd           n# 1 swxY w Y   n# t,          j        $ r Y nt,          j        $ r}$                     |$          r	                                  5 }"|"                    |(|'          })d3 |)                                D             }d&}ddd           n# 1 swxY w Y   nJ# t,          j        $ r t0                              d4           Y nw xY wt0                              d5|$           Y d}$~$nd}$~$ww xY w|sEd6 |                                D             p|g}*g }+g },|*D ]i}|                    d7d8                              d9d:                              d;d<          }-|+                    d=           |,d9|- d9d9|- d9d9|- d9gz  },jd>d?                    |+           dg}.|s|.                    d	           |K|.                    dd
                    d@ |D                        d           |,	                    |           |K|.                    dd
                    dA |D                        d           |,	                    |           |rK|.                    dd
                    dB |D                        d           |,	                    |           dCd                    |.           dD}/|,	                    ||g           |*dE         g|,z   },                                 5 }"|"                    |/|,          }0dF |0                                D             }ddd           n# 1 swxY w Y   n	                                  5 }"|"                    ||          }1dG |1                                D             }ddd           n# 1 swxY w Y   n# t,          j        $ r g cY S t,          j        $ r}$                     |$          s                                  5 }"|"                    ||          }1dH |1                                D             }ddd           n# 1 swxY w Y   Y d}$~$nd}$~$ww xY w                                 }2|2t?          |          |k     r	                       ||t?          |          z
  ||||I          }3dJ |D             B|	                    BfdK|3D                        n7# t,          j        $ r%}$t0                              dL|$           Y d}$~$nd}$~$ww xY w|s|st          |          rd|v s|                    d                                          }4 j        r" !                    |4dM|||||||N	  	        }5|5r|5}|s= j        r6 "                    |          r! !                    |4|||||||O          }6|6r|6}AdPAv r|ndQ}7|7D ]=}8	                                  5 }"|"                    dR|8dS         |8dS         f          }9g }:|9                                D ]};|;dT         }< #                    |<          }=t	          |=tH                    rBdU |=D             }>d                    dV |>D                                                       }?|?pdW}@nt	          |=t
                    r|=}@ndX}@|:                    |;dY         |@ddZ         d[           	 ddd           n# 1 swxY w Y   |:|8dP<   )# tJ          $ r	 g |8dP<   Y ;w xY w|D ]}8|8&                    dTd           AAfd\|D             }|S )]u  
        Full-text search across session messages using FTS5.

        Supports FTS5 query syntax:
          - Simple keywords: "docker deployment"
          - Phrases: '"exact phrase"'
          - Boolean: "docker OR kubernetes", "python NOT java"
          - Prefix: "deploy*"

        Returns matching messages with session metadata, content snippet,
        and surrounding context (1 message before and after the match).
        ``fields`` selects a result projection; omitting it preserves the
        complete legacy result. Context is only loaded when that projection
        consumes it.

        ``sort`` controls temporal ordering:
          - ``None`` (default): FTS5 BM25 relevance only. Time-neutral.
          - ``"newest"``: order by message timestamp DESC, then by rank.
          - ``"oldest"``: order by message timestamp ASC, then by rank.

        The short-CJK LIKE fallback already orders by timestamp DESC and
        ignores ``sort``. The trigram CJK path honours ``sort`` like the main
        FTS5 path.

        Rewound (``active=0``, ``compacted=0``) rows are excluded by default —
        the user took those back. Compaction-archived rows (``active=0``,
        ``compacted=1``) ARE included by default: they were summarized away from
        the live context but remain part of the conversation's record, so the
        pre-compaction transcript stays discoverable after in-place compaction
        (#38763). Pass ``include_inactive=True`` to search every row regardless.
        )newestoldestNr  zORDER BY m.timestamp DESC, rankr  zORDER BY m.timestamp ASC, rankzORDER BY rankzmessages_fts MATCH ?r  r  c              3      K   | ]}d V  dS r  r%   r  s     r)   r*   z;SessionSearchMixin._search_messages_impl.<locals>.<genexpr>  s"      *F*F13*F*F*F*F*F*Fr+   r  r   c              3      K   | ]}d V  dS r  r%   r  s     r)   r*   z;SessionSearchMixin._search_messages_impl.<locals>.<genexpr>  s"      +I+IAC+I+I+I+I+I+Ir+   r  c              3      K   | ]}d V  dS r  r%   r  s     r)   r*   z;SessionSearchMixin._search_messages_impl.<locals>.<genexpr>  s"      (B(B(B(B(B(B(B(Br+   r  r  a  
            SELECT
                m.id,
                m.session_id,
                m.role,
                snippet(messages_fts, -1, '>>>', '<<<', '...', 40) AS snippet,
                m.content,
                m.timestamp,
                m.tool_name,
                s.source,
                s.model,
                s.started_at AS session_started
            FROM messages_fts
            JOIN messages m ON m.id = messages_fts.rowid
            JOIN sessions s ON s.id = m.session_id
            WHERE r  r  rY  c                 j    g | ]/}|                                 d v                    |          -|0S r  r  r  s     r)   rz   z<SessionSearchMixin._search_messages_impl.<locals>.<listcomp>  sK     ! ! !7799$888T=O=OPQ=R=R8 888r+   c              3   J   K   | ]}                     |          d k     V  dS r  r  r  s     r)   r*   z;SessionSearchMixin._search_messages_impl.<locals>.<genexpr>  sE       ! !+,""Q&! ! ! ! ! !r+   Ftool>   r  r  r  r  rC  zmessages_fts_cjk MATCH ?c              3      K   | ]}d V  dS r  r%   r  s     r)   r*   z;SessionSearchMixin._search_messages_impl.<locals>.<genexpr>  "      =Y=Yac=Y=Y=Y=Y=Y=Yr+   c              3      K   | ]}d V  dS r  r%   r  s     r)   r*   z;SessionSearchMixin._search_messages_impl.<locals>.<genexpr>  "      A_A_!#A_A_A_A_A_A_r+   c              3      K   | ]}d V  dS r  r%   r  s     r)   r*   z;SessionSearchMixin._search_messages_impl.<locals>.<genexpr>   "      ;U;UAC;U;U;U;U;U;Ur+   a  
                    SELECT
                        m.id,
                        m.session_id,
                        m.role,
                        snippet(messages_fts_cjk, -1, '>>>', '<<<', '...', 40) AS snippet,
                        m.content,
                        m.timestamp,
                        m.tool_name,
                        s.source,
                        s.model,
                        s.started_at AS session_started
                    FROM messages_fts_cjk
                    JOIN messages m ON m.id = messages_fts_cjk.rowid
                    JOIN sessions s ON s.id = m.session_id
                    WHERE z
                    z6
                    LIMIT ? OFFSET ?
                c                 ,    g | ]}t          |          S r%   r  r  s     r)   rz   z<SessionSearchMixin._search_messages_impl.<locals>.<listcomp>9      "N"N"N499"N"N"Nr+   Tz;messages_fts_cjk query failed; falling back to trigram/LIKE)exc_infoc                 ,    g | ]}t          |          S r%   r  r  s     r)   rz   z<SessionSearchMixin._search_messages_impl.<locals>.<listcomp>K  +     +" +" +"25DII+" +" +"r+   zYCJK-bigram FTS search still failing after in-place rebuild; falling back to trigram/LIKE.zuCJK-bigram FTS search hit a corruption error (%s) and no in-place rebuild was possible; falling back to trigram/LIKE.r  zmessages_fts_trigram MATCH ?c              3      K   | ]}d V  dS r  r%   r  s     r)   r*   z;SessionSearchMixin._search_messages_impl.<locals>.<genexpr>s  r  r+   c              3      K   | ]}d V  dS r  r%   r  s     r)   r*   z;SessionSearchMixin._search_messages_impl.<locals>.<genexpr>v  r  r+   c              3      K   | ]}d V  dS r  r%   r  s     r)   r*   z;SessionSearchMixin._search_messages_impl.<locals>.<genexpr>y  r  r+   a  
                    SELECT
                        m.id,
                        m.session_id,
                        m.role,
                        snippet(messages_fts_trigram, -1, '>>>', '<<<', '...', 40) AS snippet,
                        m.content,
                        m.timestamp,
                        m.tool_name,
                        s.source,
                        s.model,
                        s.started_at AS session_started
                    FROM messages_fts_trigram
                    JOIN messages m ON m.id = messages_fts_trigram.rowid
                    JOIN sessions s ON s.id = m.session_id
                    WHERE c                 ,    g | ]}t          |          S r%   r  r  s     r)   rz   z<SessionSearchMixin._search_messages_impl.<locals>.<listcomp>  r  r+   c                 ,    g | ]}t          |          S r%   r  r  s     r)   rz   z<SessionSearchMixin._search_messages_impl.<locals>.<listcomp>  r  r+   zNTrigram FTS search still failing after in-place rebuild; falling back to LIKE.zjTrigram FTS search hit a corruption error (%s) and no in-place rebuild was possible; falling back to LIKE.c                 >    g | ]}|                                 d v|S r  r  rE  s     r)   rz   z<SessionSearchMixin._search_messages_impl.<locals>.<listcomp>  s6     ! ! !wwyy(<<< <<<r+   \\\r}   \%r{   r|   `(m.content LIKE ? ESCAPE '\' OR m.tool_name LIKE ? ESCAPE '\' OR m.tool_calls LIKE ? ESCAPE '\')r   z OR c              3      K   | ]}d V  dS r  r%   r  s     r)   r*   z;SessionSearchMixin._search_messages_impl.<locals>.<genexpr>  s"      >Z>Zqs>Z>Z>Z>Z>Z>Zr+   c              3      K   | ]}d V  dS r  r%   r  s     r)   r*   z;SessionSearchMixin._search_messages_impl.<locals>.<genexpr>  s"      B`B`13B`B`B`B`B`B`r+   c              3      K   | ]}d V  dS r  r%   r  s     r)   r*   z;SessionSearchMixin._search_messages_impl.<locals>.<genexpr>  s"      <V<VQS<V<V<V<V<V<Vr+   a  
                    SELECT m.id, m.session_id, m.role,
                           substr(m.content,
                                  max(1, instr(m.content, ?) - 40),
                                  120) AS snippet,
                           m.content, m.timestamp, m.tool_name,
                           s.source, s.model, s.started_at AS session_started
                    FROM messages m
                    JOIN sessions s ON s.id = m.session_id
                    WHERE zd
                    ORDER BY m.timestamp DESC
                    LIMIT ? OFFSET ?
                r   c                 ,    g | ]}t          |          S r%   r  r  s     r)   rz   z<SessionSearchMixin._search_messages_impl.<locals>.<listcomp>  s    KKKStCyyKKKr+   c                 ,    g | ]}t          |          S r%   r  r  s     r)   rz   z<SessionSearchMixin._search_messages_impl.<locals>.<listcomp>      FFFStCyyFFFr+   c                 ,    g | ]}t          |          S r%   r  r  s     r)   rz   z<SessionSearchMixin._search_messages_impl.<locals>.<listcomp>  r  r+   r:  r  r  r  c                     h | ]
}|d          S )r   r%   )r&   r  s     r)   	<setcomp>z;SessionSearchMixin._search_messages_impl.<locals>.<setcomp>  s    555AdG555r+   c              3   0   K   | ]}|d          v|V  dS )r   Nr%   )r&   r  seen_idss     r)   r*   z;SessionSearchMixin._search_messages_impl.<locals>.<genexpr>  s1      QQQ49P9Pq9P9P9P9PQQr+   z$Unindexed-gap supplement skipped: %smessages_fts_cjk)r  r  r:  r  r  r  r9  r  )r  r:  r  r  r  r9  r  r   r%   a  WITH target AS (
                               SELECT session_id, timestamp, id
                               FROM messages
                               WHERE id = ?
                           )
                           SELECT role, content
                           FROM (
                               SELECT m.id, m.timestamp, m.role, m.content
                               FROM messages m
                               JOIN target t ON t.session_id = m.session_id
                               WHERE (m.timestamp < t.timestamp)
                                  OR (m.timestamp = t.timestamp AND m.id < t.id)
                               ORDER BY m.timestamp DESC, m.id DESC
                               LIMIT 1
                           )
                           UNION ALL
                           SELECT role, content
                           FROM messages
                           WHERE id = ?
                           UNION ALL
                           SELECT role, content
                           FROM (
                               SELECT m.id, m.timestamp, m.role, m.content
                               FROM messages m
                               JOIN target t ON t.session_id = m.session_id
                               WHERE (m.timestamp > t.timestamp)
                                  OR (m.timestamp = t.timestamp AND m.id > t.id)
                               ORDER BY m.timestamp ASC, m.id ASC
                               LIMIT 1
                           )r   r  c                     g | ]F}t          |t                    r/|                    d           dk    0|                    dd          GS r=  r@  rA  s     r)   rz   z<SessionSearchMixin._search_messages_impl.<locals>.<listcomp>  sW     * * *67#-a#6#6*;<55==F;R;R !"fb 1 1;R;R;Rr+   c              3      K   | ]}||V  	d S r$   r%   rE  s     r)   r*   z;SessionSearchMixin._search_messages_impl.<locals>.<genexpr>  s'      +G+G!Q+GA+G+G+G+G+G+Gr+   rG  r  r   r  )r   r  c                 .    g | ]fd D             S )c                 *    i | ]}|v ||         S r%   r%   )r&   r'   matchs     r)   rO   zGSessionSearchMixin._search_messages_impl.<locals>.<listcomp>.<dictcomp>  s$    RRR5E>>e>>>r+   r%   )r&   r  result_fieldss    @r)   rz   z<SessionSearchMixin._search_messages_impl.<locals>.<listcomp>  s?        SRRR-RRR  r+   )'r8   r;   rK  rg  r,   r-   lowerrN  r3   r  r{  r  rL  r  r   r   r  r   r   rX   rY   rZ   r>   r   r@   r   DatabaseError_try_runtime_fts_rebuildrA   rs   re   rM  _search_unindexed_gapr  r  r   r+  r   pop)CrB   rW  r  r  r  r9  r  r  r:  r   	sort_normr  where_clausesparamssource_placeholdersexclude_placeholdersr6  	where_sqlsqlmatchesis_cjkr  	cjk_count_tokens_for_check_any_short_cjk_trigram_succeeded_wants_tool_rowsr  r  r  	cjk_query	cjk_where
cjk_paramscjk_sqlr_   
cjk_cursorrC   r  r  r  r  r  non_op_tokenstoken_clauseslike_paramsesc
like_wherelike_sqllike_cursorrQ  rebuild_statusgap_matches	_fb_querycjk_fbtri_matchescontext_matchesr  
ctx_cursorcontext_msgsrN   r  rT  rU  r?  rJ  r  r  sC   `                                                                @@r)   r  z(SessionSearchMixin._search_messages_implc  s)   V 33F;;  	I 	EKKMM 	I))%00 	I
 dC   	

**,,I 444 	I   <LL("";LL*L 00w 	F   !DEEE$"%((*F*F*F*F*F"F"F  !G1D!G!G!GHHHMM-(((&#&88+I+I+I+I+I#I#I   !L5I!L!L!LMMMMM/*** 	' #(B(Bk(B(B(B B B  !C/@!C!C!CDDDMM+&&&LL//	ufo&&&    !  : )+##E** Z	GC((..00I	22I! ! ! !$??,,! ! ! ! ! ! ! !0A! ! !  N "'
  $K00JV{5J 'R(R ..y99R
 #**! I ICyy{{&:::S))))S3;;sD+A+A%AC%GHHHHHHUOO	78	$-;
' J$$%HIII ,$$%\SXX=Y=Y==Y=Y=Y5Y5Y%\%\%\]]]%%m444".$$%bA_A_A_A_A_9_9_%b%b%bccc%%o666 3$$%X388;U;U;U;U;U3U3U%X%X%XYYY%%k222 #<<	22   "!  & !!5&/222$)) 2T%)\\':%F%F
"N"N
8K8K8M8M"N"N"N-1*2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 /    LL'15 !      ,    44S99 !%!1!1 :T-1\\$+Z." ."
+" +"9C9L9L9N9N+" +" +" 6: 2: : : : : : : : : : : : : : :  '4   "NN!0     <=@  )6 '[NN& #+ # ) # #**! I ICyy{{&:::S))))S3;;sD+A+A%AC%GHHHH #;<	$1?
' J$$%HIII ,$$%\SXX=Y=Y==Y=Y=Y5Y5Y%\%\%\]]]%%m444".$$%bA_A_A_A_A_9_9_%b%b%bccc%%o666 3$$%X388;U;U;U;U;U3U3U%X%X%XYYY%%k222 #<<	22   "!  & !!5&/222()) 2T%)\\':%F%F
"N"N
8K8K8M8M"N"N"N-1*2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 /   D,       44S99 !%!1!1 :T-1\\$+Z." ."
+" +"9C9L9L9N9N+" +" +" 6: 2: : : : : : : : : : : : : : :  '4   "NN!J     ,-0  9 B & 3L! !(00! ! ! !!  [  !#$&( H HC++dF33;;CGGOOPSUZ[[C!((}    JJJJ
C


JJJJ#GGKK?&++m"<"<???@
' K %%&IJJJ ,%%&]chh>Z>ZM>Z>Z>Z6Z6Z&]&]&]^^^&&}555".%%&c#((B`B`P_B`B`B`:`:`&c&c&cddd&&777 4%%&YCHH<V<V+<V<V<V4V4V&Y&Y&YZZZ&&{333 #<<
33   ""E6?333,Q/0;>^^%% L"&,,x"E"EKKKK4H4H4J4JKKKGL L L L L L L L L L L L L L LG^^%% G!\\#v66FFFFOO4E4EFFFGG G G G G G G G G G G G G G G +   			( G G G 44S99 ^^%% G!\\#v66FFFFOO4E4EFFFGG G G G G G G G G G G G G G GG, 0022%#g,,*>*>J"88CLL(%5"/$3 + 9   65W555QQQQ+QQQQQQQ+ J J JCSIIIIIIIIJ* $	*$	* +&&$	* ,2[+@+@C((..00I& %11,!-%5"/$3 +! 2 
 
  %$G*+* 11%88*
 #66!-%5"/$3 +! 7 	 	  *)G %,	]0J0JGGPR 	 % :	& :	&E9&^^%% 5!% < teDk2? "  "JB $&L'0022  	l"&"6"6s";"; &gt44 
)* *;B* * *J $'88+G+Gz+G+G+G#G#G#M#M#O#OD&*&D.DGG'55 )&-GG&(G$++%&vY74C4=II   !G5 5 5 5 5 5 5 5 5 5 5 5 5 5 5l $0i   & & &#%i   &  	' 	'EIIi&&&&$   $  G
 s  4S 7S?S SS SS +W W W)V=7V 4V V	VV	VW)V85W7V88WW >`! 7`	`! ``! ``! !d2d db9*7b-!b9-b1	1b94b1	5b98d9)c%"d$c%%dd5mmmn< /5n0$n< 0n44n< 7n48n< <qq*q5q<qq	qq	qq
As# #t2tt{5$C2{#{5#{'	'{5*{'	+{55||r  	fts_queryc                P   |                                  }||dk    rg S |d         |d         }	}g }
t          j        d|          D ]j}|                    d                              d                                          }|r|                                dv rU|
                    |           k|
sg S d	g}||	g}|
D ]b}|                    d
d                              dd                              dd          }|                    d           |d| dgdz  z  }c|s|                    d           |K|                    dd                    d |D                        d           |                    |           |K|                    dd                    d |D                        d           |                    |           |rK|                    dd                    d |D                        d           |                    |           dd                    |           d}|
d         g|z   |gz   }| 	                                5 }|
                    ||                                          }ddd           n# 1 swxY w Y   d |D             S )u  LIKE-scan the rows the deferred rebuild hasn't indexed yet.

        Only touches ids in (fts_rebuild_progress, fts_rebuild_high_water] —
        a range that shrinks to nothing as the backfill advances. The FTS
        query is degraded to per-token substring terms (AND-joined; quoted
        phrases kept whole), which is deliberately recall-over-precision:
        temporary results beat silently missing ones mid-rebuild.
        Nr   rU   rT   z"[^"]+"|\S+rY  r[  >   r  r  r  NEARzm.id > ? AND m.id <= ?r  r  r}   r  r{   r|   r  r  r  r  r  c              3      K   | ]}d V  dS r  r%   r  s     r)   r*   z;SessionSearchMixin._search_unindexed_gap.<locals>.<genexpr>  s"      1M1M!#1M1M1M1M1M1Mr+   r   r  c              3      K   | ]}d V  dS r  r%   r  s     r)   r*   z;SessionSearchMixin._search_unindexed_gap.<locals>.<genexpr>  s"      5S5Sac5S5S5S5S5S5Sr+   r  c              3      K   | ]}d V  dS r  r%   r  s     r)   r*   z;SessionSearchMixin._search_unindexed_gap.<locals>.<genexpr>  s"      /I/I/I/I/I/I/I/Ir+   a  
            SELECT m.id, m.session_id, m.role,
                   substr(m.content,
                          max(1, instr(m.content, ?) - 40),
                          120) AS snippet,
                   m.content, m.timestamp, m.tool_name,
                   s.source, s.model, s.started_at AS session_started
            FROM messages m
            JOIN sessions s ON s.id = m.session_id
            WHERE r  zC
            ORDER BY m.timestamp DESC
            LIMIT ?
        c                 ,    g | ]}t          |          S r%   r  rM   s     r)   rz   z<SessionSearchMixin._search_unindexed_gap.<locals>.<listcomp>  s    &&&AQ&&&r+   )re   r]  findallrK  r   rN  r   r3   r  rX   rY   rZ   )rB   r#  r9  r:  r  r  r  r   rc   rb   termsraw_tokr  wherer  termr  r  r_   rR  s                       r)   r  z(SessionSearchMixin._search_unindexed_gap  s7   $ ((**>UaZZI%i0&/* z.)<< 	 	G--$$**3//5577C #))++)EEELL 	I)* *- 	' 	'D,,tV,,44S%@@HHeTTCLL6   z3zzzlQ&&FF 	>LL<===$LLP1M1M}1M1M1M)M)MPPPQQQMM-(((&LLVSXX5S5S?5S5S5S-S-SVVVWWWMM/*** 	'LLLsxx/I/I[/I/I/I'I'ILLLMMMMM+&&& <<&&   (f$w.^^ 	8<<V,,5577D	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8&&&&&&s   )JJJinclude_archivedr   sourcesc           
      |  	
 |pd                                                                 		r|dk    rg S |                     |||t          |dz  |          d|d	          }dt          t
          t          f         dt          f	fd
t          t          |          
fd	
          }d |d|         D             S )a  Search surfaced sessions by exact/prefix/substring session id.

        Desktop search uses this alongside FTS message search so users can paste
        a session id from logs, CLI output, or another Hermes surface and jump
        straight to that conversation.  Matching also checks ``_lineage_root_id``
        for projected compression-chain tips, so an old root id still resolves to
        the live continuation row.
        r  r      T)r   r0  r  r9  r  r/  order_by_last_activeid_queryr`   r    c                 $   t          |                     d          pd          t          |                     d          pd          g}d |D             }t          fd|D                       rdS t          fd|D                       rdS d	S )
Nr   r  _lineage_root_idc                 :    g | ]}||                                 S r%   )r  )r&   rL   s     r)   rz   zKSessionSearchMixin.search_sessions_by_id.<locals>.score.<locals>.<listcomp>  s%    BBBEEB%++--BBBr+   c              3   $   K   | ]
}|k    V  d S r$   r%   r&   rL   needles     r)   r*   zJSessionSearchMixin.search_sessions_by_id.<locals>.score.<locals>.<genexpr>  s'      ;;u5F?;;;;;;r+   r   c              3   B   K   | ]}|                               V  d S r$   )
startswithr9  s     r)   r*   zJSessionSearchMixin.search_sessions_by_id.<locals>.score.<locals>.<genexpr>  s1      DD5##F++DDDDDDr+   r      )r-   r[   r  )r`   ids
normalizedr:  s      r)   scorez7SessionSearchMixin.search_sessions_by_id.<locals>.score  s    swwt}}*++S9K1L1L1RPR-S-STCBBSBBBJ;;;;
;;;;; qDDDDDDDDD q1r+   c                 6     | d                   | d         fS )Nr   r   r%   )itemr@  s    r)   <lambda>z:SessionSearchMixin.search_sessions_by_id.<locals>.<lambda>  s    eeDGnnd1g6 r+   )rK   c                     g | ]\  }}|S r%   r%   )r&   r{   r`   s      r)   rz   z<SessionSearchMixin.search_sessions_by_id.<locals>.<listcomp>  s    1113111r+   N)
rK  r  list_sessions_richr   r   r-   r   r\   r4   r_  )rB   rW  r9  r/  r   r0  r  
candidatesrankedr:  r@  s            @@r)   search_sessions_by_idz(SessionSearchMixin.search_sessions_by_id  s    " +2$$&&,,.. 	!I ,,+eai''-!% - 	
 	

	tCH~ 	# 	 	 	 	 	 	 j!!6666
 
 
 21&%.1111r+   namec                 p    	 | j                             d| d           dS # t          j        $ r Y dS w xY w)z6True if an FTS5 virtual table is queryable in this DB.zSELECT 1 FROM z LIMIT 0TF)r   rY   r>   r  )rB   rI  s     r)   _fts_table_existsz$SessionSearchMixin._fts_table_exists  sU    	J>>>>???4$ 	 	 	
 55	s   " 55c           	      H   d}| j         5  | j        D ]w}|                     |          s	 | j                            d| d| d           |dz  }@# t
          j        $ r&}t                              d||           Y d}~pd}~ww xY w	 ddd           n# 1 swxY w Y   |S )u  Merge fragmented FTS5 b-tree segments into one per index.

        FTS5 indexes grow as a series of incremental segments — one per
        ``INSERT`` batch driven by the message triggers. Over tens of
        thousands of messages these segments accumulate, which both bloats
        the ``*_data`` shadow tables and slows ``MATCH`` queries that must
        scan every segment. The special ``'optimize'`` command rewrites each
        index as a single merged segment.

        This is purely a maintenance operation — it changes neither search
        results nor ``snippet()`` output, only on-disk layout and query
        speed. It is complementary to VACUUM: ``optimize`` compacts the FTS
        index internally, then VACUUM returns the freed pages to the OS.

        Skips any FTS table that does not exist (e.g. the trigram index when
        disabled via ``HERMES_DISABLE_FTS_TRIGRAM`` or not yet created), so
        it is safe to call unconditionally.

        Returns the number of FTS indexes that were optimized.
        r   r   r   z) VALUES('optimize')r   zFTS optimize failed for %s: %sN)	r   _FTS_TABLESrK  r   rY   r>   r   r@   rA   )rB   	optimizedr   rC   s       r)   optimize_ftszSessionSearchMixin.optimize_fts*  s:   * 	Z 	 	'  --c22 
 J&&FsFFSFFF   NII/   NN8#s       	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 s:    B&ABB!B=BBBBBc           	         d}| j         5  | j        D ]}|                     |          s	 | j                            d| d| d           | j                                         |dz  }Y# t          j        $ r?}| j                                         t          
                    d||           Y d}~d}~ww xY w	 ddd           n# 1 swxY w Y   |S )aN  Rebuild FTS5 indexes from the canonical ``messages`` table.

        Uses the FTS5 ``'rebuild'`` command, which rewrites the internal
        b-tree segments from the content rows. This is the documented
        recovery for a corrupt FTS index that rejects message writes while
        reads still succeed (issue #50502). Unlike ``optimize_fts`` (which
        merges existing segments), ``rebuild`` discards and recreates the
        index data entirely.

        Safe to call when FTS tables don't exist (skips them).
        Returns the number of FTS indexes that were rebuilt.
        r   r   r   z) VALUES('rebuild')r   zFTS rebuild failed for %s: %sN)r   rM  rK  r   rY   r   r>   r   rollbackr@   rA   )rB   rebuiltr   rC   s       r)   rebuild_ftszSessionSearchMixin.rebuild_ftsQ  s_    Z 	 	'  --c22 
J&&EsEESEEE   J%%'''qLGG/   J'')))NN7c       	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 s:    C	?A+*C	+B9:5B4/C	4B99C		CC)max_commandsr:   rT  c          
         t          |t                    st          |t                    st          d          |dk    rt	          d          || j        }t          |t                    st          |t                    st          d          |dk    rt	          d          d}| j        5  | j        D ]}|                     |          st          | dd          s!| j
                            d	| d
| d           t          |          D ]K}| j
        j        }| j
                            d	| d
| d|f           |dz  }| j
        j        |z
  dk     r nLd| _        ddd           n# 1 swxY w Y   |S )u  Run bounded FTS5 ``'merge'`` commands against each present index.

        A positive merge rank tells SQLite to stop after approximately that
        many output pages, so each command holds the write lock for
        milliseconds regardless of index size — unlike ``'optimize'``, which
        rewrites the whole index in one transaction (measured 9-18 s per
        index on a 10 GB production DB, long enough to exhaust a competing
        writer's entire lock-retry patience).

        Protocol (SQLite FTS5 §6.8-6.9):

        - ``usermerge`` is lowered to its minimum of 2 (persisted in the
          ``%_config`` shadow table, applied once per instance) so a
          positive merge acts on ANY level holding >= 2 segments. With the
          default of 4, levels below that threshold are never merged by a
          positive-rank command and a fragmented index cannot converge.
        - Up to *max_commands* merge commands run per index, stopping early
          on the documented no-progress signal: the delta in
          ``total_changes`` is < 2 (the command's own INSERT accounts
          for 1 change; >= 2 means real merge work happened).

        Each command is its own implicit transaction (the connection runs
        with ``isolation_level=None``), so the SQLite write lock is released
        between commands and competing processes can interleave writes
        mid-pass. Missing tables are valid schema variants (FTS variants are
        optional, and ``optimize_fts_storage`` legitimately drops + backfills
        these tables while writers keep running) and are skipped, mirroring
        ``optimize_fts``. Other SQLite errors propagate to the caller.

        Returns the number of merge commands executed.
        zmax_pages must be an integerr   z#max_pages must be greater than zeroNzmax_commands must be an integerz&max_commands must be greater than zero_fts_usermerge_floor_appliedFr   r   z, rank) VALUES('usermerge', 2)z, rank) VALUES('merge', ?)r   r=  T)r,   r   r\   r.   r2   _FTS_MERGE_COMMANDS_PER_PASSr   rM  rK  getattrr   rY   rangetotal_changesrV  )rB   r:   rT  executedr   r{   befores          r)   r<   z+SessionSearchMixin._merge_fts_incrementallyp  s   D i&& 	<jC.H.H 	<:;;;>>BCCC<LlD)) 	?L#1N1N 	?=>>>1EFFFZ 	5 	5'  --c22 
 t%CUKK J&&1s 1 1S 1 1 1   |,,  A!Z5FJ&&LsLLSLLL"   MHz/&81<< =04D--	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5. s   /B5E00E47E4)r    N)r   r  r  )r8  F)NNNr8  r   NFN)r8  TNNN)5__name__
__module____qualname____doc__r1   classmethodr   r   r-   r	   r8   rD   r   r   re   rv   r   r   r   r   r   r   r   r   r   r   r\   r   r   r   r   r   r  r7  r   rV  staticmethodrg  rx  r{  r  r  r  r  r  r  r  r  rH  rK  rO  rS  r<   r%   r+   r)   r   r   #   s
	       HH%! 
jo.
	%S/	"
 
 
 [
 D D D DVHT#s(^$< V V V V80O 0O 0O 0Od($ ( ( ( (TB$ B B B BHVc3h(@ V V V V$,d , , , ,\O O O O4%$ %$ %$ %$Nt    :d       ( @E - - - - - - - -^4! 4! 4! 4!l'L 'L 'L 'L 'LRBS B B B BN CG	]3 ]3 ]3 hS#X'7'=>?]3 	]3
 
c3h]3 ]3 ]3 ]3F 0Ex
 x
x
 x
 	x

 x
 U38_-x
 
c3hx
 x
 x
 x
z !&	A AA A 	A
 
d38n	A A A AF L!C L!C L! L! L! \L!\ (c (d ( ( ( \( C D    \ Gc Gc G G G [G c d    [" A A A A A \A( , $(%)!%M@ M@ M@M@ 	M@
 M@ M@ CyM@ cM@ #YM@ M@ M@ 
$tCH~&	'M@ M@ M@ M@d $(%)!%!&,00 00 Cy0 c	0
 #Y0 0 0 0 0 C)0 
d38n	0 0 0 0d3 3    2 $(%)!%!&,0{ {{ Cy{ c	{
 #Y{ { { { { C){ 
d38n	{ { { {D "'-1/3+/G' G' G'G' G'
 G'  S	*G' "$s),G' d3i(G' 
d38n	G' G' G' G'X !%!%)32 3232 32 	32
 32 c32 c32 
d38n	32 32 32 32j
c 
d 
 
 
 
%c % % % %NS    @ @DE E EE/7}E	E E E E E Er+   r   )r`  loggingr!  r  r]  r>   r   typingr   r   r   r   r   r   r	   agent.skill_commandsr
   hermes_state_commonr   r   r   r   r   r   r   	getLoggerr@   r   r%   r+   r)   <module>rh     s9      				 				   I I I I I I I I I I I I I I I I I I : : : : : :                  
	>	*	*R" R" R" R" R" R" R" R" R" R"r+   