支持SQLITE3结尾的文件名

Author: yanqizhou

DebugDatabase/DebugDatabaseManager.m

@@ -227,22 +227,35 @@ - (NSDictionary*)getAllDBPathsWithDirectories:(NSArray*)directories {
     NSMutableDictionary *paths = @{}.mutableCopy;
 
     for (NSString *directory in directories) {
-        NSArray *dirList = [[[NSFileManager defaultManager] subpathsAtPath:directory] pathsMatchingExtensions:@[@"sqlite",@"SQLITE",@"db",@"DB"]];
+        NSArray *dirList = [[[NSFileManager defaultManager] subpathsAtPath:directory] pathsMatchingExtensions:[self databaseSuffixs]];
 
         for (NSString *subPath in dirList) {
-            if ([subPath hasSuffix:@"sqlite"] || [subPath hasSuffix:@"SQLITE"]|| [subPath hasSuffix:@"db"]|| [subPath hasSuffix:@"DB"]) {
-                [paths setObject:[directory stringByAppendingPathComponent:subPath] forKey:subPath.lastPathComponent];
+            if ([self checkDatabaseFile:subPath]) {
+                 [paths setObject:[directory stringByAppendingPathComponent:subPath] forKey:subPath.lastPathComponent];
             }
         }
 
-        if ([directory hasSuffix:@"sqlite"] || [directory hasSuffix:@"SQLITE"]|| [directory hasSuffix:@"db"]|| [directory hasSuffix:@"DB"]) {
+        if ([self checkDatabaseFile:directory]) {
             [paths setObject:directory forKey:directory.lastPathComponent];
         }
     }
 
     return paths;
 }
 
+- (BOOL)checkDatabaseFile:(NSString*)fileName {
+    for (NSString *suffix in [self databaseSuffixs]) {
+        if ([fileName hasSuffix:suffix]) {
+            return YES;
+        }
+    }
+    return NO;
+}
+
+- (NSArray*)databaseSuffixs {
+    return @[@"sqlite", @"SQLITE", @"db", @"DB", @"sqlite3", @"SQLITE3"];
+}
+
 - (NSString*)getTableNameFromQuery:(NSString*)query {
     NSArray *words = [query  componentsSeparatedByString:@" "];
     NSString *operator = [[words firstObject] lowercaseString];

YYDebugDatabase.podspec

@@ -8,7 +8,7 @@
 
 Pod::Spec.new do |s|
   s.name             = 'YYDebugDatabase'
-  s.version          = '2.0.4'
+  s.version          = '2.0.5'
   s.summary          = 'easy way to process splite database'
 
   s.homepage         = 'https://y500.me'