osx/Helper/KBLogger.m - master - githubmirror/client

osx/Helper/KBLogger.m
81 строка2 KB
Gabriel Handford
Log helper tool output to file, so we can log send (#3581)
26 июл 2016, 02:53
26 июл 2016, 02:53d71e895
//
//  KBLogger.m
//  Keybase
//
//  Created by Gabriel on 7/22/16.
//  Copyright © 2016 Keybase. All rights reserved.
//

#import "KBLogger.h"

#import <syslog.h>

@interface KBFileLogger : NSObject
@property NSFileHandle *currentLogFileHandle;
@property NSDateFormatter *dateFormatter;
- (void)log:(NSString *)message;
@end

void KBLog(NSString *msg, ...) {
  if (!msg) return;
  va_list args;
  va_start(args, msg);

  NSString *string = [[NSString alloc] initWithFormat:msg arguments:args];

  va_end(args);

  // Log to as many places as possible: syslog, stdout and file
  syslog(LOG_NOTICE, "%s", [string UTF8String]);

  NSLog(@"%@", string);

  static dispatch_once_t onceToken;
  static KBFileLogger *logger;
  dispatch_once(&onceToken, ^{
    logger = [[KBFileLogger alloc] init];
  });
  [logger log:string];
}

@implementation KBFileLogger

- (id)init {
  if ((self = [super init])) {
    // Ensure log file exists
    NSString * const logPath = @"/Library/Logs/keybase.system.log";
    if (![[NSFileManager defaultManager] fileExistsAtPath:logPath]) {
      if (![[NSFileManager defaultManager] createFileAtPath:logPath contents:nil attributes:nil]) {
        NSLog(@"Unable to create log file: %@", logPath);
      }
    }
    _currentLogFileHandle = [NSFileHandle fileHandleForWritingAtPath:logPath];
    [_currentLogFileHandle seekToEndOfFile];

    _dateFormatter = [[NSDateFormatter alloc] init];
    [_dateFormatter setDateFormat:@"yyyy/MM/dd HH:mm:ss.SSS"];
  }
  return self;
}

- (void)dealloc {
  [_currentLogFileHandle synchronizeFile];
  [_currentLogFileHandle closeFile];
}

/*!
 Log to file.
 */
- (void)log:(NSString *)message {
  NSString *prefix = [_dateFormatter stringFromDate:[NSDate date]];
  NSString *logMessage = [NSString stringWithFormat:@"%@ %@\n", prefix, message];
  NSData *logData = [logMessage dataUsingEncoding:NSUTF8StringEncoding];

  @try {
    [_currentLogFileHandle writeData:logData];
  } @catch (NSException *exception) {
    NSLog(@"Error writing to log: %@", exception);
  }
}

@end