summaryrefslogtreecommitdiff
path: root/DomainDigShareExtension/ShareViewController.swift
blob: 411f9e7ddb4307a52759efc942190700acc1a52a (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
import UIKit
import UniformTypeIdentifiers

/// Receives a web URL from the share sheet, extracts its host, and hands it to
/// the app through the App Group inbox. Shows a brief confirmation and closes;
/// the app picks up the domain the next time it becomes active.
final class ShareViewController: UIViewController {
    private let label = UILabel()

    override func viewDidLoad() {
        super.viewDidLoad()
        view.backgroundColor = .systemBackground

        label.text = "Saving…"
        label.font = .preferredFont(forTextStyle: .headline)
        label.textAlignment = .center
        label.numberOfLines = 0
        label.translatesAutoresizingMaskIntoConstraints = false
        view.addSubview(label)
        NSLayoutConstraint.activate([
            label.centerXAnchor.constraint(equalTo: view.centerXAnchor),
            label.centerYAnchor.constraint(equalTo: view.centerYAnchor),
            label.leadingAnchor.constraint(greaterThanOrEqualTo: view.leadingAnchor, constant: 24),
            label.trailingAnchor.constraint(lessThanOrEqualTo: view.trailingAnchor, constant: -24)
        ])

        extractSharedDomain { [weak self] domain in
            DispatchQueue.main.async {
                self?.finish(domain: domain)
            }
        }
    }

    private func extractSharedDomain(completion: @escaping (String?) -> Void) {
        let providers = (extensionContext?.inputItems ?? [])
            .compactMap { $0 as? NSExtensionItem }
            .flatMap { $0.attachments ?? [] }

        guard let provider = providers.first(where: {
            $0.hasItemConformingToTypeIdentifier(UTType.url.identifier)
        }) else {
            completion(nil)
            return
        }

        provider.loadItem(forTypeIdentifier: UTType.url.identifier) { item, _ in
            let url = item as? URL ?? (item as? Data).flatMap { URL(dataRepresentation: $0, relativeTo: nil) }
            let host = url?.host?.trimmingCharacters(in: .whitespacesAndNewlines)
            completion(host?.isEmpty == false ? host : nil)
        }
    }

    private func finish(domain: String?) {
        if let domain {
            DomainDigShareInbox.write(domain: domain)
            label.text = "Saved \(domain) — open DomainDig to inspect it."
        } else {
            label.text = "No web address found."
        }

        DispatchQueue.main.asyncAfter(deadline: .now() + 1.4) { [weak self] in
            self?.extensionContext?.completeRequest(returningItems: nil)
        }
    }
}