chrome/browser/ui/webui/ash/slow_trace_ui.h

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

#ifndef CHROME_BROWSER_UI_WEBUI_ASH_SLOW_TRACE_UI_H_
#define CHROME_BROWSER_UI_WEBUI_ASH_SLOW_TRACE_UI_H_

#include <string>

#include "chrome/common/webui_url_constants.h"
#include "content/public/browser/url_data_source.h"
#include "content/public/browser/web_ui_controller.h"
#include "content/public/browser/webui_config.h"
#include "content/public/common/url_constants.h"

namespace base {
class RefCountedString;
}

namespace ash {

// This class provides the source for chrome://slow_trace/.  It needs to be a
// separate handler than chrome://slow, because URLDataSource and
// WebUIDataSource are not descended from each other, and WebUIDataSource
// doesn't allow the MimeType to be dynamically specified.
class SlowTraceSource : public content::URLDataSource {
 public:
  SlowTraceSource();

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

  ~SlowTraceSource() override;

  // content::URLDataSource implementation.
  std::string GetSource() override;
  void StartDataRequest(
      const GURL& url,
      const content::WebContents::Getter& wc_getter,
      content::URLDataSource::GotDataCallback callback) override;
  std::string GetMimeType(const GURL& url) override;
  bool AllowCaching() override;

 private:
  void OnGetTraceData(content::URLDataSource::GotDataCallback callback,
                      scoped_refptr<base::RefCountedString> trace_data);
};

class SlowTraceController;

// WebUIConfig for chrome://slow_trace
class SlowTraceControllerConfig
    : public content::DefaultWebUIConfig<SlowTraceController> {
 public:
  SlowTraceControllerConfig()
      : DefaultWebUIConfig(content::kChromeUIScheme,
                           chrome::kChromeUISlowTraceHost) {}
};

class SlowTraceController : public content::WebUIController {
 public:
  explicit SlowTraceController(content::WebUI* web_ui);

  SlowTraceController(const SlowTraceController&) = delete;
  SlowTraceController& operator=(const SlowTraceController&) = delete;
};

}  // namespace ash

#endif  // CHROME_BROWSER_UI_WEBUI_ASH_SLOW_TRACE_UI_H_