1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
// Copyright 2021 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_ENTERPRISE_CONNECTORS_ANALYSIS_CONTENT_ANALYSIS_DOWNLOADS_DELEGATE_H_
#define CHROME_BROWSER_ENTERPRISE_CONNECTORS_ANALYSIS_CONTENT_ANALYSIS_DOWNLOADS_DELEGATE_H_
#include "base/functional/callback.h"
#include "base/memory/raw_ptr.h"
#include "chrome/browser/enterprise/connectors/analysis/content_analysis_delegate_base.h"
#include "components/download/public/common/download_item.h"
#include "components/enterprise/common/proto/connectors.pb.h"
namespace enterprise_connectors {
// A ContentAnalysisDelegateBase implementation meant to be used to display the
// ContentAnalysisDialog for a download that triggered a block or a warning, and
// for which a custom message must be shown to the user.
class ContentAnalysisDownloadsDelegate
: public ContentAnalysisDelegateBase,
public download::DownloadItem::Observer {
public:
ContentAnalysisDownloadsDelegate(
const std::u16string& filename,
const std::u16string& custom_message,
GURL custom_learn_more_url,
bool bypass_justification_required,
base::OnceClosure open_file_callback,
base::OnceClosure discard_file_callback,
download::DownloadItem* download_item,
const ContentAnalysisResponse::Result::TriggeredRule::CustomRuleMessage&
custom_rule_message);
~ContentAnalysisDownloadsDelegate() override;
// Called when the user opts to keep the download and open it. Should not be
// called if the result was a "block" since the option shouldn't be available
// in that case.
void BypassWarnings(
std::optional<std::u16string> user_justification) override;
// Called when the user opts to delete the downloaded file and not open it.
void Cancel(bool warning) override;
std::optional<std::u16string> GetCustomMessage() const override;
std::optional<GURL> GetCustomLearnMoreUrl() const override;
std::optional<std::vector<std::pair<gfx::Range, GURL>>>
GetCustomRuleMessageRanges() const override;
bool BypassRequiresJustification() const override;
std::u16string GetBypassJustificationLabel() const override;
std::optional<std::u16string> OverrideCancelButtonText() const override;
// download::DownloadItem::Observer:
void OnDownloadDestroyed(download::DownloadItem* download) override;
private:
// Resets |open_file_callback_| and |discard_file_callback_|, ensuring actions
// can't be attempted on a file that has already been opened or discarded
// (which may be undefined).
void ResetCallbacks();
// Called when the user opts to open the downloaded file.
void Open();
// Custom message for rule.
ContentAnalysisResponse::Result::TriggeredRule::CustomRuleMessage
custom_rule_message_;
std::u16string filename_;
std::u16string custom_message_;
GURL custom_learn_more_url_;
bool bypass_justification_required_;
base::OnceClosure open_file_callback_;
base::OnceClosure discard_file_callback_;
raw_ptr<download::DownloadItem> download_item_;
};
} // namespace enterprise_connectors
#endif // CHROME_BROWSER_ENTERPRISE_CONNECTORS_ANALYSIS_CONTENT_ANALYSIS_DOWNLOADS_DELEGATE_H_
|
