Method AddSubSession

pkg/session/store.go:1080–1118

AddSubSession creates a sub-session and links it to the parent.

(ctx context.Context, parentSessionID string, subSession *Session)

// AddSubSession creates a sub-session and links it to the parent.
func (s *SQLiteSessionStore) AddSubSession(ctx context.Context, parentSessionID string, subSession *Session) error {
	if parentSessionID == "" || subSession.ID == "" {
		return ErrEmptyID
	}

	tx, err := s.db.BeginTx(ctx, nil)
	if err != nil {
		return err
	}
	defer func() {
		_ = tx.Rollback()
	}()

	// 1. Set parent_id on sub-session
	subSession.ParentID = parentSessionID

	// 2. Insert sub-session as a new session row
	if err := s.addSessionTx(ctx, tx, subSession); err != nil {
		return fmt.Errorf("inserting sub-session: %w", err)
	}

	// 3. Recursively add all items from the sub-session
	for i, item := range subSession.Messages {
		if err := s.addItemTx(ctx, tx, subSession.ID, i, item); err != nil {
			return fmt.Errorf("inserting sub-session item %d: %w", i, err)
		}
	}

	// 4. Add reference in parent's items
	_, err = tx.ExecContext(ctx,
		`INSERT INTO session_items (session_id, position, item_type, subsession_id)
		 VALUES (?, (SELECT COALESCE(MAX(position), -1) + 1 FROM session_items WHERE session_id = ?), 'subsession', ?)`,
		parentSessionID, parentSessionID, subSession.ID)
	if err != nil {
		return fmt.Errorf("inserting subsession reference: %w", err)
	}

	return tx.Commit()
}

// addSessionTx inserts a session within a transaction.
func (s *SQLiteSessionStore) addSessionTx(ctx context.Context, tx *sql.Tx, session *Session) error {