session-ios/Signal/src/OWSMessagesBubbleImageFacto...

74 lines
2.1 KiB
Swift

//
// Copyright (c) 2017 Open Whisper Systems. All rights reserved.
//
import Foundation
import JSQMessagesViewController
import SignalServiceKit
@objc
class OWSMessagesBubbleImageFactory: NSObject {
private let jsqFactory = JSQMessagesBubbleImageFactory()!
// TODO: UIView is a little bit expensive to instantiate.
// Can we cache this value?
private var isRTL: Bool {
return UIView().isRTL()
}
public var incoming: JSQMessagesBubbleImage {
let color = UIColor.jsq_messageBubbleLightGray()!
return incoming(color: color)
}
public var outgoing: JSQMessagesBubbleImage {
let color = UIColor.ows_materialBlue()
return outgoing(color: color)
}
public var currentlyOutgoing: JSQMessagesBubbleImage {
let color = UIColor.ows_fadedBlue()
return outgoing(color: color)
}
public var outgoingFailed: JSQMessagesBubbleImage {
let color = UIColor.gray
return outgoing(color: color)
}
public func bubble(message: TSMessage) -> JSQMessagesBubbleImage {
if message is TSIncomingMessage {
return self.incoming
} else if let outgoingMessage = message as? TSOutgoingMessage {
switch outgoingMessage.messageState {
case .unsent:
return outgoingFailed
case .attemptingOut:
return currentlyOutgoing
default:
return outgoing
}
} else {
owsFail("Unexpected message type: \(message)")
return outgoing
}
}
private func outgoing(color: UIColor) -> JSQMessagesBubbleImage {
if isRTL {
return jsqFactory.incomingMessagesBubbleImage(with: color)
} else {
return jsqFactory.outgoingMessagesBubbleImage(with: color)
}
}
private func incoming(color: UIColor) -> JSQMessagesBubbleImage {
if isRTL {
return jsqFactory.outgoingMessagesBubbleImage(with: color)
} else {
return jsqFactory.incomingMessagesBubbleImage(with: color)
}
}
}