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
|
// Copyright 2019 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "remoting/host/file_transfer/directory_helpers.h"
#include "base/check_is_test.h"
#include "base/files/file_path.h"
#include "base/logging.h"
#include "base/path_service.h"
#include "build/build_config.h"
#include "remoting/protocol/file_transfer_helpers.h"
#if BUILDFLAG(IS_CHROMEOS)
// `nogncheck` prevents false positive 'missing dependency' errors on
// non-chromeos builds.
#include "chrome/common/chrome_paths.h" // nogncheck
#endif
namespace remoting {
namespace {
static base::FilePath* g_upload_directory_for_testing = nullptr;
protocol::FileTransferResult<base::FilePath> GetDirectory(int path_key) {
base::FilePath directory_path;
if (!base::PathService::Get(path_key, &directory_path)) {
LOG(ERROR) << "Failed to get path from base::PathService::Get";
return protocol::MakeFileTransferError(
FROM_HERE, protocol::FileTransfer_Error_Type_UNEXPECTED_ERROR);
}
return directory_path;
}
} // namespace
protocol::FileTransferResult<base::FilePath> GetFileUploadDirectory() {
if (g_upload_directory_for_testing) {
CHECK_IS_TEST();
return *g_upload_directory_for_testing;
}
#if BUILDFLAG(IS_CHROMEOS)
return GetDirectory(chrome::DIR_DEFAULT_DOWNLOADS_SAFE);
#else
return GetDirectory(base::DIR_USER_DESKTOP);
#endif
}
void SetFileUploadDirectoryForTesting(base::FilePath dir) {
if (g_upload_directory_for_testing) {
delete g_upload_directory_for_testing;
}
g_upload_directory_for_testing = new base::FilePath(dir);
}
} // namespace remoting
|
