♻️ cleanup

Author: igorkulman

LocalizationEditor/Providers/LocalizationProvider.swift

@@ -53,17 +53,17 @@ class LocalizationProvider {
             Log.debug?.message("Same value provided for \(string)")
             return
         }
-        
+
         Log.debug?.message("Updating \(string) with \(value) in \(localization)")
-        
+
         string.update(value: value)
-       
+
         let data = localization.translations.map { string in
             "\"\(string.key)\" = \"\(string.value.replacingOccurrences(of: "\"", with: "\\\""))\";"
-            }.reduce("") { (prev, next) in
-                "\(prev)\n\(next)"
+        }.reduce("") { prev, next in
+            "\(prev)\n\(next)"
         }
-        
+
         do {
             try data.write(toFile: localization.path, atomically: false, encoding: .utf8)
             Log.debug?.message("Localization file for \(localization) updated")

LocalizationEditor/Providers/LocalizationsDataSource.swift

@@ -13,18 +13,18 @@ class LocalizationsDataSource: NSObject, NSTableViewDataSource {
 
     // MARK: - Properties
 
-    private(set) var localizations: [Localization] = []
-
+    private var localizations: [Localization] = []
     private var masterLocalization: Localization?
     private let localizationProvider = LocalizationProvider()
     private var numberOfKeys = 0
 
     // MARK: - Action
 
-    func load(folder: URL) {
+    func load(folder: URL) -> [String] {
         localizations = localizationProvider.getLocalizations(url: folder)
         numberOfKeys = localizations.map({ $0.translations.count }).max() ?? 0
         masterLocalization = localizations.first(where: { $0.translations.count == numberOfKeys })
+        return localizations.map({ $0.language })
     }
 
     func getKey(row: Int) -> String? {

LocalizationEditor/UI/ViewController.swift

@@ -43,7 +43,7 @@ class ViewController: NSViewController {
         tableView.dataSource = dataSource
     }
 
-    func reloadData() {
+    func reloadData(with languages: [String]) {
         let columns = tableView.tableColumns
         columns.forEach {
             self.tableView.removeTableColumn($0)
@@ -54,10 +54,10 @@ class ViewController: NSViewController {
         column.width = 200
         tableView.addTableColumn(column)
 
-        dataSource.localizations.forEach { localization in
-            let column = NSTableColumn(identifier: NSUserInterfaceItemIdentifier(localization.language))
-            column.title = localization.language.uppercased()
-            column.width = (self.view.bounds.width - 200.0) / CGFloat(self.dataSource.localizations.count)
+        languages.forEach { language in
+            let column = NSTableColumn(identifier: NSUserInterfaceItemIdentifier(language))
+            column.title = language.uppercased()
+            column.width = (self.view.bounds.width - 200.0) / CGFloat(languages.count)
             self.tableView.addTableColumn(column)
         }
 
@@ -73,8 +73,8 @@ class ViewController: NSViewController {
         openPanel.begin { [unowned self] (result) -> Void in
             if result.rawValue == NSApplication.ModalResponse.OK.rawValue {
                 if let url = openPanel.url {
-                    self.dataSource.load(folder: url)
-                    self.reloadData()
+                    let languages = self.dataSource.load(folder: url)
+                    self.reloadData(with: languages)
                 }
             }
         }