//
//  LHSqlite.m
//  LHDBDemo
//
//  Created by 3wchina01 on 16/3/22.
//  Copyright © 2016年 李浩. All rights reserved.
//

#import "LHSqlite.h"
#import <sqlite3.h>
#import <libkern/OSAtomic.h>
#import <UIKit/UIKit.h>
#import "NSObject+LHModel.h"

#define Lock OSSpinLockLock(&(self->_lock))
#define UnLock OSSpinLockUnlock(&(self->_lock))
#define TryLock OSSpinLockTry(&(self->_lock))



@implementation LHSqlite{
    @package
    OSSpinLock _lock;
    sqlite3* _db;
}

static NSError* errorForDataBase(NSString* sqlString,sqlite3* db)
{
    NSError* error = [NSError errorWithDomain:[NSString stringWithUTF8String:sqlite3_errmsg(db)] code:sqlite3_errcode(db) userInfo:@{@"sqlString":sqlString}];
    return error;
}

static NSObject* dataWithDataType(int type,sqlite3_stmt * statement,int index)
{
    if (type == SQLITE_INTEGER) {
        int value = sqlite3_column_int(statement, index);
        return [NSNumber numberWithInt:value];
    }else if (type == SQLITE_FLOAT) {
        float value = sqlite3_column_double(statement, index);
        return [NSNumber numberWithFloat:value];
    }else if (type == SQLITE_BLOB) {
        const void *value = sqlite3_column_blob(statement, index);
        int bytes = sqlite3_column_bytes(statement, index);
        return [NSData dataWithBytes:value length:bytes];
    }else if (type == SQLITE_NULL) {
        return nil;
    }else if (type == SQLITE_TEXT) {
        return [NSString stringWithUTF8String:(char*)sqlite3_column_text(statement, index)];
    }else {
        return nil;
    }
}

- (instancetype)initWithPath:(NSString*)dbPath
{
    self = [super init];
    if (self) {
        _sqlPath = dbPath;
        _lock = OS_SPINLOCK_INIT;
    }
    return self;
}

+ (instancetype)sqliteWithPath:(NSString*)dbPath
{
    LHSqlite* sqlite = [[LHSqlite alloc] initWithPath:dbPath];
    return sqlite;
}

+ (instancetype)shareInstance
{
    static LHSqlite* sqlite = nil;
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        sqlite = [LHSqlite sqliteWithPath:nil];
    });
    return sqlite;
}

- (BOOL)openDB
{
    if (sqlite3_open([self.sqlPath UTF8String], &_db) == SQLITE_OK) {
        return YES;
    }else{
        sqlite3_close(_db);
        return NO;
    }
}

- (void)executeSQLWithSqlstring:(NSString*)sqlString object:(id)model executeType:(LHSqliteType)type success:(success)successBlock fail:(fail)failBlock
{
    Lock;
    if (![self openDB]) {
        if (failBlock) {
            failBlock([NSError errorWithDomain:@"打开数据库失败" code:0 userInfo:@{}]);
        }
        UnLock;
        return;
    }
    sqlite3_stmt* stmt;
    if (type == LHSqliteTypeWrite) {
        [self writeToDB:sqlString stmt:stmt model:model complete:^(NSError *error) {
            UnLock;
            if (failBlock) {
                failBlock(error);
            }
        
        }];
    }else {
        [self readFromDB:sqlString stmt:stmt success:^(NSArray *result) {
            UnLock;
            if (successBlock) {
                successBlock(result);
            }
        } fail:^(NSError *error) {
            UnLock;
            if (failBlock) {
                failBlock(error);
            }
        }];
    }
    sqlite3_close(_db);
}

- (void)writeToDB:(NSString*)sqlString stmt:(sqlite3_stmt*)stmt model:(id)model complete:(fail)complete
{
    NSDictionary* modelDic = [model lh_ModelToDictionary];
    if (sqlite3_prepare_v2(_db, [sqlString UTF8String], -1, &stmt, nil) != SQLITE_OK) {
        if (complete) {
            complete(errorForDataBase(sqlString, _db));
        }else
            UnLock;
        sqlite3_finalize(stmt);
        return;
    }
    for (int i=0; i<modelDic.allKeys.count; i++) {
        [self bindObject:modelDic[modelDic.allKeys[i]] toColumn:i+1 inStatement:stmt];
    }
    if (sqlite3_step(stmt) != SQLITE_DONE) {
        if (complete) {
            complete(errorForDataBase(sqlString, _db));
        }else
            UnLock;
    }else
        UnLock;
    sqlite3_finalize(stmt);
}

- (void)readFromDB:(NSString*)sqlString stmt:(sqlite3_stmt*)stmt success:(success)successBlock fail:(fail)failBlock
{
    NSMutableArray* dataSource = [NSMutableArray array];
    if (sqlite3_prepare_v2(_db, [sqlString UTF8String], -1, &stmt, nil) == SQLITE_OK) {
        int count = sqlite3_column_count(stmt);
        while (sqlite3_step(stmt) == SQLITE_ROW) {
            NSMutableDictionary* dataDic = [NSMutableDictionary dictionary];
            for (int i=0; i<count; i++) {
                int type = sqlite3_column_type(stmt, i);
                NSString* propertyName = [NSString stringWithUTF8String:sqlite3_column_name(stmt, i)];
                NSObject* value = dataWithDataType(type, stmt, i);
                [dataDic setValue:value forKey:propertyName];
            }
            [dataSource addObject:dataDic];
        }
        if (successBlock) {
            successBlock(dataSource);
        }else
            UnLock;
        sqlite3_finalize(stmt);

    }else {
        if (failBlock) {
            failBlock(errorForDataBase(sqlString, _db));
        }else
            UnLock;
        sqlite3_finalize(stmt);

    }
}

- (void)bindObject:(id)obj toColumn:(int)idx inStatement:(sqlite3_stmt*)pStmt {
    
    if ((!obj) || ((NSNull *)obj == [NSNull null])) {
        sqlite3_bind_null(pStmt, idx);
    }
    
    else if ([obj isKindOfClass:[NSData class]]) {
        const void *bytes = [obj bytes];
        if (!bytes) {

            bytes = "";
        }
        sqlite3_bind_blob(pStmt, idx, bytes, (int)[obj length], SQLITE_STATIC);
    }
    else if ([obj isKindOfClass:[NSDate class]]) {
        if (self.dateFormatter)
            sqlite3_bind_text(pStmt, idx, [[self.dateFormatter stringFromDate:obj] UTF8String], -1, SQLITE_STATIC);
        else
            sqlite3_bind_text(pStmt, idx, [[self stringFromDate:obj] UTF8String],-1,SQLITE_STATIC);
    }
    else if ([obj isKindOfClass:[NSNumber class]]) {
        
        if (strcmp([obj objCType], @encode(char)) == 0) {
            sqlite3_bind_int(pStmt, idx, [obj charValue]);
        }
        else if (strcmp([obj objCType], @encode(unsigned char)) == 0) {
            sqlite3_bind_int(pStmt, idx, [obj unsignedCharValue]);
        }
        else if (strcmp([obj objCType], @encode(short)) == 0) {
            sqlite3_bind_int(pStmt, idx, [obj shortValue]);
        }
        else if (strcmp([obj objCType], @encode(unsigned short)) == 0) {
            sqlite3_bind_int(pStmt, idx, [obj unsignedShortValue]);
        }
        else if (strcmp([obj objCType], @encode(int)) == 0) {
            sqlite3_bind_int(pStmt, idx, [obj intValue]);
        }
        else if (strcmp([obj objCType], @encode(unsigned int)) == 0) {
            sqlite3_bind_int64(pStmt, idx, (long long)[obj unsignedIntValue]);
        }
        else if (strcmp([obj objCType], @encode(long)) == 0) {
            sqlite3_bind_int64(pStmt, idx, [obj longValue]);
        }
        else if (strcmp([obj objCType], @encode(unsigned long)) == 0) {
            sqlite3_bind_int64(pStmt, idx, (long long)[obj unsignedLongValue]);
        }
        else if (strcmp([obj objCType], @encode(long long)) == 0) {
            sqlite3_bind_int64(pStmt, idx, [obj longLongValue]);
        }
        else if (strcmp([obj objCType], @encode(unsigned long long)) == 0) {
            sqlite3_bind_int64(pStmt, idx, (long long)[obj unsignedLongLongValue]);
        }
        else if (strcmp([obj objCType], @encode(float)) == 0) {
            sqlite3_bind_double(pStmt, idx, [obj floatValue]);
        }
        else if (strcmp([obj objCType], @encode(double)) == 0) {
            NSLog(@"%f",[obj doubleValue]);
            sqlite3_bind_double(pStmt, idx, [obj doubleValue]);
        }
        else if (strcmp([obj objCType], @encode(BOOL)) == 0) {
            sqlite3_bind_int(pStmt, idx, ([obj boolValue] ? 1 : 0));
        }
        else {
            sqlite3_bind_text(pStmt, idx, [[obj description] UTF8String], -1, SQLITE_STATIC);
        }
    }
    else if ([obj isKindOfClass:[NSArray class]]||[obj isKindOfClass:[NSDictionary class]]) {
        @try {
            NSData* data = [NSJSONSerialization dataWithJSONObject:obj options:NSJSONWritingPrettyPrinted error:nil];
            NSString* jsonStr = [[NSString alloc] initWithData:data encoding:NSUTF8StringEncoding];
            sqlite3_bind_text(pStmt, idx, [[jsonStr description] UTF8String], -1, SQLITE_STATIC);
        }
        @catch (NSException *exception) {
            
        }
        @finally {
            
        }
        
    }else if ([obj isKindOfClass:NSClassFromString(@"UIImage")]) {
        NSData* data = UIImagePNGRepresentation(obj);
        const void *bytes = [data bytes];
        if (!bytes) {
            bytes = "";
        }
        sqlite3_bind_blob(pStmt, idx, bytes, (int)[data length], SQLITE_STATIC);
    }
    else {
        sqlite3_bind_text(pStmt, idx, [[obj description] UTF8String], -1, SQLITE_STATIC);
    }
}

- (NSString*)stringFromDate:(NSDate*)date
{
    NSDateFormatter* formatter = [[NSDateFormatter alloc] init];
    formatter.dateFormat = @"yyyy-HH-dd HH:MM:ss";
    return [formatter stringFromDate:date];
}

@end