#pragma once

#include "StarFormattedJson.hpp"
#include "StarJsonPath.hpp"

namespace Star {

struct GetCommand {
  JsonPath::PathPtr path;
  bool opt;
  bool children;
};

struct SetCommand {
  JsonPath::PathPtr path;
  FormattedJson value;
};

struct AddCommand {
  JsonPath::PathPtr path;
  FormattedJson value;
};

struct RemoveCommand {
  JsonPath::PathPtr path;
};

struct EditCommand {
  JsonPath::PathPtr path;
};

typedef MVariant<GetCommand, SetCommand, AddCommand, RemoveCommand, EditCommand> Command;

struct AtBeginning {};

struct AtEnd {};

struct BeforeKey {
  String key;
};

struct AfterKey {
  String key;
};

typedef MVariant<AtBeginning, AtEnd, BeforeKey, AfterKey> InsertLocation;

STAR_CLASS(JsonInputFormat);

class JsonInputFormat {
public:
  virtual ~JsonInputFormat() {}
  virtual FormattedJson toJson(String const& input) const = 0;
  virtual String fromJson(FormattedJson const& json) const = 0;
  virtual FormattedJson getDefault() const = 0;
};

class GenericInputFormat : public JsonInputFormat {
public:
  virtual FormattedJson toJson(String const& input) const override;
  virtual String fromJson(FormattedJson const& json) const override;
  virtual FormattedJson getDefault() const override;
};

class CommaSeparatedStrings : public JsonInputFormat {
public:
  virtual FormattedJson toJson(String const& input) const override;
  virtual String fromJson(FormattedJson const& json) const override;
  virtual FormattedJson getDefault() const override;
};

class StringInputFormat : public JsonInputFormat {
public:
  virtual FormattedJson toJson(String const& input) const override;
  virtual String fromJson(FormattedJson const& json) const override;
  virtual FormattedJson getDefault() const override;
};

STAR_CLASS(Output);

class Output {
public:
  virtual ~Output() {}
  virtual void out(FormattedJson const& json) = 0;
  virtual void flush() = 0;

  function<void(FormattedJson const& json)> toFunction() {
    return [this](FormattedJson const& json) { this->out(json); };
  }
};

class OutputOnSeparateLines : public Output {
public:
  virtual void out(FormattedJson const& json) override;
  virtual void flush() override;
};

class ArrayOutput : public Output {
public:
  ArrayOutput(bool unique) : m_unique(unique), m_results() {}

  virtual void out(FormattedJson const& json) override;
  virtual void flush() override;

private:
  bool m_unique;
  List<FormattedJson> m_results;
};

struct Options {
  Options() : inPlace(false), insertLocation(), editFormat(nullptr), editorImages(), output() {}

  bool inPlace;
  InsertLocation insertLocation;
  JsonInputFormatPtr editFormat;
  List<JsonPath::PathPtr> editorImages;
  OutputPtr output;
};

struct JsonLiteralInput {
  String json;
};

struct FileInput {
  String filename;
};

struct FindInput {
  String directory;
  String filenameSuffix;
};

typedef MVariant<JsonLiteralInput, FileInput, FindInput> Input;

struct ParsedArgs {
  ParsedArgs() : inputs(), command(), options() {}

  List<Input> inputs;
  Command command;
  Options options;
};

FormattedJson addOrSet(bool add, JsonPath::PathPtr path, FormattedJson const& input, InsertLocation insertLocation, FormattedJson const& value);
String reprWithLineEnding(FormattedJson const& json);
StringList findFiles(FindInput const& findArgs);

}