88
99import Cocoa
1010
11- class HomeDataSource {
11+ struct HomeDataSource {
1212 /** 头部注释 */
1313 var fileHeaderArr : Array = [ String] ( )
14- /** json数据 */
15- var jsonDic : Dictionary = [ String: Any] ( )
1614 /** 继承 */
17- var inheritanceStr : String = " "
18- /** 内容 */
19- var contentArr : Array = [ HomeContentModel] ( )
20- /** struct名称 */
21- var structName : String = " "
15+ var inheritanceStr : String
16+ /* 内容 */
17+ var contentArr : Array < HomeContentModel >
18+ /** 文件名称 */
19+ var fileName : String
20+ /** 项目名称 */
21+ var productName : String
2222
23- init ( ) {
23+ init ( inheritanceStr: String , contentArr: [ HomeContentModel ] , fileName: String , productName: String ) {
24+ self . inheritanceStr = inheritanceStr
25+ self . contentArr = contentArr
26+ self . fileName = fileName
27+ self . productName = productName
28+
2429 let keyword = APPKeyword ( )
2530 fileHeaderArr. removeAll ( )
2631 fileHeaderArr. append ( keyword. annotationStr)
27- fileHeaderArr. append ( " " )
28- fileHeaderArr. append ( " \( keyword. annotationStr) \( App . appName) " )
32+ fileHeaderArr. append ( " \( keyword. annotationStr) \( fileName) .swift " )
33+ let product = productName. isEmpty ? " __projectName__ " : productName
34+ fileHeaderArr. append ( " \( keyword. annotationStr) \( product) " )
2935 fileHeaderArr. append ( keyword. annotationStr)
30- fileHeaderArr. append ( " " )
31- fileHeaderArr. append ( " \( keyword. annotationStr) \( App . copyright ) " )
36+ fileHeaderArr. append ( " \( keyword . annotationStr ) Created by \( NSUserName ( ) ) on \( Date ( ) . formattingDate ( Date . dateFormatSlashDay ) ) " )
37+ fileHeaderArr. append ( " \( keyword. annotationStr) Copyright © \( Date ( ) . formattingDate ( Date . dateForematYear ) ) 年 \( NSUserName ( ) ) . All rights reserved. " )
3238 fileHeaderArr. append ( keyword. annotationStr)
3339 }
40+ }
41+
42+ // struct模型
43+ struct HomeStructModel {
44+ /** 名称 */
45+ var structName : String
46+ /** 继承 */
47+ var inheritanceStr : String
48+ /** 内容 */
49+ var contentArr : Array < HomeContentModel >
3450
51+ init ( structName: String , contentArr: [ HomeContentModel ] , inheritanceStr: String = " " ) {
52+ self . structName = structName
53+ self . contentArr = contentArr
54+ self . inheritanceStr = inheritanceStr
55+ }
3556}
3657
58+
59+ // 内容
3760class HomeContentModel {
3861 /** key */
3962 var key : String
@@ -73,22 +96,31 @@ class HomeContentModel {
7396
7497struct FileDataModel {
7598
99+ // MARK:- public
76100 // 创建
77- static public func createFileSwift( _ name : String , homeData: HomeDataSource , success: ( Bool ) -> ( ) ) {
101+ static public func createFileSwift( _ fileUrl : URL , homeData: HomeDataSource , success: ( Bool ) -> ( ) ) {
78102 let manager = FileManager . default
79- let fileName = " \( name) .swift "
80- let fileUrl = getFilePath ( ) . appendingPathComponent ( fileName)
81103 if manager. fileExists ( atPath: fileUrl. path) {
82104 // 新增数据
83105 let fileHandle = FileHandle ( forUpdatingAtPath: fileUrl. path)
84106 fileHandle? . seekToEndOfFile ( )
85107
86108 // 添加模型
87- wirteStructModel ( fileHandle, dataArray: flatStructModel ( homeData) )
109+ let structModel = HomeStructModel ( structName: homeData. fileName, contentArr: homeData. contentArr, inheritanceStr: homeData. inheritanceStr)
110+ wirteStructModel ( fileHandle, dataArray: flatStructModel ( structModel) )
88111
89112 fileHandle? . closeFile ( )
90113 return
91114 }
115+
116+ do {
117+ var arr = fileUrl. path. components ( separatedBy: " / " )
118+ arr. removeLast ( )
119+ let path = arr. joined ( separator: " / " )
120+ try . createDirectory ( atPath: path, withIntermediateDirectories: true , attributes: nil )
121+ } catch {
122+ }
123+
92124 if manager. createFile ( atPath: fileUrl. path, contents: nil , attributes: nil ) {
93125 let fileHandle = FileHandle ( forUpdatingAtPath: fileUrl. path)
94126 let appKeyword = APPKeyword ( )
@@ -110,7 +142,8 @@ struct FileDataModel {
110142 }
111143
112144 // 添加模型
113- wirteStructModel ( fileHandle, dataArray: flatStructModel ( homeData) )
145+ let structModel = HomeStructModel ( structName: homeData. fileName, contentArr: homeData. contentArr, inheritanceStr: homeData. inheritanceStr)
146+ wirteStructModel ( fileHandle, dataArray: flatStructModel ( structModel) )
114147
115148 // 关闭文件
116149 fileHandle? . closeFile ( )
@@ -151,48 +184,43 @@ struct FileDataModel {
151184 return array
152185 }
153186
154- // 扁平数据处理
155- static fileprivate func flatStructModel( _ homeData: HomeDataSource ) -> [ HomeDataSource ] {
156- var array = [ HomeDataSource] ( )
157-
158- let dataSource = HomeDataSource ( )
159- dataSource. structName = homeData. structName
160- dataSource. contentArr = homeData. contentArr
161- dataSource. inheritanceStr = homeData. inheritanceStr
162- array. append ( dataSource)
187+ // MARK: - fileprivate
188+ // 数据处理
189+ static fileprivate func flatStructModel( _ structData: HomeStructModel ) -> [ HomeStructModel ] {
190+ var array = [ HomeStructModel] ( )
191+ array. append ( structData)
163192
164- for item in homeData. contentArr {
165- if item. childArr. count > 0 {
166- let source = HomeDataSource ( )
167- source. structName = item. outputType. writeStr
168- source. contentArr = item. childArr
169- source. inheritanceStr = homeData. inheritanceStr
170- array += flatStructModel ( source)
193+ for item in structData. contentArr {
194+ if item. childArr. count > 0 && !item. isIgnore {
195+ let model = HomeStructModel ( structName: item. childName, contentArr: item. childArr, inheritanceStr: structData. inheritanceStr)
196+ array += flatStructModel ( model)
171197 }
172- }
198+ }
199+
173200 return array
174201 }
175202
176203 // 添加结构体
177- static fileprivate func wirteStructModel( _ fileHandle: FileHandle ? , dataArray: [ HomeDataSource ] ) {
204+ static fileprivate func wirteStructModel( _ fileHandle: FileHandle ? , dataArray: [ HomeStructModel ] ) {
178205 dataArray. forEach {
179- addStructModel ( fileHandle, homeData : $0)
206+ addStructModel ( fileHandle, structModel : $0)
180207 }
181208 }
182209
183- static fileprivate func addStructModel( _ fileHandle: FileHandle ? , homeData: HomeDataSource ) {
210+ // 写文件
211+ static fileprivate func addStructModel( _ fileHandle: FileHandle ? , structModel: HomeStructModel ) {
184212 let appKeyword = APPKeyword ( )
185213 // 机构体开头
186214 fileHandle? . write ( Data . newlineData ( ) )
187215 var structStart : String = " "
188- if homeData . inheritanceStr == appKeyword. handy {
189- structStart = " \( appKeyword. uct) \( homeData . structName) \( appKeyword. colon) \( appKeyword. handy) \( appKeyword. lPar) "
216+ if structModel . inheritanceStr == appKeyword. handy {
217+ structStart = " \( appKeyword. uct) \( structModel . structName) \( appKeyword. colon) \( appKeyword. handy) \( appKeyword. lPar) "
190218 } else {
191- structStart = " \( appKeyword. uct) \( homeData . structName) \( appKeyword. lPar) "
219+ structStart = " \( appKeyword. uct) \( structModel . structName) \( appKeyword. lPar) "
192220 }
193221 fileHandle? . write ( structStart. wirteData)
194-
195- for model in homeData . contentArr {
222+
223+ for model in structModel . contentArr {
196224 if !model. isIgnore {
197225 if !model. annotation. isEmpty {
198226 fileHandle? . write ( Data . newlineData ( ) )
@@ -207,23 +235,7 @@ struct FileDataModel {
207235 fileHandle? . write ( Data . newlineData ( ) )
208236 fileHandle? . write ( " \( appKeyword. rPar) " . wirteData)
209237 fileHandle? . write ( Data . newlineData ( ) )
210-
211- }
212-
213- // 获取桌面路径 创建时间文件夹
214- static fileprivate func getFilePath( ) -> URL {
215- let manager = FileManager . default
216- let urlForDocument = manager. urls ( for: . desktopDirectory, in: . userDomainMask)
217- let url = urlForDocument [ 0 ]
218- let folderUrl = url. appendingPathComponent ( Date ( ) . formattingDate ( ) )
219- if !manager. fileExists ( atPath: folderUrl. path) {
220- do {
221- try . createDirectory ( at: folderUrl, withIntermediateDirectories: true , attributes: nil )
222- } catch {
223- return url
224- }
225- }
226- return folderUrl
238+
227239 }
228240
229241}
0 commit comments