//
//  HDLPickerView.m
//  HDL_Widget_iOS
//
//  Created by HDL on 2019/9/25.
//  Copyright © 2019 JLChen. All rights reserved.
//

#import "HDLPickerView.h"
@interface HDLPickerView ()<UIPickerViewDataSource, UIPickerViewDelegate>

@property (nonatomic, strong) UIPickerView *pickerView;

//2019-9-5 新增默认选择项，和回调选中的索引
@property (nonatomic, assign) NSInteger selectIndex1;
@property (nonatomic, assign) NSInteger selectIndex2;
@property (nonatomic, assign) NSInteger selectIndex3;
//@property (nonatomic, strong) UIColor *btnColor;

@property (nonatomic, assign) BOOL isLinked; //是否不联动

@property (nonatomic, assign) NSUInteger component; // component numbers, default 0
@property (nonatomic, assign) CGFloat pickerViewHeight; // pickerView height, default 224 pt
@property (nonatomic, assign) CGFloat oneComponentRowHeight; // one component row height, default 32 pt
// property
@property (nonatomic, strong) NSMutableArray *dataList; // data list
@property (nonatomic, strong) NSMutableArray *mSecondDataList; // 二级 list
@property (nonatomic, strong) NSMutableArray *mThirdDataList; // 三级 list
//@property (nonatomic, strong) NSDictionary *selectRowTitleAttribute; // select row titlt attribute

@property (nonatomic, strong) NSDictionary *selectRowTitleAttribute0; // select row titlt attribute
@property (nonatomic, strong) NSDictionary *selectRowTitleAttribute1; // select row titlt attribute
@property (nonatomic, strong) NSDictionary *selectRowTitleAttribute2; // select row titlt attribute




@end

@implementation HDLPickerView



/*
 // Only override drawRect: if you perform custom drawing.
 // An empty implementation adversely affects performance during animation.
 - (void)drawRect:(CGRect)rect {
 // Drawing code
 }
 */
-(instancetype)initWithFrame:(CGRect)frame{
    self = [super initWithFrame:frame];
    if (self) {
        
        //        self.selectRowTitleAttribute = @{NSForegroundColorAttributeName : [UIColor blackColor], NSFontAttributeName : [UIFont systemFontOfSize:20.0f]};
        self.selectRowTitleAttribute0 = @{NSForegroundColorAttributeName : [UIColor blackColor], NSFontAttributeName : [UIFont systemFontOfSize:20.0f]};
        self.selectRowTitleAttribute1 = @{NSForegroundColorAttributeName : [UIColor blackColor], NSFontAttributeName : [UIFont systemFontOfSize:20.0f]};
        self.selectRowTitleAttribute2 = @{NSForegroundColorAttributeName : [UIColor blackColor], NSFontAttributeName : [UIFont systemFontOfSize:20.0f]};
        
        
        self.pickerViewHeight = 224.0f;
        self.oneComponentRowHeight = 32.0f;
        _dataList = [NSMutableArray array];
        _mSecondDataList = [NSMutableArray array];
        _mThirdDataList = [NSMutableArray array];
//        self.isLinked = NO;
        [self addSubview:self.pickerView];
        self.backgroundColor = [UIColor whiteColor];
    }
    return  self;
}


#pragma mark- 懒加载
- (UIPickerView *)pickerView{
    if (_pickerView == nil) {
        _pickerView = [[UIPickerView alloc] initWithFrame:CGRectMake(0, 0, self.frame.size.width, self.frame.size.height)];
        _pickerView.center = CGPointMake(self.frame.size.width/2, self.frame.size.height/2);
        _pickerView.dataSource = self;
        _pickerView.delegate = self;
        
    }
    return _pickerView;
}



#pragma mark - UIPickerView DataSource & Delegate
- (NSInteger)numberOfComponentsInPickerView:(UIPickerView *)pickerView
{
    return self.component;
}

- (NSInteger)pickerView:(UIPickerView *)pickerView numberOfRowsInComponent:(NSInteger)component
{
    
    if(self.isLinked){
        return [self getDataWithComponent:component].count;
//        return 0;
    }else{
        return [self getDataWithComponentNoLinked:component].count;
    }
    
}

- (NSString *)pickerView:(UIPickerView *)pickerView titleForRow:(NSInteger)row forComponent:(NSInteger)component
{
    NSArray *componentArray;
    if(self.isLinked){
        componentArray = [self getDataWithComponent:component];
    }else{
        componentArray = [self getDataWithComponentNoLinked:component];
    }
    
    if (componentArray.count) {
        if (row < componentArray.count) {
            id titleData = componentArray[row];
            if ([titleData isKindOfClass:[NSString class]]) {
                return titleData;
            } else if ([titleData isKindOfClass:[NSNumber class]]) {
                return [NSString stringWithFormat:@"%@", titleData];
            }
        }
    }
    return @"";
}

- (void)pickerView:(UIPickerView *)pickerView didSelectRow:(NSInteger)row inComponent:(NSInteger)component
{
    if(self.isLinked){
        if(component == 0){
            self.selectIndex1 = row;
            self.selectIndex2 = 0;
        }else if(component == 1){
            self.selectIndex2 = row;
            self.selectIndex3 = 0;
        }else if(component == 2){
            self.selectIndex3 = row;
        }
        
        [pickerView reloadAllComponents];
        for (NSUInteger i = 0; i < self.component; i++) {
            if (i > component) {
                [pickerView selectRow:0 inComponent:i animated:YES];
            }
        }
    }else{
        //        [pickerView selectRow:row inComponent:component animated:YES];
        if(component == 0){
            self.selectIndex1 = row;
            
        }else if(component == 1){
            self.selectIndex2 = row;
        }else if(component == 2){
            self.selectIndex3 = row;
        }
    }
    
    if(self.mChangedDelegate != nil){
        [self.mChangedDelegate onOptionsSelectChangeListener:_selectIndex1 selectIndex2:_selectIndex2 selectIndex3:_selectIndex3];
    }
    
}

- (CGFloat)pickerView:(UIPickerView *)pickerView rowHeightForComponent:(NSInteger)component
{
    return self.oneComponentRowHeight;
}

- (UIView *)pickerView:(UIPickerView *)pickerView viewForRow:(NSInteger)row forComponent:(NSInteger)component reusingView:(UIView *)view
{
    // set separateline color
    //    if (NO == self.isSettedSelectRowLineBackgroundColor) {
    //        UIView *topSeparateLine = [pickerView.subviews objectAtIndex:1];
    //        UIView *bottomSeparateLine = [pickerView.subviews objectAtIndex:2];
    //        if (topSeparateLine.frame.size.height < 1.0f &&
    //            bottomSeparateLine.frame.size.height < 1.0f) {
    //            topSeparateLine.backgroundColor = self.selectRowLineBackgroundColor;
    //            bottomSeparateLine.backgroundColor = self.selectRowLineBackgroundColor;
    //            self.isSettedSelectRowLineBackgroundColor = YES;
    //        } else {
    //            for (UIView *singleLine in pickerView.subviews) {
    //                if (singleLine.frame.size.height < 1.0f) {
    //                    singleLine.backgroundColor = self.selectRowLineBackgroundColor;
    //                    self.isSettedSelectRowLineBackgroundColor = YES;
    //                }
    //            }
    //        }
    //    }
    
    // custom pickerView content label
    UILabel *pickerLabel = (UILabel *)view;
    
    // discussion: this is always nil, not Reusing, it's an iOS System bug.
    // reference: https://stackoverflow.com/questions/20635949/reusing-view-in-uipickerview-with-ios-7/21039321#21039321
    if (!pickerLabel) {
        pickerLabel = [[UILabel alloc] init];
        pickerLabel.textAlignment = NSTextAlignmentCenter;
        pickerLabel.adjustsFontSizeToFitWidth = YES;
        pickerLabel.backgroundColor = [UIColor clearColor];
    }
    pickerLabel.attributedText = [self pickerView:pickerView attributedTitleForRow:row forComponent:component];
    
    return pickerLabel;
}
//
- (nullable NSAttributedString *)pickerView:(UIPickerView *)pickerView attributedTitleForRow:(NSInteger)row forComponent:(NSInteger)component
{
    NSString *normalRowString = [self pickerView:pickerView titleForRow:row forComponent:component];
    
    if(component == 1){
        return [[NSAttributedString alloc] initWithString:normalRowString attributes:self.selectRowTitleAttribute1];
    }else if(component == 2){
        return [[NSAttributedString alloc] initWithString:normalRowString attributes:self.selectRowTitleAttribute2];
    }else{
        return [[NSAttributedString alloc] initWithString:normalRowString attributes:self.selectRowTitleAttribute0];
    }
    
}

- (NSArray *)getDataWithComponentNoLinked:(NSInteger)component
{
    
    //    NSMutableArray *arrayM = [NSMutableArray array];
    if(component == 0){
        return self.dataList;
    }else if(component == 1){
        return self.mSecondDataList;
    }else if(component == 2){
        return self.mThirdDataList;
    }else{
        NSMutableArray *arrayM = [NSMutableArray array];
        return arrayM;
    }
}


- (NSArray *)getDataWithComponent:(NSInteger)component
{
    NSMutableArray *tempArray = [NSMutableArray arrayWithArray:self.dataList];
    NSMutableArray *arrayM = [NSMutableArray array];
    for (NSInteger i = 0; i <= component; i++) {
        if (i == component) {
            id data = tempArray.firstObject;
            if ([data isKindOfClass:[NSDictionary class]]) {
                NSMutableArray *tempTitleArray = [NSMutableArray arrayWithArray:tempArray];
                [tempArray removeAllObjects];
                [tempTitleArray enumerateObjectsUsingBlock:^(NSDictionary *dict, NSUInteger idx, BOOL * _Nonnull stop) {
                    [tempArray addObjectsFromArray:dict.allKeys];
                }];
                [arrayM addObjectsFromArray:tempArray];
            } else if ([data isKindOfClass:[NSString class]] ||
                       [data isKindOfClass:[NSNumber class]]){
                [arrayM addObjectsFromArray:tempArray];
            }
        } else {
            NSInteger selectRow = [self.pickerView selectedRowInComponent:i];
            if (selectRow < tempArray.count) {
                id data = tempArray[selectRow];
                if ([data isKindOfClass:[NSDictionary class]]) {
                    [tempArray removeAllObjects];
                    NSDictionary *dict = data;
                    [dict.allValues enumerateObjectsUsingBlock:^(id  _Nonnull obj, NSUInteger idx, BOOL * _Nonnull stop) {
                        if ([obj isKindOfClass:[NSArray class]]) {
                            [tempArray addObjectsFromArray:obj];
                        } else {
                            [tempArray addObject:obj];
                        }
                    }];
                }
            }
        }
    }
    return arrayM;
}

-(void)setCurrentItems:(NSInteger)selectIndex1 selectIndex2:(NSInteger)selectIndex2 selectIndex3:(NSInteger)selectIndex3{
    
    self.selectIndex1 = selectIndex1;
    self.selectIndex2 = selectIndex2;
    self.selectIndex3 = selectIndex3;
    [self.pickerView reloadAllComponents];
    [self scrollToSelectedRow];
    
}


/*
 * 2019-09-05 根据索引选中，替换原来根据传入字符选中方案
 */
- (void)scrollToSelectedRow
{
    for (NSUInteger i = 0; i < self.component; i++) {
        if(i == 0){
            [self.pickerView selectRow:self.selectIndex1 inComponent:i animated:NO];
        }else if(i == 1){
            [self.pickerView selectRow:self.selectIndex2 inComponent:i animated:NO];
        }else if(i == 2){
            [self.pickerView selectRow:self.selectIndex3 inComponent:i animated:NO];
        }
        
    }
}

/**
 重置布局
 
 @param mCGRect 布局
 
 */
-(void)initWithFrameArc:(CGRect) mCGRect{
    self.frame = mCGRect;
    _pickerView.frame =  self.frame;
    _pickerView.center = CGPointMake(self.frame.size.width/2, self.frame.size.height/2);
}


/**
 设置指定位置 分割线之间的文字的颜色
 */
-(void)setTextColorCenter:(UIColor *)mColor optionId:(NSInteger)optionId{
    if(optionId == 0){
        self.selectRowTitleAttribute0 = @{NSForegroundColorAttributeName : mColor, NSFontAttributeName : [UIFont systemFontOfSize:20.0f]};
    }else if(optionId == 1){
        self.selectRowTitleAttribute1 = @{NSForegroundColorAttributeName : mColor, NSFontAttributeName : [UIFont systemFontOfSize:20.0f]};
    }else if(optionId == 2){
        self.selectRowTitleAttribute2 = @{NSForegroundColorAttributeName : mColor, NSFontAttributeName : [UIFont systemFontOfSize:20.0f]};
    }
}

/**
 设置指定位置 分割线之间的文字的颜色
 */
-(void)setTextColorCenterAll:(UIColor *)mColor{
    self.selectRowTitleAttribute0 = @{NSForegroundColorAttributeName : mColor, NSFontAttributeName : [UIFont systemFontOfSize:20.0f]};
    self.selectRowTitleAttribute1 = @{NSForegroundColorAttributeName : mColor, NSFontAttributeName : [UIFont systemFontOfSize:20.0f]};
    self.selectRowTitleAttribute2 = @{NSForegroundColorAttributeName : mColor, NSFontAttributeName : [UIFont systemFontOfSize:20.0f]};
}


/**
 单独设置指定位置 分割线之间的文字的颜色
 */
-(void)setTextColorCenterAlone:(UIColor *)mColor0 mColor1:(UIColor *)mColor1 mColor2:(UIColor *)mColor2 {
    self.selectRowTitleAttribute0 = @{NSForegroundColorAttributeName : mColor0, NSFontAttributeName : [UIFont systemFontOfSize:20.0f]};
    self.selectRowTitleAttribute1 = @{NSForegroundColorAttributeName : mColor1, NSFontAttributeName : [UIFont systemFontOfSize:20.0f]};
    self.selectRowTitleAttribute2 = @{NSForegroundColorAttributeName : mColor2, NSFontAttributeName : [UIFont systemFontOfSize:20.0f]};
}

- (void)setNPicker:(nonnull NSArray *)dataList
       mSecondList:(nullable NSArray *)mSecondList
        mThirdList:(nullable NSArray *)mThirdList{
    self.isLinked = NO;
    //配置关键数据
    if (dataList && dataList.count > 0) {
        self.component = 1;
        [_dataList removeAllObjects];
        [_dataList addObjectsFromArray:dataList];
    }
    
    
    
    if (mSecondList && mSecondList.count > 0) {
        self.component = 2;
        [_mSecondDataList removeAllObjects];
        [_mSecondDataList addObjectsFromArray:mSecondList];
    }
    
    
    if (mThirdList && mThirdList.count > 0) {
        self.component = 3;
        [_mThirdDataList removeAllObjects];
        [_mThirdDataList addObjectsFromArray:mThirdList];
    }
    
}

-(void)setPicker:(nonnull NSArray *)dataList{
    
    // no data
    if (!dataList || dataList.count == 0) {
        return;
    }
    
    [_dataList removeAllObjects];
    [_dataList addObjectsFromArray:dataList];
    // calculate component num
    id data = dataList.firstObject;
    if ([data isKindOfClass:[NSString class]] ||
        [data isKindOfClass:[NSNumber class]] ) {
        self.component = 1;
    } else if ([data isKindOfClass:[NSDictionary class]]) {
        self.component ++;
        [self handleDictDataList:dataList];
    } else {
        NSLog(@"ZJPickerView error tip：\"Unsupported data type\"");
        return;
    }
    
    self.isLinked = YES;
}
    
#pragma mark - private method
- (void)handleDictDataList:(NSArray *)list{
        id data = list.firstObject;
        if ([data isKindOfClass:[NSDictionary class]]) {
            NSDictionary *dict = data;
            id value = dict.allValues.firstObject;
            if ([value isKindOfClass:[NSArray class]]) {
                self.component++;
                [self handleDictDataList:value];
            }
        }
    
}
    

@end