import { getTemplateUrl, ArtosAPIError } from '@artosai/sdk'

async function safeGetTemplate(documentId) {
    try {
        const result = await getTemplateUrl({
            documentId,
            accessToken: process.env.ARTOS_ACCESS_TOKEN,
            baseUrl: 'https://api.artosai.com'
        })
        return { success: true, data: result }
    } catch (error) {
        if (error instanceof ArtosAPIError) {
            console.error(`API Error (${error.status}): ${error.message}`)

            switch (error.status) {
                case 401:
                    return { success: false, error: 'Invalid or expired access token' }
                case 403:
                    return { success: false, error: 'Access denied to this document' }
                case 404:
                    return { success: false, error: 'Document not found' }
                default:
                    return { success: false, error: `API error: ${error.message}` }
            }
        }

        console.error('Unexpected error:', error)
        return { success: false, error: 'An unexpected error occurred' }
    }
}

async function withRetry(fn, maxRetries = 3) {
    for (let i = 0; i < maxRetries; i++) {
        try {
            return await fn()
        } catch (error) {
            if (i === maxRetries - 1) throw error
            await new Promise(r => setTimeout(r, 1000 * (i + 1)))
        }
    }
}

// Usage
const result = await withRetry(() => getTemplateUrl({
    documentId: 'doc-123',
    accessToken: token,
    baseUrl: 'https://api.artosai.com'
}))