[rbdr/captura] / Captura / Data / OutputFormatSetting.swift

/*
 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://captura.tranquil.systems.
 */
import Foundation

enum OutputFormatSetting: Int {
  case gifOnly = 0
  case mp4Only = 1
  case all = 2

  init?(_ string: String) {
    switch string {
    case "gif":
      self = .gifOnly
    case "mp4":
      self = .mp4Only
    case "all":
      self = .all
    default:
      return nil
    }
  }

  func shouldSaveGif() -> Bool {
    return self == .gifOnly || self == .all
  }

  func shouldSaveMp4() -> Bool {
    return self == .mp4Only || self == .all
  }

  func toString() -> String {
    switch self {
    case .gifOnly:
      return "gif"
    case .mp4Only:
      return "mp4"
    case .all:
      return "all"
    }
  }
}