Отримати значення параметрів URL

Я намагаюся отримати параметри з URL-адреси за допомогою Swift. Скажімо, у мене є така URL-адреса:

http://mysite3994.com?test1=blah&test2=blahblah

Як я можу отримати значення test1 та test2?

Ви можете використовувати код нижче, щоб отримати параметр

func getQueryStringParameter(url: String, param: String) -> String? {
  guard let url = URLComponents(string: url) else { return nil }
  return url.queryItems?.first(where: { $0.name == param })?.value
}

Викличте метод типу let test1 = getQueryStringParameter(url, param: "test1")

Інший спосіб з розширенням:

extension URL {
    public var queryParameters: [String: String]? {
        guard
            let components = URLComponents(url: self, resolvingAgainstBaseURL: true),
            let queryItems = components.queryItems else { return nil }
        return queryItems.reduce(into: [String: String]()) { (result, item) in
            result[item.name] = item.value
        }
    }
}

Крок 1: Створіть розширення URL-адреси

extension URL {
    func valueOf(_ queryParamaterName: String) -> String? {
        guard let url = URLComponents(string: self.absoluteString) else { return nil }
        return url.queryItems?.first(where: { $0.name == queryParamaterName })?.value
    }
}

Крок 2: Як використовувати розширення

let newURL = URL(string: "http://mysite3994.com?test1=blah&test2=blahblah")!

newURL.valueOf("test1") // Output i.e "blah"
newURL.valueOf("test2") // Output i.e "blahblah"

Я також зробив розширення URL-адреси, але помістив пошук параметрів запиту в індекс.

extension URL {
    subscript(queryParam:String) -> String? {
        guard let url = URLComponents(string: self.absoluteString) else { return nil }
        return url.queryItems?.first(where: { $0.name == queryParam })?.value
    }
}

Використання:

let url = URL(string: "http://some-website.com/documents/127/?referrer=147&mode=open")!

let referrer = url["referrer"]  // "147"
let mode     = url["mode"]      // "open"

Здається, жодна з існуючих відповідей не працює, коли посилання веде на веб-сайт, створений на Angular. Це тому, що шляхи Angular часто містять #символ (хеш) у всіх посиланнях, що призводить до того, що url.queryItemsзавжди повертає нуль.

Якщо посилання виглядає так: http://example.com/path/#/morepath/aaa?test1=blah&test2=blahblah

Тоді параметри можна отримати лише з url.fragment. З додаванням додаткової логіки розбору до розширення @ Matt, більш універсальний код буде виглядати так:

extension URL {
    subscript(queryParam: String) -> String? {
        guard let url = URLComponents(string: self.absoluteString) else { return nil }
        if let parameters = url.queryItems {
            return parameters.first(where: { $0.name == queryParam })?.value
        } else if let paramPairs = url.fragment?.components(separatedBy: "?").last?.components(separatedBy: "&") {
            for pair in paramPairs where pair.contains(queryParam) {
                return pair.components(separatedBy: "=").last
            }
            return nil
        } else {
            return nil
        }
    }
}

Використання залишається незмінним:

let url = URL(string: "http://example.com/path/#/morepath/aaa?test1=blah&test2=blahblah")!

let referrer = url["test1"]  // "blah"
let mode     = url["test2"]  // "blahblah"

Іншим способом зробити це є створення розширення на URL-адресі для повернення компонентів, а потім створення розширення на [URLQueryItem] для отримання значення з queryItems.

extension URL {
    var components: URLComponents? {
        return URLComponents(url: self, resolvingAgainstBaseURL: false)
    }
}

extension Array where Iterator.Element == URLQueryItem {
    subscript(_ key: String) -> String? {
        return first(where: { $0.name == key })?.value
    }
}

І це приклад того, як це можна використовувати:

if let urlComponents = URL(string: "http://mysite3994.com?test1=blah&test2=blahblah")?.components,
    let test1Value = urlComponents.queryItems?["test1"] {
    print(test1Value)
}