Swift 重构:如何优雅的实现折叠列表

嵌套列表麻烦在数据处理,通常无非是字典+数组的方式处理; 灵光一闪能不能用元祖解决这个问题呢? 遂动手实现,效果极好;
实现优点:
1、因为使用元祖,省去了嵌套数据的处理逻辑,降低程序复杂度;
2、代码量极少;
3、将子列表的实体用泛型代替,可以随意传入任意数据类型;
4、可以通过 cellForRowBlock 传入自定义 cell、可以通过 headerViewForSectionBlock 传入自定义 headerView 极大的扩展重用用范围;

需求: 实现一个简单的系统字体选择页面

examle:

           let vc = NNFoldTableController<String>()
            vc.data = UIFont.familyNames
                .map({ return ($0, UIFont.fontNames(forFamilyName: $0), false) })
                .sorted{$0.0 < $1.0}
            vc.block = {
                DDLog($0)
            }
            
//            vc.headerViewForSectionBlock = { UITableView, Int, arg2 in
//                let view = UILabel()
//                view.backgroundColor = .red
//
//                view.text = "\(arg2.0)(\(arg2.1.count))"
//                return view
//            }
            navigationController?.pushViewController(vc, animated: true)

截图:

Simulator Screen Shot - iPhone 11 Pro - 2021-06-19 at 12.13.13.png

源码

//
//  NNFoldTableController.swift
//  SwiftTemplet
//
//  Created by Bin Shang on 2021/6/18.
//  Copyright © 2021 BN. All rights reserved.
//

import UIKit
import SwiftExpand

///折叠列表选择器
class NNFoldTableController<E: Any>: UIViewController, UITableViewDataSource, UITableViewDelegate {
    
        
    public var cellForRowBlock: ((UITableView, IndexPath, E) -> UITableViewCell)?
    
    public var headerViewForSectionBlock: ((UITableView, Int, (String, [E], Bool)) -> UIView)?

    public var block: ((E?)->Void)?

    var indexP: IndexPath?

    var element: E?

    let identifier = String(describing: UITableViewCell.self)
    
    public var tableView: UITableView {
        let view = UITableView(frame: self.view.bounds, style: .plain)
        view.autoresizingMask = [.flexibleWidth, .flexibleHeight]
        view.separatorStyle = .singleLine
        view.separatorInset = .zero
        view.rowHeight = 50
//        view.register(UITableViewCell.self, forCellReuseIdentifier: NSStringFromClass(UITableViewCell.self))
        view.register(UITableViewCell.self, forCellReuseIdentifier: identifier)
        view.keyboardDismissMode = .onDrag
        view.backgroundColor = .groupTableViewBackground
//        view.tableHeaderView = UIView();
//        view.tableFooterView = UIView()

        view.estimatedRowHeight = 0;
        view.estimatedSectionHeaderHeight = 0;
        view.estimatedSectionFooterHeight = 0;
        
        view.dataSource = self
        view.delegate = self
        
        return view
    }
    
    public var heightForHeaderInSection: CGFloat = 45
    ///数据源(section 标题, 子数组, 折叠/展开)
    public var data: [(String, [E], Bool)] = []

    ///只是箭头
    var indicator: UIImage?

    // MARK: -lifecycle
    convenience init(block: ((E?)->Void)?) {
        self.init()
        self.block = block
    }
    
    override func viewDidLoad() {
        super.viewDidLoad()

        // Do any additional setup after loading the view.
        edgesForExtendedLayout = []
        title = "请选择"
        navigationItem.rightBarButtonItems = ["确定",]
            .map { UIBarButtonItem(title: $0, style: .plain, target: self, action: #selector(actionItem(_:))) }
        
        view.addSubview(tableView)
    }
    
    override func viewWillAppear(_ animated: Bool) {
        super.viewWillAppear(animated)
                
    }
    
    @objc func actionItem(_ sender: UIBarButtonItem) {
        block?(element)
//        DDLog(presentationController, presentedViewController, presentingViewController)
        if presentingViewController != nil {
            navigationController?.dismiss(animated: true, completion: nil)
            return
        }
        navigationController?.popViewController(animated: true);
    }
    
    // MARK: -tableView
    func numberOfSections(in tableView: UITableView) -> Int {
        return data.count
    }
    
    func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int {
        return data[section].2 ? data[section].1.count : 0
    }
    
    func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
        let cell = tableView.dequeueReusableCell(withIdentifier: identifier, for: indexPath)
        cell.textLabel?.text = "\(indexPath.row)"
        
        cell.backgroundColor = .white.withAlphaComponent(0.5)

        let element = data[indexPath.section].1[indexPath.row]
        guard let newCell = cellForRowBlock?(tableView, indexPath, element) else {
            if let text = element as? String {
                cell.textLabel?.text = text
                
               if let font = UIFont(name: text, size: UIFont.labelFontSize) {
                    cell.textLabel?.font = font
                }

                cell.accessoryType = indexP != nil && indexP == indexPath ? .checkmark : .none
                cell.textLabel?.textColor = indexP != nil && indexP == indexPath ? .systemBlue : .black
            }
            return cell }
        
        return newCell
    }
    
    func tableView(_ tableView: UITableView, heightForHeaderInSection section: Int) -> CGFloat {
        return heightForHeaderInSection
    }
    
    func tableView(_ tableView: UITableView, viewForHeaderInSection section: Int) -> UIView? {
        let label = UILabel()
        label.text = data[section].0

        let imageView = UIImageView(image: indicator ?? UIImage(named: "img_arrowRight_gray"))
        imageView.isUserInteractionEnabled = true
        imageView.isHidden = data[section].1.count == 0
        imageView.transform = data[section].2 == false ? .identity : imageView.transform.rotated(by: CGFloat(Double.pi/2))
        
        let items = [UILabel(), label, imageView, UILabel()].filter { $0.isHidden == false }
        let view = UIStackView.init(subviews: items,
                                    axis: .horizontal,
                                    spacing: 8,
                                    alignment: .center,
                                    distribution: .fill)
        view.addGestureTap { reco in
            self.data[section].2.toggle()
            tableView.reloadSections([section], with: .automatic)
        }
        
        view.backgroundColor = .white
        
        guard let headerView = headerViewForSectionBlock?(tableView, section, data[section]) else {
            return view
        }
        headerView.addGestureTap { reco in
            self.data[section].2.toggle()
            tableView.reloadSections([section], with: .automatic)
        }
        return headerView
    }
    
    func tableView(_ tableView: UITableView, didSelectRowAt indexPath: IndexPath) {
        element = data[indexPath.section].1[indexPath.row]
//        print("\(Date()) \(#function) \(#line): \(element)")
        
        if indexP == nil {
            let cell = tableView.cellForRow(at: indexPath)
            cell?.accessoryType = .checkmark
            indexP = indexPath

        } else if indexP != indexPath  {
            let newCell = tableView.cellForRow(at: indexPath)
            newCell?.accessoryType = .checkmark
            newCell?.textLabel?.textColor = .systemBlue

            let oldCell = tableView.cellForRow(at: indexP!)
            oldCell?.accessoryType = .none
            oldCell?.textLabel?.textColor = .black
            indexP = indexPath
        }
    }
}

NNFoldTableController.swift

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 217,509评论 6 504
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,806评论 3 394
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 163,875评论 0 354
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,441评论 1 293
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,488评论 6 392
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,365评论 1 302
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,190评论 3 418
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 39,062评论 0 276
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,500评论 1 314
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,706评论 3 335
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,834评论 1 347
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,559评论 5 345
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,167评论 3 328
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,779评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,912评论 1 269
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,958评论 2 370
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,779评论 2 354

推荐阅读更多精彩内容