// Copyright 2026 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <vector>

#include "src/utils/span.h"

namespace dawn::internal {

class RawStream {
 public:
  // Expected rewrite:
  // virtual void fct(dawn::Span<char> param) = 0;
  virtual void fct(dawn::Span<char> param) = 0;
};

class RawBufferedStream : public RawStream {
 public:
  // Expected rewrite:
  // void fct(dawn::Span<char> param) override
  void fct(dawn::Span<char> param) override {
    // Expected rewrite:
    // param[0] = 'a';
    param[0] = 'a';

    // This leads param to be rewritten.
    param[1] = 'b';
  }
};

class SimpleBufferedStream : public RawStream {
 public:
  // Expected rewrite:
  // void fct(dawn::Span<char> param) override
  void fct(dawn::Span<char> param) override {}
};

// Expected rewrite:
// dawn::Span<char> get(int index = 0)
dawn::Span<char> get(int index = 0) {
  // Expected rewrite:
  // return {};
  return {};
}

void fct2() {
  std::vector<char> buf;
  RawBufferedStream stream;

  // Expected rewrite:
  // stream.fct(buf);
  stream.fct(buf);

  // Expected rewrite:
  // dawn::Span<char> ptr = get();
  dawn::Span<char> ptr = get();
  // Buffer expression leading ptr and get return type to be rewritten.
  ptr[3] = 'c';

  // Expected rewrite:
  // get()[0] = 'a'
  get()[0] = 'a';

  int index = 0;
  // Expected rewrite:
  // get(index)[0] = 'x';",
  get(index)[0] = 'x';
}

}  // namespace dawn::internal