let cellIdentifier = "CellIdentifier"

var sections: [DemoSection]!

// MARK: LifeCycle Methods

override func viewDidLoad() {

super.viewDidLoad()

sections = [

DemoSection(title: "Plain HTML Editor", rows: [

DemoRow(title: "Standard Demo", action: { self.showEditorDemo(filename: "content") }),

DemoRow(title: "Captions Demo", action: { self.showEditorDemo(filename: "captions") }),

DemoRow(title: "Image Overlays", action: { self.showEditorDemo(filename: "imagesOverlays") }),

DemoRow(title: "Video Demo", action: { self.showEditorDemo(filename: "video", wordPressMode: false) }),

DemoRow(title: "Failed Media", action: { self.showEditorDemo(filename: "failedMedia") }),

DemoRow(title: "Big Lists", action: { self.showEditorDemo(filename: "bigLists") }),

DemoRow(title: "Underline NBSP", action: { self.showEditorDemo(filename: "underline")}),

DemoRow(title: "Empty Demo", action: { self.showEditorDemo() })

]

),

DemoSection(title: "WordPressEditor (Calypso & Gutenberg)", rows: [

DemoRow(title: "Gutenberg Demo", action: { self.showEditorDemo(filename: "gutenberg", wordPressMode: true) }),

DemoRow(title: "Video Shortcode Demo", action: { self.showEditorDemo(filename: "videoShortcodes", wordPressMode: true) }),

DemoRow(title: "Gutenberg Gallery", action: { self.showEditorDemo(filename: "gallery", wordPressMode: true) }),

DemoRow(title: "Empty Demo", action: { self.showEditorDemo(wordPressMode: true) })

]

),

]

}

// MARK: Actions

func showEditorDemo(filename: String? = nil, wordPressMode: Bool = false) {

let controller: EditorDemoController

if let filename = filename {

let sampleHTML = getSampleHTML(fromHTMLFileNamed: filename)

controller = EditorDemoController(withSampleHTML: sampleHTML, wordPressMode: wordPressMode)

} else {

controller = EditorDemoController(wordPressMode: wordPressMode)

}

navigationController?.pushViewController(controller, animated: true)

}

// MARK: Sample HTML

func getSampleHTML(fromHTMLFileNamed fileName: String) -> String {

let htmlFilePath = Bundle.main.path(forResource: fileName, ofType: "html")!

let fileContents: String

do {

fileContents = try String(contentsOfFile: htmlFilePath)

} catch {

fatalError("Could not load the sample HTML. Check the file exists in the target and that it has the correct name.")

}

return fileContents

}

// MARK: TableView Methods

override func numberOfSections(in tableView: UITableView) -> Int {

return sections.count

}

override func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int {

return sections[section].rows.count

}

override func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {

let cell = tableView.dequeueReusableCell(withIdentifier: cellIdentifier)!

cell.accessoryType = .disclosureIndicator

//cell.backgroundColor = UIColor.

let row = sections[indexPath.section].rows[indexPath.row]

cell.textLabel?.text = row.title

return cell

}

override func tableView(_ tableView: UITableView, heightForHeaderInSection section: Int) -> CGFloat {

return 40

}

override func tableView(_ tableView: UITableView, titleForHeaderInSection section: Int) -> String? {

return sections[section].title

}

override func tableView(_ tableView: UITableView, didSelectRowAt indexPath: IndexPath) {

tableView.deselectRow(at: indexPath, animated: true)

sections[indexPath.section].rows[indexPath.row].action()

}