Revision control
Copy as Markdown
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
import UIKit
import DesignSystem
protocol FindInPageBarDelegate: AnyObject {
func findInPage(_ findInPage: FindInPageBar, didTextChange text: String)
func findInPage(_ findInPage: FindInPageBar, didFindPreviousWithText text: String)
func findInPage(_ findInPage: FindInPageBar, didFindNextWithText text: String)
func findInPageDidPressClose(_ findInPage: FindInPageBar)
}
class FindInPageBar: UIView {
weak var delegate: FindInPageBarDelegate?
private let searchText = UITextField()
private let matchCountView = UILabel()
private let previousButton = UIButton()
private let nextButton = UIButton()
var currentResult = 0 {
didSet {
if totalResults > 500 {
matchCountView.text = "\(currentResult)/500+"
} else {
matchCountView.text = "\(currentResult)/\(totalResults)"
}
}
}
var totalResults = 0 {
didSet {
if totalResults > 500 {
matchCountView.text = "\(currentResult)/500+"
} else {
matchCountView.text = "\(currentResult)/\(totalResults)"
}
previousButton.isEnabled = totalResults > 1
nextButton.isEnabled = previousButton.isEnabled
}
}
var text: String? {
get {
return searchText.text
}
set {
searchText.text = newValue
didTextChange(searchText)
}
}
override init(frame: CGRect) {
super.init(frame: frame)
backgroundColor = .foundation
searchText.addTarget(self, action: #selector(didTextChange), for: .editingChanged)
searchText.textColor = .primaryText
searchText.tintColor = .primaryText
searchText.font = .body16
searchText.autocapitalizationType = .none
searchText.autocorrectionType = .no
searchText.inputAssistantItem.leadingBarButtonGroups = []
searchText.inputAssistantItem.trailingBarButtonGroups = []
searchText.enablesReturnKeyAutomatically = true
searchText.returnKeyType = .search
searchText.accessibilityIdentifier = "FindInPage.searchField"
searchText.delegate = self
addSubview(searchText)
matchCountView.textColor = .primaryText.withAlphaComponent(0.6)
matchCountView.font = .body16
matchCountView.isHidden = true
matchCountView.accessibilityIdentifier = "FindInPage.matchCount"
addSubview(matchCountView)
previousButton.setImage(.findPrevious, for: [])
previousButton.setTitleColor(.white, for: [])
previousButton.accessibilityLabel = UIConstants.strings.findInPagePreviousLabel
previousButton.addTarget(self, action: #selector(didFindPrevious), for: .touchUpInside)
previousButton.accessibilityIdentifier = "FindInPage.find_previous"
previousButton.isEnabled = false
addSubview(previousButton)
nextButton.setImage(.findNext, for: [])
nextButton.setTitleColor(.white, for: [])
nextButton.accessibilityLabel = UIConstants.strings.findInPageNextLabel
nextButton.addTarget(self, action: #selector(didFindNext), for: .touchUpInside)
nextButton.accessibilityIdentifier = "FindInPage.find_next"
nextButton.isEnabled = false
addSubview(nextButton)
let closeButton = UIButton()
closeButton.setImage(.stopMenu, for: [])
closeButton.setTitleColor(.white, for: [])
closeButton.accessibilityLabel = UIConstants.strings.findInPageDoneLabel
closeButton.addTarget(self, action: #selector(didPressClose), for: .touchUpInside)
closeButton.accessibilityIdentifier = "FindInPage.close"
addSubview(closeButton)
let topBorder = UIView()
topBorder.backgroundColor = .foundation
addSubview(topBorder)
searchText.snp.makeConstraints { make in
make.leading.top.bottom.equalTo(self).inset(UIConstants.layout.findInPageSearchTextInsets)
}
searchText.setContentHuggingPriority(.defaultLow, for: .horizontal)
searchText.setContentCompressionResistancePriority(.defaultLow, for: .horizontal)
matchCountView.snp.makeConstraints { make in
make.leading.equalTo(searchText.snp.trailing)
make.centerY.equalTo(self)
}
matchCountView.setContentHuggingPriority(.defaultHigh, for: .horizontal)
matchCountView.setContentCompressionResistancePriority(.defaultHigh, for: .horizontal)
previousButton.snp.makeConstraints { make in
make.leading.equalTo(matchCountView.snp.trailing).offset(UIConstants.layout.findInPagePreviousButtonOffset)
make.size.equalTo(self.snp.height)
make.centerY.equalTo(self)
}
nextButton.snp.makeConstraints { make in
make.leading.equalTo(previousButton.snp.trailing)
make.size.equalTo(self.snp.height)
make.centerY.equalTo(self)
}
closeButton.snp.makeConstraints { make in
make.leading.equalTo(nextButton.snp.trailing)
make.size.equalTo(self.snp.height)
make.trailing.centerY.equalTo(self)
}
topBorder.snp.makeConstraints { make in
make.height.equalTo(1)
make.left.right.top.equalTo(self)
}
}
required init?(coder aDecoder: NSCoder) {
fatalError("init(coder:) has not been implemented")
}
override func pressesBegan(_ presses: Set<UIPress>, with event: UIPressesEvent?) {
guard let key = presses.first?.key else { return }
switch key.keyCode {
case .keyboardEscape:
delegate?.findInPageDidPressClose(self)
default:
super.pressesBegan(presses, with: event)
}
}
@discardableResult
override func becomeFirstResponder() -> Bool {
searchText.becomeFirstResponder()
return super.becomeFirstResponder()
}
@objc
private func didFindPrevious(_ sender: UIButton) {
delegate?.findInPage(self, didFindPreviousWithText: searchText.text ?? "")
}
@objc
private func didFindNext(_ sender: UIButton) {
delegate?.findInPage(self, didFindNextWithText: searchText.text ?? "")
}
@objc
private func didTextChange(_ sender: UITextField) {
matchCountView.isHidden = searchText.text?.trimmingCharacters(in: .whitespaces).isEmpty ?? true
delegate?.findInPage(self, didTextChange: searchText.text ?? "")
}
@objc
private func didPressClose(_ sender: UIButton) {
delegate?.findInPageDidPressClose(self)
}
}
extension FindInPageBar: UITextFieldDelegate {
func textFieldShouldReturn(_ textField: UITextField) -> Bool {
textField.endEditing(true)
return true
}
}