LogFormatter.swift

//
//  LogFormatter.swift
//
//  Created by Ryoga Kitagawa on 3/25/16.
//  Copyright © 2016 Givery. All rights reserved.
//

import Foundation
import CocoaLumberjackSwift
import CocoaLumberjack.DDDispatchQueueLogFormatter

class LogFormatter: DDDispatchQueueLogFormatter {

    let dateFormatter: NSDateFormatter

    static func applyFormatter() {
        #if CONFIGURATION_AppStore
            defaultDebugLevel = DDLogLevel.Warning
        #else
            defaultDebugLevel = DDLogLevel.Verbose
        #endif
        DDTTYLogger.sharedInstance().logFormatter = LogFormatter()
        DDLog.addLogger(DDTTYLogger.sharedInstance())
    }

    override init!() {
        self.dateFormatter = NSDateFormatter()
        self.dateFormatter.dateFormat = "yyyy/MM/dd HH:mm:ss:SSS"
        super.init()
    }

    override func formatLogMessage(logMessage: DDLogMessage!) -> String! {

        var logLevel: String
        switch logMessage.flag {
        case DDLogFlag.Error:
            logLevel = "E"
        case DDLogFlag.Warning:
            logLevel = "W"
        case DDLogFlag.Info:
            logLevel = "I"
        case DDLogFlag.Debug:
            logLevel = "D"
        case DDLogFlag.Verbose:
            logLevel = "V"
        default:
            logLevel = "U"  // Unknown
        }

        let dateAndTime = self.dateFormatter.stringFromDate(logMessage.timestamp)
        let queueLabel = self.queueThreadLabelForLogMessage(logMessage)
        let fileName = (logMessage.file as NSString).lastPathComponent
        let lineNumber = logMessage.line
        let function = logMessage.function
        let message = logMessage.message

        return "\(dateAndTime) [\(logLevel)] [\(queueLabel)] [\(fileName):\(lineNumber) \(function)] \n\(message) "
    }
}