FixRekordboxPaths.swift

// Copyright (c) 2024 Hidden Spectrum, LLC

// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:

// The above copyright notice and this permission notice shall be included in all
// copies or substantial portions of the Software.

// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
// SOFTWARE.

// See instructions in comments. Requires latest Xcode installed (available on App Store)
// Place this file in your home directory, open terminal, and run `swift ~/FixRekordboxPaths.swift`

import Cocoa
import Darwin

// CHANGE THESE VALUES
let exportedXmlFilePath = "/path/to/RekordboxLibrary.xml"
let newXmlFilePath = "/path/to/RekordboxLibraryFixed.xml"
let oldTracksPath = "/incorrect/path/to/tracks" // No trailing slash
let fixedTracksPath = "/correct/path/to/tracks" // No trailing slash

let xmlDocument = try XMLDocument(contentsOf: URL(fileURLWithPath: exportedXmlFilePath))

guard let root = xmlDocument.rootElement()?.elements(forName: "COLLECTION").first,
      let tracks = root.children
else {
    print("No root / tracks")
    exit(0)
}

for (index, trackNode) in tracks.enumerated() {
    guard let track = trackNode as? XMLElement,
          let existingUrlString = track.attribute(forName: "Location")?.stringValue,
          let existingUrl = URL(string: existingUrlString)
    else {
        continue
    }
    
    let correctedPath = existingUrl.path.replacingOccurrences(of: oldTracksPath, with: fixedTracksPath)
    guard let newFilePath = correctedPath.addingPercentEncoding(withAllowedCharacters: .urlPathAllowed) else {
        print("Couldn't escape file path")
        continue
    }
    let attributes = track.attributes
    
    guard let locationAttributeIndex = attributes?.firstIndex(where: { node in
        node.name == "Location"
    }) else {
        continue
    }
    
    print("[\(index + 1) / \(tracks.count)] Setting new file path: \(newFilePath)")
    track.attributes?[locationAttributeIndex].objectValue = "file://localhost\(newFilePath)"
}

try xmlDocument.xmlData.write(to: URL(fileURLWithPath: newXmlFilePath))