aboutsummaryrefslogtreecommitdiff
path: root/Map/Data/MapDocument.swift
blob: 6cc84d746516655fc32760cc62bb0c7617004a43 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
// Copyright (C) 2024 Rubén Beltrán del Río

// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.

// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License
// along with this program. If not, see https://map.tranquil.systems.
import SwiftUI
import UniformTypeIdentifiers

extension UTType {
  static var wmap: UTType {
    UTType(importedAs: "systems.tranquil.map.wmap")
  }
}

struct MapDocument: FileDocument {
  var text: String

  init(
    text: String? = nil
  ) {
    if let text = text {
      self.text = text
    } else {
      // Check for pending template content first
      if let pendingContent = UserDefaults.standard.string(forKey: "pendingTemplateContent") {
        self.text = pendingContent
      } else if let templatesData = UserDefaults.standard.data(forKey: "mapTemplates"),
        let templates = try? JSONDecoder().decode([Template].self, from: templatesData),
        let defaultTemplate = templates.first(where: { $0.isDefault })
      {
        self.text = defaultTemplate.content
      } else {
        self.text = String(localized: "map_template.default.content")
      }
    }
  }

  static var readableContentTypes: [UTType] { [.wmap] }

  var parsed: [Wmap.Entity] {
    let lexer = Wmap.Lexer(self.text)
    let parser = Wmap.Parser(lexer: lexer)
    return parser.parse()
  }

  init(configuration: ReadConfiguration) throws {
    guard let data = configuration.file.regularFileContents,
      let string = String(data: data, encoding: .utf8)
    else {
      throw CocoaError(.fileReadCorruptFile)
    }
    text = string
  }

  func fileWrapper(configuration: WriteConfiguration) throws -> FileWrapper {
    let data = text.data(using: .utf8)!
    return .init(regularFileWithContents: data)
  }

  @MainActor
  func exportAsImage(withEvolution selectedEvolution: StageType) -> NSImage? {
    let renderView = MapRenderView(
      entities: self.parsed,
      evolution: .constant(selectedEvolution))
    let renderer = ImageRenderer(content: renderView)

    return renderer.nsImage
  }
}