chromium/chrome/browser/chromeos/policy/dlp/dlp_files_controller_lacros.h

// Copyright 2023 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_CHROMEOS_POLICY_DLP_DLP_FILES_CONTROLLER_LACROS_H_
#define CHROME_BROWSER_CHROMEOS_POLICY_DLP_DLP_FILES_CONTROLLER_LACROS_H_

#include <optional>

#include "base/files/file_path.h"
#include "base/gtest_prod_util.h"
#include "chrome/browser/chromeos/policy/dlp/dlp_files_controller.h"
#include "chrome/browser/chromeos/policy/dlp/dlp_rules_manager.h"

namespace policy {
class DlpFilesControllerLacros : public DlpFilesController {
 public:
  explicit DlpFilesControllerLacros(const DlpRulesManager& rules_manager);

  ~DlpFilesControllerLacros() override;

 protected:
  // DlpFilesController overrides:
  std::optional<data_controls::Component> MapFilePathToPolicyComponent(
      Profile* profile,
      const base::FilePath& file_path) override;
  bool IsInLocalFileSystem(const base::FilePath& file_path) override;
  void ShowDlpBlockedFiles(std::optional<uint64_t> task_id,
                           std::vector<base::FilePath> blocked_files,
                           dlp::FileAction action) override;

  // TODO(b/284122497): Cleanup friend for testing
  FRIEND_TEST_ALL_PREFIXES(DlpFilesControllerLacrosTest,
                           MapFilePathToPolicyComponentTest);

  FRIEND_TEST_ALL_PREFIXES(DlpFilesControllerLacrosBlockUITest,
                           ShowDlpBlockedFilesTest);
};
}  // namespace policy

#endif  // CHROME_BROWSER_CHROMEOS_POLICY_DLP_DLP_FILES_CONTROLLER_LACROS_H_