richardfrk / ericsmark Goto Github PK

func perform(with invocation: XCSourceEditorCommandInvocation, completionHandler: @escaping (Error?) -> Void ) -> Void
{
    var lineClassMark:Int?
    var linePropertieMark:Int?
    var lineIBOutletMark:Int?
    var lineSuperMethodMark:Int?
    var lineIBActionMark:Int?
    var linePrivateMethodMark:Int?
    var lineExtensionMark:Int?
    
    var bRemove  = true
    let array = ["//MARK: - IBOutlets","//MARK: - Properties","//MARK: - Super Methods","//MARK: - Private Methods","//MARK: - Extensions","//MARK: - IBActions"]
    while ( bRemove == true)
    {
        bRemove = false
        for i in 0..<invocation.buffer.lines.count
        {
            let line = invocation.buffer.lines[i] as! String
    
            if array.contains(line) == true
            {
                invocation.buffer.lines.removeObject(at: i)
                bRemove = true
                break
            }
        }
    }