import Foundation

public struct SentryInterpolatedString: ExpressibleByStringInterpolation {
    public let message: String
    public let attributes: [String: SentryLog.Attribute]
    
    public init(stringLiteral value: String) {
        self.message = value
        self.attributes = [:]
    }
    
    public init(stringInterpolation: StringInterpolation) {
        self.message = stringInterpolation.message
        self.attributes = stringInterpolation.attributes
    }
    
    public struct StringInterpolation: StringInterpolationProtocol {
        var message: String = ""
        var attributes: [String: SentryLog.Attribute] = [:]
        private var interpolationCounter = 0
        
        public init(literalCapacity: Int, interpolationCount: Int) {
            message.reserveCapacity(literalCapacity * 2)
            attributes.reserveCapacity(interpolationCount)
        }
        
        public mutating func appendLiteral(_ literal: String) {
            message.append(literal)
        }
        
        // MARK: - Supported SentryLog.Attribute types only
        
        /// Append String interpolation
        public mutating func appendInterpolation(_ value: String) {
            let key = "interpolation_\(interpolationCounter)"
            interpolationCounter += 1
            
            message.append(value)
            attributes[key] = .string(value)
        }
        
        // Other attibute types
        ...
        
        /// Append interpolation without tracking (for any other type or sensitive data)
        public mutating func appendInterpolation<T>(untracked value: T) {
            message.append(String(describing: value))
        }
    }
}

// MARK: - SentryLogger Extensions

extension SentryLogger {
    /// Logs a trace-level message using interpolated string that automatically captures attributes
    public func trace(_ interpolatedString: SentryInterpolatedString) {
        captureLog(level: .trace, body: interpolatedString.message, attributes: interpolatedString.attributes.mapValues { $0.value })
    }

    ...
}

// Automatic attribute tracking for supported types
let userName = "Henry"
logger.info("User \(userName) just logged in...")