1/*
2
3PrintTransport
4
5Copyright (c) 2004 Haiku.
6
7Authors:
8	Philippe Houdoin
9	Michael Pfeiffer
10
11Permission is hereby granted, free of charge, to any person obtaining a copy of
12this software and associated documentation files (the "Software"), to deal in
13the Software without restriction, including without limitation the rights to
14use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
15of the Software, and to permit persons to whom the Software is furnished to do
16so, subject to the following conditions:
17
18The above copyright notice and this permission notice shall be included in all
19copies or substantial portions of the Software.
20
21THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
22IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
23FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
24AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
25LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
26OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
27THE 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
40PrintTransport::PrintTransport()
41	: fDataIO(NULL)
42	, fAddOnID(-1)
43	, fExitProc(NULL)
44{
45}
46
47PrintTransport::~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
60status_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	const directory_which paths[] = {
74		B_USER_NONPACKAGED_ADDONS_DIRECTORY,
75		B_USER_ADDONS_DIRECTORY,
76		B_SYSTEM_NONPACKAGED_ADDONS_DIRECTORY,
77		B_SYSTEM_ADDONS_DIRECTORY,
78	};
79	BPath path;
80	for (uint32 i = 0; i < sizeof(paths) / sizeof(paths[0]); ++i) {
81		if (find_directory(paths[i], &path) != B_OK)
82			continue;
83	path.Append("Print/transport");
84	path.Append(transportName.String());
85	fAddOnID = load_add_on(path.Path());
86		if (fAddOnID >= 0)
87			break;
88	}
89
90	if (fAddOnID < 0) {
91		// failed to load transport add-on
92		return B_ERROR;
93	}
94
95	// get init & exit proc
96	BDataIO* (*initProc)(BMessage*);
97	get_image_symbol(fAddOnID, "init_transport", B_SYMBOL_TYPE_TEXT, (void **) &initProc);
98	get_image_symbol(fAddOnID, "exit_transport", B_SYMBOL_TYPE_TEXT, (void **) &fExitProc);
99
100	if (initProc == NULL || fExitProc == NULL) {
101		// transport add-on has not the proper interface
102		return B_ERROR;
103	}
104
105	// now, initialize the transport add-on
106	node_ref   ref;
107	BDirectory dir;
108
109	printerFolder->GetNodeRef(&ref);
110	dir.SetTo(&ref);
111
112	if (path.SetTo(&dir, NULL) != B_OK) {
113		return B_ERROR;
114	}
115
116	// request BDataIO object from transport add-on
117	BMessage input('TRIN');
118	input.AddString("printer_file", path.Path());
119	fDataIO = (*initProc)(&input);
120	return B_OK;
121}
122
123BDataIO*
124PrintTransport::GetDataIO()
125{
126	return fDataIO;
127}
128
129bool PrintTransport::IsPrintToFileCanceled() const
130{
131	// The BeOS "Print To File" transport add-on returns a non-NULL BDataIO *
132	// even after user filepanel cancellation!
133	BFile* file = dynamic_cast<BFile*>(fDataIO);
134	return fDataIO == NULL || (file != NULL && file->InitCheck() != B_OK);
135}
136