#import "MDDQuery.h"

#import "MDDQuery+Private.h"

#import "MDDColumn.h"

#import "MDDTableInfo.h"

#import "MDDIndex.h"

#import "MDDItem.h"

#import "MDDDescription.h"

#import "MDDRange.h"

#import "MDDSort.h"

#import "MDDConditionSet+Private.h"

@implementation MDDQuery

@dynamic tableInfo;

+ (instancetype)queryWithTableInfo:(id<MDDTableInfo>)tableInfo objectClass:(Class<MDDObject>)objectClass;{

MDDQuery *query = [self descriptorWithTableInfo:tableInfo];

query.transform = ^id(NSDictionary *result) {

return [objectClass objectWithDictionary:result];

};

return query;

}

- (id)transformValue:(NSDictionary *)value;{

if (_transform) return _transform(value);

return value;

}

- (id<MDDTableInfo>)tableInfo{

return super.tableInfo ?: _conditionSet.tableInfo;

}

- (NSString *)description{

return [[self dictionaryWithValuesForKeys:@[@"tableInfo", @"set", @"conditionSet", @"range", @"property", @"sorts"]] description];

}

#pragma mark - accessor

- (MDDDescription *)SQLDescription{

NSMutableArray *values = [NSMutableArray array];

NSMutableArray<NSString *> *columns = [NSMutableArray<NSString *> array];

NSMutableSet<MDDTableInfo> *tableInfos = [NSMutableSet<MDDTableInfo> setWithObject:self.tableInfo];

[tableInfos unionSet:_conditionSet.mutableTableInfos];

for (MDDItem *property in _properties) {

MDDDescription *description = property.SQLDescription;

if (property.tableInfo) [tableInfos addObject:property.tableInfo];

[columns addObject:description.SQL];

[values addObjectsFromArray:description.values];

}

for (MDDSort *sort in _sorts) {

if (sort.tableInfo) [tableInfos addObject:sort.tableInfo];

}

NSString *property = [columns componentsJoinedByString:@", "];

property = [property length] ? property : @" * ";

MDDIndex *index = _conditionSet.index;

NSString *indexString = index ? [NSString stringWithFormat:@" INDEXED BY %@ ", index.name] : @"";

MDDDescription *description = _set.SQLDescription;

NSString *tableSet = nil;

if (description) {

tableSet = description.SQL;

[values addObjectsFromArray:description.values];

} else {

tableSet = [[tableInfos.allObjects valueForKey:@MDDKeyPath(MDDTableInfo, name)] componentsJoinedByString:@" , "];

}

NSMutableString *SQL = [NSMutableString stringWithFormat:@" SELECT %@ FROM %@ %@ ", property, tableSet, indexString];

if (_set) description = [_conditionSet SQLDescriptionInSet:_set];

else description = _conditionSet.SQLDescription;

if (description.SQL) {

[SQL appendFormat:@" WHERE %@ ", description.SQL];

[values addObjectsFromArray:description.values];

}

description = [MDDSort descriptionWithSorts:_sorts];

if (description.SQL) {

[SQL appendFormat:@" ORDER BY %@ ", description.SQL ?: @""];

}

NSRange range = _range;

if (range.location || range.length) {

range.length = range.length ?: INT_MAX;

[SQL appendFormat:@" LIMIT %lu OFFSET %ld ", (unsigned long)range.length, (unsigned long)range.location];

}

return [MDDDescription descriptionWithSQL:SQL values:values];

}