Something went wrong. Try again.
Monorepo for Aesthetic.Computer aesthetic.computer
Something went wrong. Try again.
Swift
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164//// MessagesViewController.swift// aesthetic//// Created by Rumi Tarighian on 12/5/23.//
import UIKitimport Messagesimport WebKit
class MessagesViewController: MSMessagesAppViewController, WKScriptMessageHandler{ func userContentController(_ userContentController: WKUserContentController, didReceive message: WKScriptMessage) { if message.name == "iMessageExtension", let jsonString = message.body as? String { print("Received iMessageExtension: \(jsonString)") // Convert JSON string to Dictionary if let data = jsonString.data(using: .utf8) { do { if let dictionary = try JSONSerialization.jsonObject(with: data, options: []) as? [String: Any] { // Now 'dictionary' is a Swift dictionary print("JSON as dictionary: \(dictionary)") // Handle the dictionary as needed if let type = dictionary["type"] as? String, type == "tap" { if let image = convertDataUrlToImage(dataUrl: dictionary["body"] as! String) { displayFixedImageSticker(picture: image) } else { print("Image not found in the assets.") } } } } catch { print("Error parsing JSON: \(error)") } } } else if message.name == "iMessageExtensionLog" { print("JavaScript Log: \(message.body)") } } func convertDataUrlToImage(dataUrl: String) -> UIImage? { // Check if the string is a valid data URL guard dataUrl.hasPrefix("data:image/png;base64,"), let base64String = dataUrl.split(separator: ",").last, let imageData = Data(base64Encoded: String(base64String)) else { print("Invalid data URL") return nil }
// Create an image from the data let image = UIImage(data: imageData) return image } var webView: WKWebView! let grey: CGFloat = 32/255; override func viewDidLoad() { super.viewDidLoad() let config = WKWebViewConfiguration() let userScript = WKUserScript(source: "console.log = function() { window.webkit.messageHandlers.iMessageExtensionLog.postMessage([...arguments].join(' ')); }", injectionTime: .atDocumentStart, forMainFrameOnly: false) config.userContentController.addUserScript(userScript) config.userContentController.add(self, name: "iMessageExtension") config.userContentController.add(self, name: "iMessageExtensionLog") webView = WKWebView(frame: view.bounds, configuration: config) webView.load(URLRequest(url: URL(string: "https://aesthetic.computer/imessage")!)) webView.customUserAgent = "AestheticExtension" webView.backgroundColor = UIColor(red: grey, green: grey, blue: grey, alpha: 1) webView.isOpaque = false view.addSubview(webView) webView.backgroundColor = .black } func displayFixedImageSticker(picture: UIImage) { // Ensure there is an active conversation guard let conversation = activeConversation else { print("Active conversation is nil") return }
let tempDirectory = NSTemporaryDirectory() let tempFileURL = URL(fileURLWithPath: tempDirectory).appendingPathComponent("tempImage.png")
do { // Write the image data to the temporary file if let pngData = picture.pngData() { try pngData.write(to: tempFileURL) // Insert the attachment conversation.insertAttachment(tempFileURL, withAlternateFilename: nil, completionHandler: { error in if let error = error { print("Error inserting attachment: \(error)") } }) } else { print("Failed to get PNG data from image") } } catch { print("Failed to write image to temporary file: \(error)") } } override func willBecomeActive(with conversation: MSConversation) { // Called when the extension is about to move from the inactive to active state. // This will happen when the extension is about to present UI. // Use this method to configure the extension and restore previously stored state. } override func didResignActive(with conversation: MSConversation) { // Called when the extension is about to move from the active to inactive state. // This will happen when the user dismisses the extension, changes to a different // conversation or quits Messages. // Use this method to release shared resources, save user data, invalidate timers, // and store enough state information to restore your extension to its current state // in case it is terminated later. } override func didReceive(_ message: MSMessage, conversation: MSConversation) { // Called when a message arrives that was generated by another instance of this // extension on a remote device. // Use this method to trigger UI updates in response to the message. } override func didStartSending(_ message: MSMessage, conversation: MSConversation) { // Called when the user taps the send button. } override func didCancelSending(_ message: MSMessage, conversation: MSConversation) { // Called when the user deletes the message without sending it. // Use this to clean up state related to the deleted message. } override func willTransition(to presentationStyle: MSMessagesAppPresentationStyle) { // Called before the extension transitions to a new presentation style. // Use this method to prepare for the change in presentation style. } override func didTransition(to presentationStyle: MSMessagesAppPresentationStyle) { super.didTransition(to: presentationStyle) switch presentationStyle { case .compact: print("Extension has contracted") webView.evaluateJavaScript("iMessageExtensionResize('contract')", completionHandler: nil) case .expanded: print("Extension has expanded") webView.evaluateJavaScript("iMessageExtensionResize('expand')", completionHandler: nil) case .transcript: print("whatever...") @unknown default: fatalError("Unknown presentation style encountered") } } }