@ -1,26 +1,41 @@
import FeedKit
// TODO: M o v e t h e R S S f e e d l o g i c i n t o i t s o w n f i l e
@objc ( LKGroupChatPoller )
public final class LokiGroupChatPoller : NSObject {
private let group : UInt
private let server : String
private let group : LokiGroupChat
private var pollForNewMessagesTimer : Timer ? = nil
private var pollForDeletedMessagesTimer : Timer ? = nil
private var hasStarted = false
private let pollForNewMessagesInterval : TimeInterval = 4
private let pollForDeletedMessagesInterval : TimeInterval = 120
private lazy var pollForNewMessagesInterval : TimeInterval = {
switch group . kind {
case . publicChat ( _ ) : return 4
case . rss ( _ ) : return 8 * 60
}
} ( )
@objc ( initForGroup : onServer : )
public init ( for group : UInt , on server : String ) {
private lazy var pollForDeletedMessagesInterval : TimeInterval = {
switch group . kind {
case . publicChat ( _ ) : return 32 * 60
case . rss ( _ ) : preconditionFailure ( )
}
} ( )
@objc ( initForGroup : )
public init ( for group : LokiGroupChat ) {
self . group = group
self . server = server
super . init ( )
}
@objc public func startIfNeeded ( ) {
if hasStarted { return }
pollForNewMessagesTimer = Timer . scheduledTimer ( withTimeInterval : pollForNewMessagesInterval , repeats : true ) { [ weak self ] _ in self ? . pollForNewMessages ( ) }
pollForDeletedMessagesTimer = Timer . scheduledTimer ( withTimeInterval : pollForDeletedMessagesInterval , repeats : true ) { [ weak self ] _ in self ? . pollForDeletedMessages ( ) }
pollForNewMessages ( ) // P e r f o r m i n i t i a l u p d a t e
if group . isPublicChat {
pollForDeletedMessagesTimer = Timer . scheduledTimer ( withTimeInterval : pollForDeletedMessagesInterval , repeats : true ) { [ weak self ] _ in self ? . pollForDeletedMessages ( ) }
}
hasStarted = true
}
@ -32,28 +47,57 @@ public final class LokiGroupChatPoller : NSObject {
private func pollForNewMessages ( ) {
let group = self . group
let server = self . server
let _ = LokiGroupChatAPI . getMessages ( for : group , on : server ) . map { messages in
messages . reversed ( ) . map { message in
let id = " \( server ) . \( group ) " . data ( using : String . Encoding . utf8 ) !
let x1 = SSKProtoGroupContext . builder ( id : id , type : . deliver )
x1 . setName ( NSLocalizedString ( " Loki Public Chat " , comment : " " ) )
let x2 = SSKProtoDataMessage . builder ( )
x2 . setTimestamp ( message . timestamp )
x2 . setGroup ( try ! x1 . build ( ) )
x2 . setBody ( message . body )
let x3 = SSKProtoContent . builder ( )
x3 . setDataMessage ( try ! x2 . build ( ) )
let x4 = SSKProtoEnvelope . builder ( type : . ciphertext , timestamp : message . timestamp )
let senderHexEncodedPublicKey = message . hexEncodedPublicKey
let endIndex = senderHexEncodedPublicKey . endIndex
let cutoffIndex = senderHexEncodedPublicKey . index ( endIndex , offsetBy : - 8 )
let senderDisplayName = " \( message . displayName ) (... \( senderHexEncodedPublicKey [ cutoffIndex . . < endIndex ] ) ) "
x4 . setSource ( senderDisplayName )
x4 . setSourceDevice ( OWSDevicePrimaryDeviceId )
x4 . setContent ( try ! x3 . build ( ) . serializedData ( ) )
OWSPrimaryStorage . shared ( ) . dbReadWriteConnection . readWrite { transaction in
SSKEnvironment . shared . messageManager . throws_processEnvelope ( try ! x4 . build ( ) , plaintextData : try ! x3 . build ( ) . serializedData ( ) , wasReceivedByUD : false , transaction : transaction )
func parseGroupMessage ( body : String , timestamp : UInt64 , senderDisplayName : String ) {
let id = group . id . data ( using : String . Encoding . utf8 ) !
let x1 = SSKProtoGroupContext . builder ( id : id , type : . deliver )
x1 . setName ( group . displayName )
let x2 = SSKProtoDataMessage . builder ( )
x2 . setTimestamp ( timestamp )
x2 . setGroup ( try ! x1 . build ( ) )
x2 . setBody ( body )
let x3 = SSKProtoContent . builder ( )
x3 . setDataMessage ( try ! x2 . build ( ) )
let x4 = SSKProtoEnvelope . builder ( type : . ciphertext , timestamp : timestamp )
x4 . setSource ( senderDisplayName )
x4 . setSourceDevice ( OWSDevicePrimaryDeviceId )
x4 . setContent ( try ! x3 . build ( ) . serializedData ( ) )
OWSPrimaryStorage . shared ( ) . dbReadWriteConnection . readWrite { transaction in
SSKEnvironment . shared . messageManager . throws_processEnvelope ( try ! x4 . build ( ) , plaintextData : try ! x3 . build ( ) . serializedData ( ) , wasReceivedByUD : false , transaction : transaction )
}
}
switch group . kind {
case . publicChat ( let id ) :
let _ = LokiGroupChatAPI . getMessages ( for : id , on : group . server ) . done { messages in
messages . reversed ( ) . forEach { message in
let senderHexEncodedPublicKey = message . hexEncodedPublicKey
let endIndex = senderHexEncodedPublicKey . endIndex
let cutoffIndex = senderHexEncodedPublicKey . index ( endIndex , offsetBy : - 8 )
let senderDisplayName = " \( message . displayName ) (... \( senderHexEncodedPublicKey [ cutoffIndex . . < endIndex ] ) ) "
parseGroupMessage ( body : message . body , timestamp : message . timestamp , senderDisplayName : senderDisplayName )
}
}
case . rss ( _ ) :
let url = URL ( string : group . server ) !
FeedParser ( URL : url ) . parseAsync { wrapper in
guard case . rss ( let feed ) = wrapper , let items = feed . items else { return print ( " [Loki] Failed to parse RSS feed for: \( group . server ) " ) }
items . reversed ( ) . forEach { item in
guard let title = item . title , let description = item . description , let date = item . pubDate else { return }
let timestamp = UInt64 ( date . timeIntervalSince1970 * 1000 )
let regex = try ! NSRegularExpression ( pattern : " <a \\ s+(?:[^>]*? \\ s+)?href= \" ([^ \" ]*) \" .*?>(.*?)<.*? \\ /a> " )
var bodyAsHTML = " <b> \( title ) </b> \( description ) "
while true {
guard let match = regex . firstMatch ( in : bodyAsHTML , options : [ ] , range : NSRange ( location : 0 , length : bodyAsHTML . utf16 . count ) ) else { break }
let matchRange = match . range ( at : 0 )
let urlRange = match . range ( at : 1 )
let descriptionRange = match . range ( at : 2 )
let url = ( bodyAsHTML as NSString ) . substring ( with : urlRange )
let description = ( bodyAsHTML as NSString ) . substring ( with : descriptionRange )
bodyAsHTML = ( bodyAsHTML as NSString ) . replacingCharacters ( in : matchRange , with : " \( description ) ( \( url ) ) " ) as String
}
guard let bodyAsData = bodyAsHTML . data ( using : String . Encoding . unicode ) else { return }
let options = [ NSAttributedString . DocumentReadingOptionKey . documentType : NSAttributedString . DocumentType . html ]
guard let body = try ? NSAttributedString ( data : bodyAsData , options : options , documentAttributes : nil ) else { return }
parseGroupMessage ( body : body . string , timestamp : timestamp , senderDisplayName : NSLocalizedString ( " Loki " , comment : " " ) )
}
}
}
gmbnt
6 years ago
committed by
GitHub