1 /*
2  * Copyright (c) 2022 Huawei Device Co., Ltd.
3  * Licensed under the Apache License, Version 2.0 (the "License");
4  * you may not use this file except in compliance with the License.
5  * You may obtain a copy of the License at
6  *
7  *     http://www.apache.org/licenses/LICENSE-2.0
8  *
9  * Unless required by applicable law or agreed to in writing, software
10  * distributed under the License is distributed on an "AS IS" BASIS,
11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12  * See the License for the specific language governing permissions and
13  * limitations under the License.
14  */
15 
16 #ifndef PRINT_MANAGER_CLIENT_H
17 #define PRINT_MANAGER_CLIENT_H
18 
19 #include <condition_variable>
20 #include <map>
21 #include <mutex>
22 
23 #include "want.h"
24 #include "iprint_callback.h"
25 #include "iprint_service.h"
26 #include "iremote_object.h"
27 #include "print_extension_callback_stub.h"
28 #include "print_extension_info.h"
29 #include "print_job.h"
30 #include "print_sa_death_recipient.h"
31 #include "printer_info.h"
32 #include "refbase.h"
33 #ifdef PDFIUM_ENABLE
34 #include "fpdfview.h"
35 #endif // PDFIUM_ENABLE
36 
37 namespace OHOS::Print {
38 class PrintManagerClient : public RefBase {
39 public:
40     PrintManagerClient();
41     ~PrintManagerClient();
42     static PrintManagerClient* GetInstance();
43 
44     void OnRemoteSaDied(const wptr<IRemoteObject> &object);
45 
46     int32_t Init();
47     int32_t Release();
48     int32_t StartPrint(const std::vector<std::string> &fileList,
49         const std::vector<uint32_t> &fdList, std::string &taskId);
50     int32_t StopPrint(const std::string &taskId);
51     int32_t QueryAllExtension(std::vector<PrintExtensionInfo> &extensionInfos);
52     int32_t StartDiscoverPrinter(const std::vector<std::string> &extensionList);
53     int32_t StopDiscoverPrinter();
54     int32_t AddPrinters(const std::vector<PrinterInfo> &printerInfos);
55     int32_t RemovePrinters(const std::vector<std::string> &printerIds);
56     int32_t UpdatePrinters(const std::vector<PrinterInfo> &printerInfos);
57     int32_t ConnectPrinter(const std::string &printerId);
58     int32_t DisconnectPrinter(const std::string &printerId);
59     int32_t StartPrintJob(PrintJob &jobinfo);
60     int32_t CancelPrintJob(const std::string &jobId);
61     int32_t UpdatePrinterState(const std::string &printerId, uint32_t state);
62     int32_t UpdatePrintJobStateForNormalApp(const std::string &jobId, uint32_t state, uint32_t subState);
63     int32_t UpdatePrintJobStateOnlyForSystemApp(const std::string &jobId, uint32_t state, uint32_t subState);
64     int32_t UpdateExtensionInfo(const std::string &extensionId);
65     int32_t RequestPreview(const PrintJob &jobinfo, std::string &previewResult);
66     int32_t QueryPrinterCapability(const std::string &printerId);
67     int32_t QueryAllPrintJob(std::vector<PrintJob> &printJobs);
68     int32_t QueryPrintJobById(std::string &printJobId, PrintJob &printjob);
69     int32_t AddPrinterToCups(const std::string &printerUri, const std::string &printerName,
70         const std::string &printerMake);
71     int32_t QueryPrinterCapabilityByUri(const std::string &printerUri, const std::string &printerId,
72         PrinterCapability &printerCaps);
73     int32_t NotifyPrintServiceEvent(std::string &jobId, uint32_t event);
74     int32_t SetDefaultPrinter(const std::string &printerId, uint32_t type);
75     int32_t DeletePrinterFromCups(const std::string &printerName);
76     int32_t QueryPrinterInfoByPrinterId(const std::string &printerId, PrinterInfo &info);
77     int32_t QueryAddedPrinter(std::vector<std::string> &printerNameList);
78     int32_t QueryPrinterProperties(const std::string &printerId, const std::vector<std::string> &keyList,
79         std::vector<std::string> &valueList);
80     int32_t StartNativePrintJob(PrintJob &printJob);
81     int32_t GetPrinterPreference(const std::string &printerId, std::string &printerPreference);
82     int32_t SetPrinterPreference(const std::string &printerId, const std::string &printerPreference);
83     int32_t DiscoverUsbPrinters(std::vector<PrinterInfo> &printers);
84     int32_t AddPrinterToDiscovery(const PrinterInfo &printerInfo);
85     int32_t UpdatePrinterInDiscovery(const PrinterInfo &printerInfo);
86     int32_t RemovePrinterFromDiscovery(const std::string &printerId);
87     int32_t UpdatePrinterInSystem(const PrinterInfo &printerInfo);
88 
89     int32_t On(const std::string &taskId, const std::string &type, const sptr<IPrintCallback> &listener);
90     int32_t Off(const std::string &taskId, const std::string &type);
91     int32_t Print(const std::string &printJobName, const sptr<IPrintCallback> &listener,
92         const std::shared_ptr<PrintAttributes> &printAttributes);
93     int32_t Print(const std::string &printJobName, const sptr<IPrintCallback> &listener,
94         const std::shared_ptr<PrintAttributes> &printAttributes, void* contextToken);
95     int32_t Print(const std::string &printJobName, const sptr<IPrintCallback> &listener,
96         const PrintAttributes &printAttributes);
97     int32_t Print(const std::string &printJobName, const sptr<IPrintCallback> &listener,
98         const PrintAttributes &printAttributes, void* contextToken);
99     int32_t Print(const std::string &printJobName, const sptr<IPrintCallback> &listener,
100         const PrintAttributes &printAttributes, std::string &taskId);
101     int32_t Print(const std::string &printJobName, const sptr<IPrintCallback> &listener,
102         const PrintAttributes &printAttributes, std::string &taskId, void* contextToken);
103     int32_t StartGetPrintFile(const std::string &jobId, const PrintAttributes &printAttributes,
104         const uint32_t fd);
105     int32_t NotifyPrintService(const std::string &jobId, const std::string &type);
106 
107     int32_t RegisterExtCallback(const std::string &extensionId, uint32_t callbackId, PrintExtCallback cb);
108     int32_t RegisterExtCallback(const std::string &extensionId, uint32_t callbackId, PrintJobCallback cb);
109     int32_t RegisterExtCallback(const std::string &extensionId, uint32_t callbackId, PrinterCallback cb);
110     int32_t RegisterExtCallback(const std::string &extensionId, uint32_t callbackId, PrinterCapabilityCallback cb);
111     int32_t UnregisterAllExtCallback(const std::string &extensionId);
112     int32_t LoadExtSuccess(const std::string &extensionId);
113 
114     int32_t SetNativePrinterChangeCallback(const std::string &type, NativePrinterChangeCallback cb);
115 
116     void LoadServerSuccess();
117     void LoadServerFail();
118     void SetProxy(const sptr<IRemoteObject> &obj);
119     void ResetProxy();
120 
121 private:
122     void SetWantParam(AAFwk::Want &want, std::string &taskId);
123     bool LoadServer();
124     bool GetPrintServiceProxy();
125     int32_t runBase(const char* callerFunName, std::function<int32_t(sptr<IPrintService>)> func);
126 #define CALL_COMMON_CLIENT(func) runBase(__func__, func)
127 
128 private:
129     std::recursive_mutex proxyLock_;
130     sptr<IPrintService> printServiceProxy_;
131     sptr<PrintSaDeathRecipient> deathRecipient_;
132 
133     std::map<std::string, sptr<PrintExtensionCallbackStub>> extCallbackMap_;
134 
135     std::mutex loadMutex_;
136     std::mutex conditionMutex_;
137     std::condition_variable syncCon_;
138     bool ready_ = false;
139     static constexpr int LOAD_SA_TIMEOUT_MS = 15000;
140 };
141 }  // namespace OHOS::Print
142 #endif  // PRINT_MANAGER_CLIENT_H
143