//
//  CSVUtils.swift
//  TournamentStats
//
//  Created by Laurent Morvillier on 03/06/2019.
//  Copyright © 2019 Stax River. All rights reserved.
//

import Foundation

extension Array where Element : CSVRepresentable {
    
    func writeCSV(file: String, limit: Int? = 10) {
        
        var string = Element.csvHeaders()
        let max = limit ?? Int.max
        for (index, rep) in self.enumerated() {
            string.append("/n")
            string.append(rep.csv())
            
            if index >= max {
                break
            }
        }
        
        do {
            try FileWriter.writeToDocumentDirectory(content: string, fileName: file)
        } catch {
            print(error)
        }
        
    }
    
}

extension Array where Element : HTMLRepresentable {
    
    func writeHTML(fileName: String, limit: Int? = 10) {
        
        var html = "<table class=\"wp-block-table alignwide\">\n"
        html.append("<thead>")
        html.append(Element.htmlHeaders())
        html.append("</thead>\n<tbody>")
        let max = limit ?? Int.max
        for (index, rep) in self.enumerated() {
            html.append("\n")
            html.append(rep.html())
            
            if index + 1 >= max {
                break
            }
        }
        html.append("\n</tbody></table>")
        
        do {
            try FileWriter.writeToDocumentDirectory(content: html, fileName: fileName)
        } catch {
            print(error)
        }
        
    }
    
}

public protocol CSVRepresentable {
    static func csvHeaders() -> String
    func csv() -> String
}

public protocol HTMLRepresentable {
    static func htmlHeaders() -> String
    func html() -> String
}