361 lines
14 KiB
Swift
361 lines
14 KiB
Swift
//
|
||
// LocationPickerVC.swift
|
||
// QuickLocation
|
||
//
|
||
// Created by 八条 on 2026/6/24.
|
||
//
|
||
|
||
import UIKit
|
||
import RxSwift
|
||
import RxCocoa
|
||
import CoreLocation
|
||
import SwiftyUserDefaults
|
||
#if !targetEnvironment(simulator)
|
||
import AMapNaviKit
|
||
import AMapSearchKit
|
||
#endif
|
||
|
||
struct PickedLocation {
|
||
let name: String
|
||
let address: String
|
||
let coordinate: CLLocationCoordinate2D
|
||
var province: String = ""
|
||
var city: String = ""
|
||
var district: String = ""
|
||
var street: String = ""
|
||
var country: String = ""
|
||
var formatted_address: String = ""
|
||
}
|
||
|
||
class LocationPickerVC: BaseViewController {
|
||
|
||
override var isNavigationBarHidden: Bool { true }
|
||
|
||
var onPickedLocation: ((PickedLocation) -> Void)?
|
||
var initialLocation: PickedLocation?
|
||
|
||
fileprivate var rootView: LocationPickerView!
|
||
|
||
override func loadView() {
|
||
rootView = LocationPickerView(frame: UIScreen.main.bounds)
|
||
view = rootView
|
||
}
|
||
|
||
private var isShowPoi: Bool = false
|
||
override func viewDidDisappear(_ animated: Bool) {
|
||
super.viewDidDisappear(animated)
|
||
// 页面被 pop/dismiss 时清理地图资源
|
||
if isMovingFromParent || isBeingDismissed {
|
||
#if !targetEnvironment(simulator)
|
||
rootView.cleanupMap()
|
||
#endif
|
||
}
|
||
}
|
||
|
||
override func viewDidLoad() {
|
||
super.viewDidLoad()
|
||
setupMap()
|
||
setupKeyboard()
|
||
setupSearch()
|
||
// 如果有初始位置，显示在底部面板
|
||
if let loc = initialLocation {
|
||
selectedLocation = loc
|
||
rootView.bottomView.isHidden = false
|
||
rootView.poiNameLab.text = loc.name
|
||
rootView.poiAddressLab.text = loc.address
|
||
if CLLocationCoordinate2DIsValid(loc.coordinate) {
|
||
rootView.mapView.setCenter(loc.coordinate, animated: false)
|
||
rootView.mapView.setZoomLevel(19, animated: false)
|
||
addLocationAnnotation(coordinate: loc.coordinate)
|
||
}
|
||
}
|
||
}
|
||
|
||
// MARK: - Keyboard
|
||
private var keyboardHeight: CGFloat = 0
|
||
|
||
private func setupKeyboard() {
|
||
NotificationCenter.default.rx.notification(UIResponder.keyboardWillShowNotification)
|
||
.subscribe(onNext: { [weak self] noti in
|
||
guard let self = self,
|
||
let frame = noti.userInfo?[UIResponder.keyboardFrameEndUserInfoKey] as? CGRect
|
||
else { return }
|
||
self.keyboardHeight = frame.height
|
||
})
|
||
.disposed(by: disposeBag)
|
||
}
|
||
|
||
// MARK: - Search
|
||
#if !targetEnvironment(simulator)
|
||
private let searchAPI = AMapSearchAPI()
|
||
private var searchPoiRequest = AMapPOIKeywordsSearchRequest()
|
||
#endif
|
||
private var selectedLocation: PickedLocation?
|
||
private var searchKeyword: String = ""
|
||
private var poiItems: [AMapPOI] = []
|
||
private var maxTableHeight: CGFloat = 300
|
||
private var currentLocation: CLLocationCoordinate2D?
|
||
|
||
private func setupSearch() {
|
||
#if !targetEnvironment(simulator)
|
||
searchAPI?.delegate = self
|
||
|
||
searchPoiRequest.city = ""
|
||
searchPoiRequest.offset = 10
|
||
if let loc = currentLocation {
|
||
searchPoiRequest.location = AMapGeoPoint.location(withLatitude: CGFloat(loc.latitude), longitude: CGFloat(loc.longitude))
|
||
}
|
||
#endif
|
||
|
||
rootView.resultTableView.delegate = self
|
||
rootView.resultTableView.dataSource = self
|
||
|
||
// 输入联想
|
||
rootView.searchField.rx.text.orEmpty
|
||
.debounce(.milliseconds(400), scheduler: MainScheduler.instance)
|
||
// .filter { $0.count >= 2 }
|
||
.subscribe(onNext: { [weak self] keyword in
|
||
self?.poiSearch(keyword: keyword)
|
||
})
|
||
.disposed(by: disposeBag)
|
||
rootView.searchField.rx.controlEvent(.editingDidBegin)
|
||
.subscribe(onNext: {
|
||
self.isShowPoi = false
|
||
})
|
||
.disposed(by: disposeBag)
|
||
|
||
rootView.searchField.rx.controlEvent(.editingDidEndOnExit)
|
||
.subscribe(onNext: { [weak self] _ in
|
||
guard let self = self, !self.poiItems.isEmpty else { return }
|
||
self.selectPOI(at: 0)
|
||
self.rootView.searchField.resignFirstResponder()
|
||
self.rootView.resultTableView.isHidden = true
|
||
})
|
||
.disposed(by: disposeBag)
|
||
|
||
// 搜索按钮 → 选中第一条联想
|
||
rootView.searchBtn.rx.tap
|
||
.subscribe(onNext: { [weak self] _ in
|
||
guard let self = self, !self.poiItems.isEmpty else { return }
|
||
self.selectPOI(at: 0)
|
||
self.rootView.searchField.resignFirstResponder()
|
||
self.rootView.resultTableView.isHidden = true
|
||
})
|
||
.disposed(by: disposeBag)
|
||
|
||
rootView.backBtn.rx.tap
|
||
.subscribe(onNext: { [weak self] _ in
|
||
self?.dismiss(animated: true)
|
||
})
|
||
.disposed(by: disposeBag)
|
||
|
||
rootView.confirmBtn.rx.tap
|
||
.subscribe(onNext: { [weak self] _ in
|
||
guard let self = self, let loc = self.selectedLocation else { return }
|
||
self.onPickedLocation?(loc)
|
||
self.dismiss(animated: true)
|
||
})
|
||
.disposed(by: disposeBag)
|
||
}
|
||
|
||
private func poiSearch(keyword: String) {
|
||
searchKeyword = keyword
|
||
#if !targetEnvironment(simulator)
|
||
searchPoiRequest.keywords = keyword
|
||
searchAPI?.aMapPOIKeywordsSearch(searchPoiRequest)
|
||
#endif
|
||
}
|
||
|
||
private func selectPOI(at index: Int) {
|
||
guard index < poiItems.count else { return }
|
||
selectLocation(poi: poiItems[index])
|
||
}
|
||
|
||
private func selectLocation(poi: AMapPOI) {
|
||
let coord = CLLocationCoordinate2D(latitude: CGFloat(poi.location.latitude), longitude: CGFloat(poi.location.longitude))
|
||
guard CLLocationCoordinate2DIsValid(coord) else { return }
|
||
isShowPoi = true
|
||
selectedLocation = PickedLocation(
|
||
name: poi.name ?? "",
|
||
address: poi.address ?? "",
|
||
coordinate: coord,
|
||
province: poi.province ?? "",
|
||
city: poi.city ?? "",
|
||
district: poi.district ?? "",
|
||
street: "",
|
||
formatted_address: poi.address ?? ""
|
||
)
|
||
rootView.resultTableView.isHidden = true
|
||
rootView.bottomView.isHidden = false
|
||
rootView.poiNameLab.text = poi.name
|
||
rootView.poiAddressLab.text = poi.address
|
||
rootView.mapView.setCenter(coord, animated: true)
|
||
rootView.mapView.setZoomLevel(19, animated: true)
|
||
addLocationAnnotation(coordinate: coord)
|
||
// 逆地理编码补充 street
|
||
#if !targetEnvironment(simulator)
|
||
let regeo = AMapReGeocodeSearchRequest()
|
||
regeo.location = AMapGeoPoint.location(withLatitude: CGFloat(coord.latitude),
|
||
longitude: CGFloat(coord.longitude))
|
||
regeo.requireExtension = true
|
||
searchAPI?.aMapReGoecodeSearch(regeo)
|
||
#endif
|
||
}
|
||
|
||
private func updateTableHeight(itemCount: Int) {
|
||
let contentH = CGFloat(itemCount) * 50
|
||
let searchBarBottom = kStatusBarHeight + 8 + 44 + 10
|
||
let bottomOffset: CGFloat = keyboardHeight > 0 ? keyboardHeight : 200
|
||
let available = kScreenHeight - searchBarBottom - bottomOffset - 10
|
||
maxTableHeight = max(80, available)
|
||
let h = min(contentH, maxTableHeight)
|
||
rootView.resultTableView.layoutChain.height(h)
|
||
}
|
||
|
||
// MARK: - Map
|
||
private func setupMap() {
|
||
#if !targetEnvironment(simulator)
|
||
rootView.mapView.delegate = self
|
||
rootView.mapView.showsUserLocation = false
|
||
if let lat = Defaults[\.currentLatitude], let lon = Defaults[\.currentLongitude] {
|
||
let coord = CLLocationCoordinate2D(latitude: lat, longitude: lon)
|
||
if CLLocationCoordinate2DIsValid(coord) {
|
||
currentLocation = coord
|
||
rootView.mapView.setCenter(coord, animated: false)
|
||
rootView.mapView.setZoomLevel(18, animated: false)
|
||
}
|
||
}
|
||
#endif
|
||
}
|
||
|
||
private func addLocationAnnotation(coordinate: CLLocationCoordinate2D) {
|
||
#if !targetEnvironment(simulator)
|
||
for ann in rootView.mapView.annotations?.compactMap({ $0 as? MAPointAnnotation }) ?? [] {
|
||
rootView.mapView.removeAnnotation(ann)
|
||
}
|
||
let ann = MAPointAnnotation()
|
||
ann.coordinate = coordinate
|
||
rootView.mapView.addAnnotation(ann)
|
||
#endif
|
||
}
|
||
}
|
||
|
||
#if !targetEnvironment(simulator)
|
||
// MARK: - AMapSearchDelegate
|
||
extension LocationPickerVC: AMapSearchDelegate {
|
||
func onPOISearchDone(_ request: AMapPOISearchBaseRequest!, response: AMapPOISearchResponse!) {
|
||
// POI ID 搜索（来自地图点击）直接选中
|
||
if request is AMapPOIIDSearchRequest, let poi = response.pois?.first {
|
||
selectLocation(poi: poi)
|
||
return
|
||
}
|
||
|
||
// 关键字搜索
|
||
guard response.pois.count > 0, isShowPoi == false else { return }
|
||
poiItems = response.pois
|
||
rootView.resultTableView.isHidden = false
|
||
rootView.resultTableView.reloadData()
|
||
updateTableHeight(itemCount: poiItems.count)
|
||
}
|
||
|
||
func onReGeocodeSearchDone(_ request: AMapReGeocodeSearchRequest!, response: AMapReGeocodeSearchResponse!) {
|
||
guard let regeo = response.regeocode else { return }
|
||
let address = regeo.formattedAddress ?? ""
|
||
// 取第一个 POI 名称作为地点名，没有则取街道/区
|
||
let poiName = regeo.pois?.first?.name
|
||
?? regeo.pois?.first?.address
|
||
?? regeo.pois?.first?.district
|
||
?? ""
|
||
selectedLocation = PickedLocation(
|
||
name: poiName,
|
||
address: regeo.pois?.first?.address ?? "",
|
||
coordinate: selectedLocation?.coordinate ?? kCLLocationCoordinate2DInvalid,
|
||
province: regeo.pois?.first?.province ?? "",
|
||
city: regeo.pois?.first?.city ?? "",
|
||
district: regeo.pois?.first?.district ?? "",
|
||
street: regeo.addressComponent?.streetNumber?.street ?? "",
|
||
formatted_address: regeo.pois?.first?.address ?? ""
|
||
)
|
||
rootView.bottomView.isHidden = false
|
||
rootView.poiNameLab.text = poiName
|
||
rootView.poiAddressLab.text = address
|
||
}
|
||
|
||
func aMapSearchRequest(_ request: Any!, didFailWithError error: Error!) {
|
||
print("AMapSearch error: \(error.localizedDescription)")
|
||
}
|
||
}
|
||
|
||
// MARK: - MAMapViewDelegate
|
||
extension LocationPickerVC: MAMapViewDelegate {
|
||
func mapView(_ mapView: MAMapView!, didSingleTappedAt coordinate: CLLocationCoordinate2D) {
|
||
isShowPoi = true
|
||
selectedLocation = PickedLocation(
|
||
name: "",
|
||
address: "",
|
||
coordinate: coordinate
|
||
)
|
||
addLocationAnnotation(coordinate: coordinate)
|
||
rootView.mapView.setCenter(coordinate, animated: true)
|
||
// 逆地理编码获取地址
|
||
#if !targetEnvironment(simulator)
|
||
let regeo = AMapReGeocodeSearchRequest()
|
||
regeo.location = AMapGeoPoint.location(withLatitude: CGFloat(coordinate.latitude),
|
||
longitude: CGFloat(coordinate.longitude))
|
||
regeo.requireExtension = true
|
||
searchAPI?.aMapReGoecodeSearch(regeo)
|
||
#endif
|
||
}
|
||
|
||
func mapView(_ mapView: MAMapView!, viewFor annotation: MAAnnotation!) -> MAAnnotationView! {
|
||
if annotation is MAUserLocation { return nil }
|
||
guard annotation is MAPointAnnotation else { return nil }
|
||
let identifier = "LocationPin"
|
||
var view = mapView.dequeueReusableAnnotationView(withIdentifier: identifier)
|
||
if view == nil {
|
||
view = MAAnnotationView(annotation: annotation, reuseIdentifier: identifier)
|
||
} else {
|
||
view?.annotation = annotation
|
||
}
|
||
view?.image = UIImage(named: "Schedule/location")
|
||
view?.centerOffset = CGPoint(x: 0, y: -30)
|
||
return view
|
||
}
|
||
}
|
||
|
||
// MARK: - UITableViewDataSource / Delegate
|
||
extension LocationPickerVC: UITableViewDataSource, UITableViewDelegate {
|
||
func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int {
|
||
poiItems.count
|
||
}
|
||
|
||
func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
|
||
let cell = tableView.dequeueReusableCell(withIdentifier: "cell") ?? UITableViewCell(style: .subtitle, reuseIdentifier: "cell")
|
||
let poi = poiItems[indexPath.row]
|
||
let name = poi.name ?? ""
|
||
if !searchKeyword.isEmpty, let range = name.lowercased().range(of: searchKeyword.lowercased()) {
|
||
let attr = NSMutableAttributedString(string: name)
|
||
attr.addAttribute(.foregroundColor, value: UIColor(hexStr: "#16B3FF"), range: NSRange(range, in: name))
|
||
cell.textLabel?.attributedText = attr
|
||
} else {
|
||
cell.textLabel?.text = name
|
||
}
|
||
cell.textLabel?.font = .systemFont(ofSize: 14)
|
||
cell.detailTextLabel?.text = poi.address
|
||
cell.detailTextLabel?.font = .systemFont(ofSize: 12)
|
||
cell.detailTextLabel?.textColor = .gray
|
||
let arrow = UIImageView(image: UIImage(named: "Schedule/arrow"))
|
||
arrow.frame = CGRect(x: 0, y: 0, width: 10, height: 10)
|
||
cell.accessoryView = arrow
|
||
return cell
|
||
}
|
||
|
||
func tableView(_ tableView: UITableView, didSelectRowAt indexPath: IndexPath) {
|
||
tableView.deselectRow(at: indexPath, animated: true)
|
||
rootView.searchField.resignFirstResponder()
|
||
selectPOI(at: indexPath.row)
|
||
rootView.searchField.text = poiItems[indexPath.row].name
|
||
}
|
||
}
|
||
#endif
|