/*
* Copyright (c) 2020, 2021 Jonathan Schleifer <js@nil.im>
*
* https://fl.nil.im/objmatrix
*
* Permission to use, copy, modify, and/or distribute this software for any
* purpose with or without fee is hereby granted, provided that the above
* copyright notice and this permission notice appear in all copies.
*
* THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES WITH
* REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
* AND FITNESS. IN NO EVENT SHALL ISC BE LIABLE FOR ANY SPECIAL, DIRECT,
* INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
* LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
* OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
* PERFORMANCE OF THIS SOFTWARE.
*/
#import <ObjSQLite3/ObjSQLite3.h>
#import "MTXSQLite3Storage.h"
@implementation MTXSQLite3Storage
{
SL3Connection *_conn;
SL3PreparedStatement *_nextBatchSetStatement, *_nextBatchGetStatement;
SL3PreparedStatement *_joinedRoomsAddStatement;
SL3PreparedStatement *_joinedRoomsRemoveStatement;
SL3PreparedStatement *_joinedRoomsGetStatement;
}
+ (instancetype)storageWithPath: (OFString *)path
{
return [[[self alloc] initWithPath: path] autorelease];
}
- (instancetype)initWithPath: (OFString *)path
{
self = [super init];
@try {
void *pool = objc_autoreleasePoolPush();
_conn = [[SL3Connection alloc] initWithPath: path];
[self createTables];
_nextBatchSetStatement = [[_conn prepareStatement:
@"INSERT OR REPLACE INTO next_batch (\n"
@" device_id, next_batch\n"
@") VALUES (\n"
@" $device_id, $next_batch\n"
@")"] retain];
_nextBatchGetStatement = [[_conn prepareStatement:
@"SELECT next_batch FROM next_batch\n"
@"WHERE device_id=$device_id"] retain];
_joinedRoomsAddStatement = [[_conn prepareStatement:
@"INSERT OR REPLACE INTO joined_rooms (\n"
@" user_id, room_id\n"
@") VALUES (\n"
@" $user_id, $room_id\n"
@")"] retain];
_joinedRoomsRemoveStatement = [[_conn prepareStatement:
@"DELETE FROM joined_rooms\n"
@"WHERE user_id=$user_id AND room_id=$room_id"] retain];
_joinedRoomsGetStatement = [[_conn prepareStatement:
@"SELECT room_id FROM joined_rooms\n"
@"WHERE user_id=$user_id"] retain];
objc_autoreleasePoolPop(pool);
} @catch (id e) {
[self release];
@throw e;
}
return self;
}
- (void)dealloc
{
[_nextBatchSetStatement release];
[_nextBatchGetStatement release];
[_joinedRoomsAddStatement release];
[_joinedRoomsRemoveStatement release];
[_joinedRoomsGetStatement release];
[_conn release];
[super dealloc];
}
- (void)createTables
{
[_conn executeStatement:
@"CREATE TABLE IF NOT EXISTS next_batch (\n"
@" device_id TEXT PRIMARY KEY,\n"
@" next_batch TEXT\n"
@");\n"
@"CREATE TABLE IF NOT EXISTS joined_rooms (\n"
@" user_id TEXT,\n"
@" room_id TEXT,\n"
@" PRIMARY KEY (user_id, room_id)\n"
@");"];
}
- (void)transactionWithBlock: (MTXStorageTransactionBlock)block
{
[_conn transactionWithBlock: block];
}
- (void)setNextBatch: (OFString *)nextBatch forDeviceID: (OFString *)deviceID
{
void *pool = objc_autoreleasePoolPush();
[_nextBatchSetStatement reset];
[_nextBatchSetStatement bindWithDictionary: @{
@"$device_id": deviceID,
@"$next_batch": nextBatch
}];
[_nextBatchSetStatement step];
objc_autoreleasePoolPop(pool);
}
- (OFString *)nextBatchForDeviceID: (OFString *)deviceID
{
void *pool = objc_autoreleasePoolPush();
[_nextBatchGetStatement reset];
[_nextBatchGetStatement bindWithDictionary: @{
@"$device_id": deviceID
}];
if (![_nextBatchGetStatement step])
return nil;
OFString *nextBatch =
[_nextBatchGetStatement.rowDictionary[@"next_batch"] retain];
objc_autoreleasePoolPop(pool);
return [nextBatch autorelease];
}
- (void)addJoinedRoom: (OFString *)roomID forUser: (OFString *)userID
{
void *pool = objc_autoreleasePoolPush();
[_joinedRoomsAddStatement reset];
[_joinedRoomsAddStatement bindWithDictionary: @{
@"$room_id": roomID,
@"$user_id": userID
}];
[_joinedRoomsAddStatement step];
objc_autoreleasePoolPop(pool);
}
- (void)removeJoinedRoom: (OFString *)roomID forUser: (OFString *)userID
{
void *pool = objc_autoreleasePoolPush();
[_joinedRoomsRemoveStatement reset];
[_joinedRoomsRemoveStatement bindWithDictionary: @{
@"$room_id": roomID,
@"$user_id": userID
}];
[_joinedRoomsRemoveStatement step];
objc_autoreleasePoolPop(pool);
}
- (OFArray<OFString *> *)joinedRoomsForUser: (OFString *)userID
{
OFMutableArray *joinedRooms = [OFMutableArray array];
void *pool = objc_autoreleasePoolPush();
[_joinedRoomsGetStatement reset];
[_joinedRoomsGetStatement bindWithDictionary: @{ @"$user_id": userID }];
while ([_joinedRoomsGetStatement step])
[joinedRooms addObject:
_joinedRoomsGetStatement.rowDictionary[@"room_id"]];
objc_autoreleasePoolPop(pool);
return [joinedRooms autorelease];
}
@end