Signal-iOS/SignalServiceKit/src/Devices/OWSStickerPackSyncMessage.m
2021-06-07 11:11:30 -04:00

76 lines
2.2 KiB
Objective-C

//
// Copyright (c) 2021 Open Whisper Systems. All rights reserved.
//
#import <SignalServiceKit/OWSStickerPackSyncMessage.h>
#import <SignalServiceKit/SignalServiceKit-Swift.h>
NS_ASSUME_NONNULL_BEGIN
@interface OWSStickerPackSyncMessage ()
@property (nonatomic, readonly) NSArray<StickerPackInfo *> *packs;
@property (nonatomic, readonly) StickerPackOperationType operationType;
@end
#pragma mark -
@implementation OWSStickerPackSyncMessage
- (nullable instancetype)initWithCoder:(NSCoder *)coder
{
return [super initWithCoder:coder];
}
- (instancetype)initWithThread:(TSThread *)thread
packs:(NSArray<StickerPackInfo *> *)packs
operationType:(StickerPackOperationType)operationType
{
self = [super initWithThread:thread];
if (!self) {
return self;
}
_packs = packs;
_operationType = operationType;
return self;
}
- (nullable SSKProtoSyncMessageBuilder *)syncMessageBuilderWithTransaction:(SDSAnyReadTransaction *)transaction
{
SSKProtoSyncMessageStickerPackOperationType operationType;
switch (self.operationType) {
case StickerPackOperationType_Install:
operationType = SSKProtoSyncMessageStickerPackOperationTypeInstall;
break;
case StickerPackOperationType_Remove:
operationType = SSKProtoSyncMessageStickerPackOperationTypeRemove;
break;
}
SSKProtoSyncMessageBuilder *syncMessageBuilder = [SSKProtoSyncMessage builder];
for (StickerPackInfo *pack in self.packs) {
SSKProtoSyncMessageStickerPackOperationBuilder *packOperationBuilder =
[SSKProtoSyncMessageStickerPackOperation builderWithPackID:pack.packId packKey:pack.packKey];
[packOperationBuilder setType:operationType];
NSError *error;
SSKProtoSyncMessageStickerPackOperation *_Nullable packOperationProto =
[packOperationBuilder buildAndReturnError:&error];
if (error || !packOperationProto) {
OWSFailDebug(@"could not build protobuf: %@", error);
return nil;
}
[syncMessageBuilder addStickerPackOperation:packOperationProto];
}
return syncMessageBuilder;
}
@end
NS_ASSUME_NONNULL_END