iOS 头文件引用排序、去重和分堆小工具

整理陈旧的项目代码时,常会面对一个 class 引用的头文件特别多。里面可能也偶尔会有重复引用。这个脚本用于自动排序和按一定规则分堆 iOS 项目中的头文件引用,并且支持去除重复引用。适用于 Objective-C 项目。使得代码清晰,整洁,头文件引用去除冗余。

功能特点

  • 头文件引用排序和分堆:根据一定的规则对头文件引用进行排序,提高代码可读性,也整洁。

  • 去除重复引用:自动检测并去除重复的头文件引用。

  • 头文件引用少于 5 个的文件跳过。

排序和分堆规则

  1. 如果头文件名以 View 结尾,归为一堆。(不区分大小写,以下同样

  2. 如果头文件名以 btnbutton 结尾,归为一堆。

  3. 如果头文件名以 VCViewcontroller 结尾,归为一堆。

  4. 如果头文件名中包含 helpermanager,归为一堆。

  5. 如果头文件名中包含 + 号,归为一堆。

  6. 其它头文件按字母排序,归为一堆。 (当然你可以修改脚本,自行定义。

主要代码

python 复制代码
def start_processing(directory):

    for root, _ *, files in os.walk(directory):*

*for file in files:*

*if any(file.endswith(suffix) for suffix in ['.m', '.mm', '.h']):*

*file_path = os.path.join(root, file)*

*print(f"处理文件: {file_path}")*

*header_refs = find_header_refs(file_path)*

  


                # 如果头文件引用数目少于 5 个,则跳过此文件

                if len(header_refs) < 5:

                    print(f"跳过文件 {file_path},头文件引用数目少于 5 个")

                    continue

  


                sorted_header_groups = sort_and_group_headers(header_refs)

  


                # 读取文件内容

                with open(file_path, 'r') as f:

                    lines = f.readlines()

  


                # 查找插入排序后的起始位置

                first_import_position = -1

                last_import_position = -1

                for i, line in enumerate(lines):

                    if line.startswith('#import "') or line.startswith('#import <'):

                        if first_import_position == -1:

                            first_import_position = i

                        last_import_position = i

  


                # 删除原始的引用头文件

                del lines[first_import_position:last_import_position + 1]

  


                count = len(sorted_header_groups)

                # 插入sort后的头文件引用

                for i in range(count):

                    group = sorted_header_groups[i]

                    if group:

                        if i == count - 1:

                            lines.insert(first_import_position, '\n'.join(group) + '\n')

                        else:

                            lines.insert(first_import_position, '\n'.join(group) + '\n' + '\n')

                            first_import_position += 1

  


                # 把修改后的内容写回文件

                with open(file_path, 'w') as f:

                    f.writelines(lines)

示例

排序分堆去重前:

objective-c 复制代码
//

//  THSortImportHelper.m

//

//  Created by triple on 20xx/xx/xx.

//

  


#import "THAdd.h"

#import "THSortImportClass.h"

#import "ViewController.h"

#import "ViewController.h"

#import "ViewController.h"

#import "ViewController.h"

#import "TH1Btn.h"

#import "THCaange.h"

#import "THChange.h"

#import "NSString+str2.h"

#import "NSString+str2.h"

#import "THFind.h"

#import "TH2View.h"

#import "NSString+str1.h"

  


@implementation THSortImportHelper

  


@end

排序分堆去重后:

objective-c 复制代码
//

//  THSortImportHelper.m

//

//  Created by tripleEye on 20xx/xx/xx.

//

  


#import "TH2View.h"

  


#import "TH1Btn.h"

  


#import "ViewController.h"

  


#import "NSString+str1.h"

#import "NSString+str2.h"

  


#import "THAdd.h"

#import "THCaange.h"

#import "THChange.h"

#import "THFind.h"

#import "THSortImportClass.h"

  


@implementation THSortImportHelper

  


@end

注意事项

  • 脚本会修改原始文件,请提前备份项目。

  • 建议在对整个项目进行操作之前,在一个小范围的测试项目上尝试使用脚本。

项目地址

sort_iOS_header_refs

相关推荐
nbsaas-boot12 分钟前
Java 正则表达式白皮书:语法详解、工程实践与常用表达式库
开发语言·python·mysql
仗剑_走天涯15 分钟前
基于pytorch.nn模块实现线性模型
人工智能·pytorch·python·深度学习
chao_78916 分钟前
二分查找篇——搜索旋转排序数组【LeetCode】两次二分查找
开发语言·数据结构·python·算法·leetcode
一斤代码43 分钟前
vue3 下载图片(标签内容可转图)
前端·javascript·vue
中微子1 小时前
React Router 源码深度剖析解决面试中的深层次问题
前端·react.js
光影少年1 小时前
从前端转go开发的学习路线
前端·学习·golang
中微子1 小时前
React Router 面试指南:从基础到实战
前端·react.js·前端框架
3Katrina1 小时前
深入理解 useLayoutEffect:解决 UI "闪烁"问题的利器
前端·javascript·面试
前端_学习之路2 小时前
React--Fiber 架构
前端·react.js·架构
伍哥的传说3 小时前
React 实现五子棋人机对战小游戏
前端·javascript·react.js·前端框架·node.js·ecmascript·js