Initial Commit

Author: Rubenfer

.gitignore

@@ -0,0 +1,6 @@
+.DS_Store
+/.build
+/Packages
+/*.xcodeproj
+xcuserdata/
+/.swiftpm

Package.resolved

@@ -0,0 +1,29 @@
+// swift-tools-version:5.1
+// The swift-tools-version declares the minimum version of Swift required to build this package.
+
+import PackageDescription
+
+let package = Package(
+    name: "XMLTV",
+    products: [
+        // Products define the executables and libraries produced by a package, and make them visible to other packages.
+        .library(
+            name: "XMLTV",
+            targets: ["XMLTV"]),
+    ],
+    dependencies: [
+        // Dependencies declare other packages that this package depends on.
+        // .package(url: /* package url */, from: "1.0.0"),
+        .package(url: "https://github.com/Rubenfer/LightXMLParser.git", from: "0.1.0"),
+    ],
+    targets: [
+        // Targets are the basic building blocks of a package. A target can define a module or a test suite.
+        // Targets can depend on other targets in this package, and on products in packages which this package depends on.
+        .target(
+            name: "XMLTV",
+            dependencies: ["LightXMLParser"]),
+        .testTarget(
+            name: "XMLTVTests",
+            dependencies: ["XMLTV"]),
+    ]
+)

Package.swift

@@ -0,0 +1,34 @@
+# XMLTV
+
+XMLTV is a Swift package which allows you to parse XML files using the EPG XMLTV structure.
+
+## Usage
+Add XMLTV to your project using Swift Package Manager.
+
+XMLTV provides two structs representing channels and programs.
+
+```swift
+public struct TVChannel {
+    public let id: String
+    public let name: String?
+    public let url: String?
+    public let icon: String?
+}
+
+public struct TVProgram {
+    public let start: Date?
+    public let stop: Date?
+    public let channel: TVChannel
+    public let title: String?
+    public let description: String?
+    public let credits: [String:String]
+    public let date: String?
+    public let categories: [String]
+    public let country: String?
+    public let episode: String?
+    public let icon: String?
+    public let rating: String?
+}
+```
+
+To parse files, you need to create an XMLTV object using the file data. Then you can use `getChannels() -> [TVChannel]` and `getPrograms(channel:) -> [TVProgram]` to get the information.

README.md

@@ -0,0 +1,10 @@
+import Foundation
+
+extension Date {
+    static func parse(tvDate: String?) -> Date? {
+        guard let tvDate = tvDate else { return nil }
+        let dateFormatter = DateFormatter()
+        dateFormatter.dateFormat = "yyyyMMddHHmmss Z"
+        return dateFormatter.date(from: tvDate)
+    }
+}

Sources/XMLTV/Date+XMLTV.swift

@@ -0,0 +1,8 @@
+import Foundation
+
+public struct TVChannel {
+    public let id: String
+    public let name: String?
+    public let url: String?
+    public let icon: String?
+}

Sources/XMLTV/TVChannel.swift

@@ -0,0 +1,17 @@
+import Foundation
+
+public struct TVProgram {
+    public let start: Date?
+    public let stop: Date?
+    public let channel: TVChannel
+    public let title: String?
+    public let description: String?
+    public let credits: [String:String]
+    public let date: String?
+    public let categories: [String]
+    public let country: String?
+    public let episode: String?
+    public let icon: String?
+    public let rating: String?
+}
+

Sources/XMLTV/TVProgram.swift

@@ -0,0 +1,60 @@
+import Foundation
+import LightXMLParser
+
+private extension XML {
+    func children(name: String) -> [XML] {
+        return children.filter { $0.name == name }
+    }
+}
+
+public class XMLTV {
+    
+    private let xml: XML
+    
+    public init(data: Data) {
+        self.xml = XML(data: data)
+    }
+    
+    public func getChannels() -> [TVChannel] {
+        var channels: [TVChannel] = []
+        
+        let xmlChannels = xml.children(name: "channel")
+        xmlChannels.forEach { channel in
+            if let id = channel.attributes["id"] {
+                let name = channel.children(name: "display-name").first?.value
+                let url = channel.children(name: "url").first?.value
+                let iconSrc = channel.children(name: "icon").first?.attributes["src"]
+                channels.append(TVChannel(id: id, name: name, url: url, icon: iconSrc))
+            }
+        }
+        
+        return channels
+    }
+    
+    public func getPrograms(channel: TVChannel) -> [TVProgram] {
+        var programs: [TVProgram] = []
+        
+        let xmlPrograms = xml.children.filter { $0.name == "programme" && $0.attributes["channel"] == channel.id }
+        xmlPrograms.forEach { program in
+            let startDate = Date.parse(tvDate: program.attributes["start"])
+            let stopDate = Date.parse(tvDate: program.attributes["stop"])
+            let title = program.children(name: "title").first?.value
+            let description = program.children(name: "desc").first?.value
+            let icon = program.children(name: "icon").first?.attributes["src"]
+            let rating = program.children(name: "star-rating").first?.children(name: "value").first?.value
+            let date = program.children(name: "date").first?.value
+            let episode = program.children(name: "episode-num").first?.value
+            let categories = program.children(name: "category").map { $0.value }
+            let country = program.children(name: "country").first?.value
+            let credits = program.children(name: "credits").first?.children.reduce([String:String]()) { dict, credit -> [String:String] in
+                var dict = dict
+                dict[credit.name] = credit.value
+                return dict
+                } ?? [:]
+            programs.append(TVProgram(start: startDate, stop: stopDate, channel: channel, title: title, description: description, credits: credits, date: date, categories: categories, country: country, episode: episode, icon: icon, rating: rating))
+        }
+        
+        return programs
+    }
+    
+}

Sources/XMLTV/XMLTV.swift

@@ -0,0 +1,7 @@
+import XCTest
+
+import XMLTVTests
+
+var tests = [XCTestCaseEntry]()
+tests += XMLTVTests.allTests()
+XCTMain(tests)

Tests/LinuxMain.swift

@@ -0,0 +1,9 @@
+import XCTest
+
+#if !canImport(ObjectiveC)
+public func allTests() -> [XCTestCaseEntry] {
+    return [
+        testCase(XMLTVTests.allTests),
+    ]
+}
+#endif