xref: /haiku/src/kits/print/PrintTransport.cpp (revision b671e9bbdbd10268a042b4f4cc4317ccd03d105e)
1 /*
2 
3 PrintTransport
4 
5 Copyright (c) 2004 OpenBeOS.
6 
7 Authors:
8 	Philippe Houdoin
9 	Michael Pfeiffer
10 
11 Permission is hereby granted, free of charge, to any person obtaining a copy of
12 this software and associated documentation files (the "Software"), to deal in
13 the Software without restriction, including without limitation the rights to
14 use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
15 of the Software, and to permit persons to whom the Software is furnished to do
16 so, subject to the following conditions:
17 
18 The above copyright notice and this permission notice shall be included in all
19 copies or substantial portions of the Software.
20 
21 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
22 IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
23 FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
24 AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
25 LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
26 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
27 THE SOFTWARE.
28 
29 */
30 
31 #include "PrintTransport.h"
32 
33 #include <Directory.h>
34 #include <File.h>
35 #include <FindDirectory.h>
36 #include <Path.h>
37 #include <String.h>
38 
39 // implementation of class PrintTransport
40 PrintTransport::PrintTransport()
41 	: fDataIO(NULL)
42 	, fAddOnID(-1)
43 	, fExitProc(NULL)
44 {
45 }
46 
47 PrintTransport::~PrintTransport()
48 {
49 	if (fExitProc) {
50 		(*fExitProc)();
51 		fExitProc = NULL;
52 	}
53 
54 	if (fAddOnID >= 0) {
55 		unload_add_on(fAddOnID);
56 		fAddOnID = -1;
57 	}
58 }
59 
60 status_t PrintTransport::Open(BNode* printerFolder)
61 {
62 	// already opened?
63 	if (fDataIO != NULL) {
64 		return B_ERROR;
65 	}
66 
67 	// retrieve transport add-on name from printer folder attribute
68 	BString transportName;
69 	if (printerFolder->ReadAttrString("transport", &transportName) != B_OK) {
70 		return B_ERROR;
71 	}
72 
73 	// try first in user add-ons directory
74 	BPath path;
75 	find_directory(B_USER_ADDONS_DIRECTORY, &path);
76 	path.Append("Print/transport");
77 	path.Append(transportName.String());
78 	fAddOnID = load_add_on(path.Path());
79 
80 	if (fAddOnID < 0) {
81 		// on failure try in system add-ons directory
82 		find_directory(B_BEOS_ADDONS_DIRECTORY, &path);
83 		path.Append("Print/transport");
84 		path.Append(transportName.String());
85 		fAddOnID = load_add_on(path.Path());
86 	}
87 
88 	if (fAddOnID < 0) {
89 		// failed to load transport add-on
90 		return B_ERROR;
91 	}
92 
93 	// get init & exit proc
94 	BDataIO* (*initProc)(BMessage*);
95 	get_image_symbol(fAddOnID, "init_transport", B_SYMBOL_TYPE_TEXT, (void **) &initProc);
96 	get_image_symbol(fAddOnID, "exit_transport", B_SYMBOL_TYPE_TEXT, (void **) &fExitProc);
97 
98 	if (initProc == NULL || fExitProc == NULL) {
99 		// transport add-on has not the proper interface
100 		return B_ERROR;
101 	}
102 
103 	// now, initialize the transport add-on
104 	node_ref   ref;
105 	BDirectory dir;
106 
107 	printerFolder->GetNodeRef(&ref);
108 	dir.SetTo(&ref);
109 
110 	if (path.SetTo(&dir, NULL) != B_OK) {
111 		return B_ERROR;
112 	}
113 
114 	// request BDataIO object from transport add-on
115 	BMessage input('TRIN');
116 	input.AddString("printer_file", path.Path());
117 	fDataIO = (*initProc)(&input);
118 	return B_OK;
119 }
120 
121 BDataIO*
122 PrintTransport::GetDataIO()
123 {
124 	return fDataIO;
125 }
126 
127 bool PrintTransport::IsPrintToFileCanceled() const
128 {
129 	// The BeOS "Print To File" transport add-on returns a non-NULL BDataIO *
130 	// even after user filepanel cancellation!
131 	BFile* file = dynamic_cast<BFile*>(fDataIO);
132 	return fDataIO == NULL || file != NULL && file->InitCheck() != B_OK;
133 }
134