database.h

Go to the documentation of this file.

#pragma once

 
/* Storage capacities in elements; text buffers include the terminating NUL. */
#define DISCOVERY_DATABASE_SESSION_STRING_BUFFER_SIZE 64
 
#include <ascii-chat/crypto/sizes.h>
#include <stdint.h>
#include <stdbool.h>
#include <sqlite3.h>
// NOTE: Use explicit path to avoid Windows include resolution picking up options/common.h
#include "../common.h"
#include "../discovery/session.h"
#include "discovery-service/main.h"
#include "../network/acip/acds.h"
 
asciichat_error_t database_session_transport_set(sqlite3 *db, const uint8_t session_id[SESSION_ID_SIZE],
                                                 const void *metadata, size_t length);
asciichat_error_t database_session_transport_get(sqlite3 *db, const uint8_t session_id[SESSION_ID_SIZE], void *metadata,
                                                 size_t capacity, size_t *length);

typedef sqlite3 discovery_database_t;
 
// ============================================================================
// Database Lifecycle
// ============================================================================

asciichat_error_t database_init(const char *db_path, sqlite3 **db);

void database_close(sqlite3 *db);
 
// ============================================================================
// Session Operations (SQLite as single source of truth)
// ============================================================================

asciichat_error_t database_session_create(sqlite3 *db, const acip_session_create_t *req, const acds_config_t *config,
                                          acip_session_created_t *resp);

asciichat_error_t database_session_create_keys(sqlite3 *db, const acip_session_create_t *req, const uint8_t (*keys)[32],
                                               size_t count, const acds_config_t *config, acip_session_created_t *resp);

asciichat_error_t database_session_create_keys_with_transport(sqlite3 *db, const acip_session_create_t *req,
                                                              const uint8_t (*keys)[32], size_t count,
                                                              const acds_config_t *config, acip_session_created_t *resp,
                                                              const void *metadata, size_t metadata_length);

asciichat_error_t database_session_lookup(sqlite3 *db, const char *session_string, const acds_config_t *config,
                                          acip_session_info_t *resp);

asciichat_error_t database_session_join(sqlite3 *db, const acip_session_join_t *req, const acds_config_t *config,
                                        acip_session_joined_t *resp);

asciichat_error_t database_session_leave(sqlite3 *db, const uint8_t session_id[SESSION_ID_SIZE],
                                         const uint8_t participant_id[SESSION_ID_SIZE]);

session_entry_t *database_session_find_by_id(sqlite3 *db, const uint8_t session_id[SESSION_ID_SIZE]);

session_entry_t *database_session_find_by_string(sqlite3 *db, const char *session_string);

void database_session_cleanup_expired(sqlite3 *db);

asciichat_error_t database_session_update_host(sqlite3 *db, const uint8_t session_id[SESSION_ID_SIZE],
                                               const uint8_t host_participant_id[SESSION_ID_SIZE],
                                               const char *host_address, uint16_t host_port, uint8_t connection_type);

asciichat_error_t database_session_clear_host(sqlite3 *db, const uint8_t session_id[SESSION_ID_SIZE]);

asciichat_error_t database_session_start_migration(sqlite3 *db, const uint8_t session_id[SESSION_ID_SIZE]);

bool database_session_is_migration_ready(sqlite3 *db, const uint8_t session_id[SESSION_ID_SIZE],
                                         uint64_t migration_window_ms);
 
// ============================================================================
// Multi-Key Management (for session key rotation and multiple identities)
// ============================================================================

asciichat_error_t database_session_add_key(sqlite3 *db, const char *session_string,
                                           const uint8_t identity_pubkey[ED25519_PUBLIC_KEY_SIZE],
                                           uint32_t key_version);

asciichat_error_t database_session_revoke_key(sqlite3 *db, const char *session_string,
                                              const uint8_t identity_pubkey[ED25519_PUBLIC_KEY_SIZE]);

bool database_session_verify_key(sqlite3 *db, const char *session_string,
                                 const uint8_t identity_pubkey[ED25519_PUBLIC_KEY_SIZE]);

asciichat_error_t database_session_get_keys(sqlite3 *db, const char *session_string, uint8_t (*keys_out)[32],
                                            size_t max_keys, size_t *count_out);
 
// ============================================================================
// Statistics Queries
// ============================================================================

asciichat_error_t database_stats_session_count(sqlite3 *db, int64_t *count_out);

asciichat_error_t database_stats_unique_clients(sqlite3 *db, int64_t *count_out);

typedef struct {
  char session_string[DISCOVERY_DATABASE_SESSION_STRING_BUFFER_SIZE];
  uint32_t participants;
  bool host_ready;
  bool migrating;
} database_session_stats_t;
asciichat_error_t database_stats_snapshot(sqlite3 *db, database_session_stats_t *rows, size_t capacity,
                                          size_t *row_count, uint64_t *sessions, uint64_t *participants);
 
asciichat_error_t database_session_update_host_with_transport(sqlite3 *db, const uint8_t session[SESSION_ID_SIZE],
                                                              const uint8_t host[SESSION_ID_SIZE], const char *address,
                                                              uint16_t port, uint8_t type, const void *metadata,
                                                              size_t length);