#pragma once

#include "audioeffectx.h"
#include "VSTEditor.hpp"
#include <memory>
#include <optional>

#include "amuse/BooBackend.hpp"
#include "amuse/Engine.hpp"
#include "amuse/IBackendVoice.hpp"
#include "amuse/IBackendSubmix.hpp"
#include "amuse/IBackendVoiceAllocator.hpp"
#include "AudioGroupFilePresenter.hpp"

namespace amuse {
class VSTBackend;

/** Backend voice allocator implementation for AudioUnit mixer */
class VSTBackendVoiceAllocator : public BooBackendVoiceAllocator {
public:
  VSTBackendVoiceAllocator(boo::IAudioVoiceEngine& booEngine) : BooBackendVoiceAllocator(booEngine) {}
};

/** Actual plugin implementation class */
class VSTBackend : public AudioEffectX {
  std::mutex m_lock;
  std::unique_ptr<boo::IAudioVoiceEngine> m_booBackend;
  std::optional<amuse::VSTBackendVoiceAllocator> m_voxAlloc;
  std::optional<amuse::Engine> m_engine;
  std::shared_ptr<amuse::Sequencer> m_curSeq;
  int m_reqGroup = -1;
  int m_curGroup = -1;
  const AudioGroupDataCollection* m_curData = nullptr;
  size_t m_curFrame = 0;
  std::wstring m_userDir;
  int m_routeChannel = -1;
  AudioGroupFilePresenter m_filePresenter;
  VSTEditor m_editor;

public:
  VSTBackend(audioMasterCallback cb);
  ~VSTBackend();
  AEffEditor* getEditor();
  VstInt32 processEvents(VstEvents* events);
  void processReplacing(float** inputs, float** outputs, VstInt32 sampleFrames);
  VstInt32 canDo(char* text);
  VstPlugCategory getPlugCategory();
  bool getEffectName(char* text);
  bool getProductString(char* text);
  bool getVendorString(char* text);
  bool getProgramNameIndexed(VstInt32 category, VstInt32 index, char* text);
  bool getOutputProperties(VstInt32 index, VstPinProperties* properties);
  VstInt32 getNumMidiInputChannels();
  void setSampleRate(float sampleRate);
  void setBlockSize(VstInt32 blockSize);

  amuse::Engine& getAmuseEngine() { return *m_engine; }
  std::wstring_view getUserDir() const { return m_userDir; }
  AudioGroupFilePresenter& getFilePresenter() { return m_filePresenter; }

  void loadGroupFile(int collectionIdx, int fileIdx);
  void setGroup(int groupIdx, bool immediate);
  void _setNormalProgram(int programNo);
  void setNormalProgram(int programNo);
  void _setDrumProgram(int programNo);
  void setDrumProgram(int programNo);

  VstInt32 getChunk(void** data, bool isPreset);
  VstInt32 setChunk(void* data, VstInt32 byteSize, bool isPreset);
};
} // namespace amuse