如何从 Objective-C 的 WKWebView 获取选定的文本

Question

我有一个 WKWebView。

当用户右键单击它时，我可以在 Objective-C 方法中自定义上下文菜单。仅当用户在 WKWebView 中选择了某些文本时，我才想添加菜单项。当然，我稍后需要检索所选文本来处理它。

如何从 Objective-C 的 WKWebView 检索选择，确保它只是文本并获取该文本？

谢谢

Answer 1

这就是我如何做到这一点的。这不是一个完美的解决方案，但已经足够好了。

一般说明

看来WKWebView内部发生的任何事情都必须用JavaScript来管理。 Apple 提供了一个在 JavaScript 世界和 Objective-C（或 Swift）世界之间交换信息的框架。该框架基于从 JavaScript 世界发送的一些消息，并通过可安装在 WKWebView 中的消息处理程序在 Objective-C（或 Swift）世界中捕获。

第一步 - 安装消息处理程序

在 Objective-C（或 Swift）世界中，定义一个对象，负责从 JavaScript 世界接收消息。我为此使用了视图控制器。下面的代码将视图控制器安装为“用户内容控制器”，它将接收可从 JavaScript 发送的名为“newSelectionDetected”的事件

- (void)viewDidLoad
{
    [super viewDidLoad];

    //  Add self as scriptMessageHandler of the webView
    WKUserContentController *controller = self.webView.configuration.userContentController ;
    [controller addScriptMessageHandler:self
                                   name:@"newSelectionDetected"] ;
    ... the rest will come further down...

第二步 - 在视图中安装 JavaScript

此 JavaScript 将检测选择更改，并通过名为“newSelectionDetected”的消息发送新选择

- (void)    viewDidLoad
{
    ...See first part up there...

    NSURL       *scriptURL      = .. URL to file DetectSelection.js...
    NSString    *scriptString   = [NSString stringWithContentsOfURL:scriptURL
                                                           encoding:NSUTF8StringEncoding
                                                              error:NULL] ;

    WKUserScript    *script = [[WKUserScript alloc] initWithSource:scriptString
                                                     injectionTime:WKUserScriptInjectionTimeAtDocumentEnd
                                                  forMainFrameOnly:YES] ;
    [controller addUserScript:script] ;
}

和 JavaScript：

function getSelectionAndSendMessage()
{
    var txt = document.getSelection().toString() ;
    window.webkit.messageHandlers.newSelectionDetected.postMessage(txt) ;
}
document.onmouseup = getSelectionAndSendMessage ;
document.onkeyup   = getSelectionAndSendMessage ;
document.oncontextmenu  = getSelectionAndSendMessage ;

第三步-接收并处理事件

现在，每次我们在 WKWebView 中按下鼠标或按下按键时，选择（可能是空的）都会被捕获并通过消息发送到 Objective-C 世界。

我们只需要视图控制器中的处理程序来处理该消息

- (void)    userContentController:(WKUserContentController*)userContentController
          didReceiveScriptMessage:(WKScriptMessage*)message
{
    // A new selected text has been received
    if ([message.body isKindOfClass:[NSString class]])
    {
        ...Do whatever you want with message.body which is an NSString...
    }
}

我创建了一个继承自 WKWebView 的类，并具有 NSString 属性“selectedText”。所以我在这个处理程序中所做的就是将接收到的 NSString 存储在这个属性中。

第四步 - 更新上下文菜单

在我的 WKWebView 子类中，如果 selectedText 不为空，我只是重写 willOpenMenu:WithEvent: 方法来添加菜单项。

- (void)    willOpenMenu:(NSMenu*)menu withEvent:(NSEvent*)event
{
    if ([self.selectedText length]>0)
    {
        NSMenuItem  *item   = [[NSMenuItem alloc] initWithTitle:@"That works !!!"
                                                         action:@selector(myAction:)
                                                  keyEquivalent:@""] ;
        item.target = self ;
        [menu addItem:item] ;
    }
}

- (IBAction)    myAction:(id)sender
{
    NSLog(@"tadaaaa !!!") ;
}

为什么这不理想呢？好吧，如果你的网页已经设置了 onmouseup 或 onkeyup，我会覆盖它。

但正如我所说，对我来说已经足够了。

编辑：我在 JavaScript 中添加了 document.oncontextmenu 行，这解决了我有时遇到的奇怪选择行为。

Answer 2

Swift 5 翻译

webView.configuration.userContentController.add(self, name: "newSelectionDetected")
let scriptString = """
    function getSelectionAndSendMessage()
    {
        var txt = document.getSelection().toString() ;
        window.webkit.messageHandlers.newSelectionDetected.postMessage(txt);
    }
    document.onmouseup = getSelectionAndSendMessage;
    document.onkeyup = getSelectionAndSendMessage;
    document.oncontextmenu = getSelectionAndSendMessage;
"""
let script = WKUserScript(source: scriptString, injectionTime: .atDocumentEnd, forMainFrameOnly: true)
webView.configuration.userContentController.addUserScript(script)

func userContentController(_ userContentController: WKUserContentController, didReceive message: WKScriptMessage) {
    // Use message.body here
}

Answer 3

只需要评估简单的js脚本

NSString *script = @"window.getSelection().toString()";

使用

evaluateJavaScript

方法

[wkWebView evaluateJavaScript:script completionHandler:^(NSString *selectedString, NSError *error) {
    
}];

Swift 版本

let script = "window.getSelection().toString()"
wkWebView.evaluateJavaScript(script) { selectedString, error in
        
}