src/kits/print/PrintTransport.cpp

   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         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
 123 BDataIO*
 124 PrintTransport::GetDataIO()
 125 {
 126         return fDataIO;
 127 }
 128
 129 bool 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 }