#pragma once

#include <audioapi/core/AudioNode.h>
#include <audioapi/core/BaseAudioContext.h>
#include <audioapi/core/utils/worklets/WorkletsRunner.h>
#include <audioapi/utils/AudioArray.hpp>
#include <audioapi/utils/AudioArrayBuffer.hpp>
#include <audioapi/utils/AudioBuffer.hpp>
#include <jsi/jsi.h>

#include <memory>
#include <vector>

namespace audioapi {

#if RN_AUDIO_API_TEST
class WorkletProcessingNode : public AudioNode {
 public:
  explicit WorkletProcessingNode(
      std::shared_ptr<BaseAudioContext> context,
      WorkletsRunner &&workletRunner)
      : AudioNode(context) {}

 protected:
  std::shared_ptr<DSPAudioBuffer> processNode(
      const std::shared_ptr<DSPAudioBuffer> &processingBuffer,
      int framesToProcess) override {
    return processingBuffer;
  }
};
#else

using namespace facebook;

class WorkletProcessingNode : public AudioNode {
 public:
  explicit WorkletProcessingNode(
      const std::shared_ptr<BaseAudioContext> &context,
      WorkletsRunner &&workletRunner);

 protected:
  std::shared_ptr<DSPAudioBuffer> processNode(
      const std::shared_ptr<DSPAudioBuffer> &processingBuffer,
      int framesToProcess) override;

 private:
  WorkletsRunner workletRunner_;
  std::vector<std::shared_ptr<AudioArrayBuffer>> inputBuffsHandles_;
  std::vector<std::shared_ptr<AudioArrayBuffer>> outputBuffsHandles_;
};

#endif // RN_AUDIO_API_TEST

} // namespace audioapi