Skip to content

feat: add filter hooks to fully override daily memory storage backend #1050

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
chubes4 wants to merge 1 commit into from
Closed

feat: add filter hooks to fully override daily memory storage backend #1050

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
149 changes: 130 additions & 19 deletions inc/Abilities/DailyMemoryAbilities.php
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,14 @@
* Daily Memory Abilities
*
* WordPress 6.9 Abilities API primitives for daily memory operations.
* Provides read/write/list access to daily memory files (YYYY/MM/DD.md).
* Provides read/write/list/search/delete access to daily memory.
*
* Each operation fires a filter (`datamachine_daily_memory_{operation}`) that
* allows plugins to completely replace the storage backend. If the filter
* returns a non-null value, the flat-file operation is skipped entirely.
* This enables consumers like Intelligence to store daily memory as
* WordPress pages, external services, or anything else — Data Machine
* doesn't need to know or care.
*
* @package DataMachine\Abilities
* @since 0.32.0
Expand Down Expand Up @@ -250,10 +257,7 @@ private function registerAbilities(): void {
* @return array Result.
*/
public static function readDaily( array $input ): array {
$user_id = (int) ( $input['user_id'] ?? 0 );
$agent_id = (int) ( $input['agent_id'] ?? 0 );
$daily = new DailyMemory( $user_id, $agent_id );
$date = $input['date'] ?? gmdate( 'Y-m-d' );
$date = $input['date'] ?? gmdate( 'Y-m-d' );

$parts = DailyMemory::parse_date( $date );
if ( ! $parts ) {
Expand All @@ -263,6 +267,30 @@ public static function readDaily( array $input ): array {
);
}

/**
* Filters the daily memory read result.
*
* Return a non-null array to completely replace the flat-file read.
* The array should match the standard result format:
* `{ success: bool, date?: string, content?: string, message?: string }`
*
* @since 0.47.0
*
* @param array|null $result Default null (flat-file read proceeds).
* Return an array to short-circuit.
* @param string $date The requested date (YYYY-MM-DD).
* @param array $input Full input parameters (includes user_id, agent_id).
*/
$result = apply_filters( 'datamachine_daily_memory_read', null, $date, $input );

if ( null !== $result ) {
return $result;
}

$user_id = (int) ( $input['user_id'] ?? 0 );
$agent_id = (int) ( $input['agent_id'] ?? 0 );
$daily = new DailyMemory( $user_id, $agent_id );

return $daily->read( $parts['year'], $parts['month'], $parts['day'] );
}

Expand All @@ -280,12 +308,9 @@ public static function writeDaily( array $input ): array {
);
}

$user_id = (int) ( $input['user_id'] ?? 0 );
$agent_id = (int) ( $input['agent_id'] ?? 0 );
$daily = new DailyMemory( $user_id, $agent_id );
$content = $input['content'];
$date = $input['date'] ?? gmdate( 'Y-m-d' );
$mode = $input['mode'] ?? 'append';
$date = $input['date'] ?? gmdate( 'Y-m-d' );
$content = $input['content'];
$mode = $input['mode'] ?? 'append';

$parts = DailyMemory::parse_date( $date );
if ( ! $parts ) {
Expand All @@ -295,6 +320,31 @@ public static function writeDaily( array $input ): array {
);
}

/**
* Filters the daily memory write result.
*
* Return a non-null array to completely replace the flat-file write.
* The array should match: `{ success: bool, message?: string }`
*
* @since 0.47.0
*
* @param array|null $result Default null (flat-file write proceeds).
* Return an array to short-circuit.
* @param string $date The target date (YYYY-MM-DD).
* @param string $content The content to write.
* @param string $mode Write mode: 'write' or 'append'.
* @param array $input Full input parameters (includes user_id, agent_id).
*/
$result = apply_filters( 'datamachine_daily_memory_write', null, $date, $content, $mode, $input );

if ( null !== $result ) {
return $result;
}

$user_id = (int) ( $input['user_id'] ?? 0 );
$agent_id = (int) ( $input['agent_id'] ?? 0 );
$daily = new DailyMemory( $user_id, $agent_id );

if ( 'write' === $mode ) {
return $daily->write( $parts['year'], $parts['month'], $parts['day'], $content );
}
Expand All @@ -305,10 +355,28 @@ public static function writeDaily( array $input ): array {
/**
* List all daily memory files.
*
* @param array $input Input parameters (unused).
* @param array $input Input parameters.
* @return array Result.
*/
public static function listDaily( array $input ): array {
/**
* Filters the daily memory list result.
*
* Return a non-null array to completely replace the flat-file listing.
* The array should match: `{ months: { 'YYYY/MM': ['DD', ...], ... } }`
*
* @since 0.47.0
*
* @param array|null $result Default null (flat-file list proceeds).
* Return an array to short-circuit.
* @param array $input Full input parameters (includes user_id, agent_id).
*/
$result = apply_filters( 'datamachine_daily_memory_list', null, $input );

if ( null !== $result ) {
return $result;
}

$user_id = (int) ( $input['user_id'] ?? 0 );
$agent_id = (int) ( $input['agent_id'] ?? 0 );
$daily = new DailyMemory( $user_id, $agent_id );
Expand All @@ -322,12 +390,35 @@ public static function listDaily( array $input ): array {
* @return array Search results.
*/
public static function searchDaily( array $input ): array {
$query = $input['query'];
$from = $input['from'] ?? null;
$to = $input['to'] ?? null;

/**
* Filters the daily memory search result.
*
* Return a non-null array to completely replace the flat-file search.
* The array should match:
* `{ success: bool, query: string, match_count: int, matches: [...] }`
*
* @since 0.47.0
*
* @param array|null $result Default null (flat-file search proceeds).
* Return an array to short-circuit.
* @param string $query The search term.
* @param string|null $from Start date (YYYY-MM-DD) or null.
* @param string|null $to End date (YYYY-MM-DD) or null.
* @param array $input Full input parameters (includes user_id, agent_id).
*/
$result = apply_filters( 'datamachine_daily_memory_search', null, $query, $from, $to, $input );

if ( null !== $result ) {
return $result;
}

$user_id = (int) ( $input['user_id'] ?? 0 );
$agent_id = (int) ( $input['agent_id'] ?? 0 );
$daily = new DailyMemory( $user_id, $agent_id );
$query = $input['query'];
$from = $input['from'] ?? null;
$to = $input['to'] ?? null;

return $daily->search( $query, $from, $to );
}
Expand All @@ -348,10 +439,7 @@ public static function deleteDaily( array $input ): array {
);
}

$user_id = (int) ( $input['user_id'] ?? 0 );
$agent_id = (int) ( $input['agent_id'] ?? 0 );
$daily = new DailyMemory( $user_id, $agent_id );
$date = $input['date'] ?? gmdate( 'Y-m-d' );
$date = $input['date'] ?? gmdate( 'Y-m-d' );

$parts = DailyMemory::parse_date( $date );
if ( ! $parts ) {
Expand All @@ -361,6 +449,29 @@ public static function deleteDaily( array $input ): array {
);
}

/**
* Filters the daily memory delete result.
*
* Return a non-null array to completely replace the flat-file delete.
* The array should match: `{ success: bool, message?: string }`
*
* @since 0.47.0
*
* @param array|null $result Default null (flat-file delete proceeds).
* Return an array to short-circuit.
* @param string $date The target date (YYYY-MM-DD).
* @param array $input Full input parameters (includes user_id, agent_id).
*/
$result = apply_filters( 'datamachine_daily_memory_delete', null, $date, $input );

if ( null !== $result ) {
return $result;
}

$user_id = (int) ( $input['user_id'] ?? 0 );
$agent_id = (int) ( $input['agent_id'] ?? 0 );
$daily = new DailyMemory( $user_id, $agent_id );

return $daily->delete( $parts['year'], $parts['month'], $parts['day'] );
}
}
33 changes: 3 additions & 30 deletions inc/Engine/AI/System/Tasks/DailyMemoryTask.php
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -226,6 +226,9 @@ public function execute( int $jobId, array $params ): void {
* an external service, etc.). When `true` is returned, the flat-file
* write to `daily/YYYY/MM/DD.md` is skipped entirely.
*
* For a complete storage backend override (read, list, search, delete),
* see the `datamachine_daily_memory_*` filters in DailyMemoryAbilities.
*
* @since 0.46.0
*
* @param bool $handled Whether a handler has already stored the content.
Expand Down Expand Up @@ -256,36 +259,6 @@ public function execute( int $jobId, array $params ): void {

$daily->append( $parts[0], $parts[1], $parts[2], $archive_text );
}

/**
* Fires after daily memory archived content has been processed.
*
* This action fires regardless of whether the content was written to a
* flat file or handled by a `datamachine_daily_memory_pre_archive` filter.
* Use this for post-processing, notifications, or secondary storage.
*
* @since 0.46.0
*
* @param string $content The archived content extracted from MEMORY.md.
* @param string $date The archive date (YYYY-MM-DD).
* @param bool $handled Whether a filter handled storage (true = flat file skipped).
* @param array $context {
* Additional context about the daily memory operation.
*
* @type string $persistent The persistent content remaining in MEMORY.md.
* @type int $original_size Original MEMORY.md size in bytes.
* @type int $new_size New MEMORY.md size in bytes after cleanup.
* @type int $archived_size Archived content size in bytes.
* @type int $job_id The job ID for this task execution.
* }
*/
do_action(
'datamachine_daily_memory_archived',
$parsed['archived'],
$date,
$handled,
$archive_context
);
}

do_action(
Expand Down
Loading