import Aacsearch

let client = AacsearchClient(apiKey: "ss_search_your_key_here")

let config = AacsearchConfig(
    baseURL: "https://api.aacsearch.com",
    apiKey: "ss_search_your_key_here",
    timeout: 15
)
let client = AacsearchClient(config: config)

let results = try await client.search(
    "products",
    query: "nike shoes",
    queryBy: "name,description",
    filterBy: "price:>50",
    sortBy: "price:desc",
    page: 1,
    perPage: 20
)

print("Found \(results.found) results")
for hit in results.hits {
    print(hit.document)
}

let multiResults = try await client.multiSearch(searches: [
    MultiSearchQuery(q: "nike", queryBy: "name", perPage: 5),
    MultiSearchQuery(q: "adidas", queryBy: "name", perPage: 5),
])

for result in multiResults.results {
    print("Query returned \(result.found) hits")
}

// Single document
let result = try await client.upsertDocument("products", document: [
    "id": "123",
    "name": "Nike Air Max",
    "price": 129.99,
    "description": "Classic running shoes"
])

// Batch upsert (up to 5000)
let batchResult = try await client.batchUpsertDocuments("products", documents: [
    ["id": "1", "name": "Product A", "price": 19.99],
    ["id": "2", "name": "Product B", "price": 29.99],
])

// Single document
try await client.deleteDocument("products", documentId: "123")

// Batch delete
try await client.batchDeleteDocuments("products", ids: ["1", "2"])

let documents = try await client.listDocuments(
    "products",
    page: 1,
    perPage: 50
)
for doc in documents.hits {
    print(doc.id, doc.document)
}

// List indexes for a project
let indexes = try await client.listIndexes(projectId: "proj_xxx")

// Get index stats
let stats = try await client.getIndexStats("products")
print("Document count: \(stats.documentCount)")

// Create index
let newIndex = try await client.createIndex(
    slug: "products",
    displayName: "Products",
    fields: [
        ["name": "title", "type": "string"],
        ["name": "price", "type": "float", "facet": true],
        ["name": "category", "type": "string", "facet": true],
    ],
    defaultSortingField: "price"
)

// List synonyms
let synonyms = try await client.listSynonyms("products")

// Create synonym
try await client.createSynonym("products", synonym: [
    "root": "shoe",
    "synonyms": ["sneaker", "trainer"]
])

// List curations
let curations = try await client.listCurations("products")

import Aacsearch

do {
    let results = try await client.search("products", query: "laptop")
    print("Found \(results.found) results")
} catch let error as AacsearchError {
    switch error.code {
    case .rateLimited:
        // Back off and retry
        let retryAfter = error.retryAfter ?? 15
        print("Rate limited, retry after \(retryAfter)s")
    case .unauthorized:
        // API key is invalid
        print("Invalid API key: \(error.message)")
    case .notFound:
        // Index not found
        print("Index not found: \(error.message)")
    case .quotaExceeded:
        // Monthly quota exhausted
        print("Quota exceeded, upgrade your plan")
    default:
        print("Error \(error.statusCode): \(error.message)")
    }
} catch {
    print("Unexpected error: \(error)")
}