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

#ifndef EXTENSIONS_RENDERER_RESOURCE_BUNDLE_SOURCE_MAP_H_
#define EXTENSIONS_RENDERER_RESOURCE_BUNDLE_SOURCE_MAP_H_

#include <map>
#include <memory>
#include <string>
#include <string_view>

#include "base/compiler_specific.h"
#include "base/memory/raw_ptr.h"
#include "base/synchronization/lock.h"
#include "base/thread_annotations.h"
#include "extensions/renderer/source_map.h"
#include "v8/include/v8-forward.h"

namespace ui {
class ResourceBundle;
}

namespace extensions {

class ResourceBundleSourceMap : public SourceMap {
 public:
  explicit ResourceBundleSourceMap(const ui::ResourceBundle* resource_bundle);

  ResourceBundleSourceMap(const ResourceBundleSourceMap&) = delete;
  ResourceBundleSourceMap& operator=(const ResourceBundleSourceMap&) = delete;

  ~ResourceBundleSourceMap() override;

  v8::Local<v8::String> GetSource(v8::Isolate* isolate,
                                  std::string_view name) const override;
  bool Contains(std::string_view name) const override;

  // `name` must outlive `this`. Preferably, the name string has static storage
  // duration.
  void RegisterSource(std::string_view name LIFETIME_CAPTURE_BY_THIS,
                      int resource_id);

 private:
  struct ResourceInfo {
    ResourceInfo();
    explicit ResourceInfo(int in_id);
    ResourceInfo(ResourceInfo&& other);
    ~ResourceInfo();

    ResourceInfo& operator=(ResourceInfo&& other);

    int id = 0;
    // Used to cache the uncompressed contents if the resource is gzipped.
    mutable std::unique_ptr<std::string> cached;
  };

  raw_ptr<const ui::ResourceBundle, DanglingUntriaged> resource_bundle_;

  mutable base::Lock lock_;
  std::map<std::string_view, ResourceInfo> resource_map_ GUARDED_BY(lock_);
};

}  // namespace extensions

#endif  // EXTENSIONS_RENDERER_RESOURCE_BUNDLE_SOURCE_MAP_H_