usr/src/cmd/print/selector/print-service

   1 #!/usr/perl5/bin/perl
   2 #
   3 # CDDL HEADER START
   4 #
   5 # The contents of this file are subject to the terms of the
   6 # Common Development and Distribution License (the "License").
   7 # You may not use this file except in compliance with the License.
   8 #
   9 # You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
  10 # or http://www.opensolaris.org/os/licensing.
  11 # See the License for the specific language governing permissions
  12 # and limitations under the License.
  13 #
  14 # When distributing Covered Code, include this CDDL HEADER in each
  15 # file and include the License file at usr/src/OPENSOLARIS.LICENSE.
  16 # If applicable, add the following below this CDDL HEADER, with the
  17 # fields enclosed by brackets "[]" replaced with your own identifying
  18 # information: Portions Copyright [yyyy] [name of copyright owner]
  19 #
  20 # CDDL HEADER END
  21 #
  22 # Copyright 2009 Sun Microsystems, Inc.  All rights reserved.
  23 # Use is subject to license terms.
  24 #
  25 #
  26
  27 #
  28 # This program manages the "active" print service selection.
  29 #   If called as 'print-service', it takes one of four options.
  30 #   Options:
  31 #     [-s[et] service [-m]]     Select the "active" print service, optionally
  32 #                               migrating basic print queue configuration.
  33 #     [-q[uery]]                Display the "active" print service.
  34 #     [-e[xport] file]          Export basic print queue configuration to
  35 #                               a file.
  36 #     [-i[mport] file]          Import basic print queue configuration from
  37 #                               a file.
  38 #
  39 #   If called by any other name, it will look for a corresponding command
  40 #   under /usr/lib/{active-service}/bin/{command} and execute that program
  41 #   with the original arguments.
  42 #
  43
  44 use Getopt::Long;
  45 use File::Basename;
  46 use File::Copy;
  47 use File::Temp qw/ :POSIX /;
  48
  49 my $cmd = basename($0);
  50
  51 my $LPSTAT = '/usr/bin/lpstat';
  52 my $LPADMIN = '/usr/sbin/lpadmin';
  53 my $ENABLE = '/usr/bin/enable';
  54 my $ACCEPT = '/usr/sbin/accept';
  55 my $SVCADM = '/usr/sbin/svcadm';
  56 my $SVCPROP = '/usr/bin/svcprop';
  57 my $SVCCFG = '/usr/sbin/svccfg';
  58 my $SVC_LP_SCHEDULER = 'print/server';
  59 my $SVC_LP_LPD = 'print/rfc1179';
  60 my $SVC_LP_IPP = 'print/ipp-listener';
  61 my $SVC_LP_PPD = 'print/ppd-cache-update';
  62 my $SVC_CUPS_SCHEDULER = 'cups/scheduler';
  63 my $SVC_CUPS_LPD = 'cups/in-lpd';
  64
  65 sub fatal {
  66         ($ENV{"DESKTOP_LAUNCHED"}) &&
  67                 exec("/bin/zenity", "--error", "--text=@_");
  68         print STDERR @_;
  69         exit(1);
  70 }
  71
  72 sub usage {
  73         print STDERR <<EOF ;
  74 Usage:
  75   $cmd [-s[et] service [-m]]    Select the \"active\" print service,
  76                                         optionally migrating basic print queue
  77                                         configuration.
  78   $cmd [-q[uery]]               Display the "active" print service.
  79   $cmd [-e[xport] file] Export basic print queue configuration
  80                                         to a file.
  81   $cmd [-i[mport] file] Import basic print queue configuration
  82                                         from a file.
  83 EOF
  84         exit(1);
  85 }
  86
  87 sub svcprop {
  88         local ($fmri, $property) = @_;
  89         my $FH;
  90
  91         open($FH, "$SVCPROP -C -p $property $fmri 2>/dev/null |");
  92         $result = <$FH>;
  93         close($FH);
  94
  95         return ($result);
  96 }
  97
  98 sub svccfg {
  99         local ($fmri, $operation) = @_;
 100         my $FH;
 101
 102         open($FH, "$SVCCFG -s $fmri \"$operation\" 2>/dev/null |");
 103         $result = <$FH>;
 104         close($FH);
 105
 106         return ($result);
 107 }
 108
 109 sub svcadm {
 110         local ($operation, @fmris) = @_;
 111
 112         system("$SVCADM $operation -s @fmris");
 113 }
 114
 115
 116 sub print_service {
 117         my $service;
 118
 119         $service = svcprop("$SVC_CUPS_SCHEDULER:default", "general/active");
 120         ($service =~ /true/) && ($service = 'cups') || ($service = 'lp');
 121
 122         return ($service);
 123 }
 124
 125 sub print_command {
 126         local($command, @av) = @_;
 127         my $service = print_service();
 128
 129         if (!defined($service)) {
 130                 fatal("failed to detect active print service: $!\n");
 131         }
 132
 133         if (! -d "/usr/lib/$service/bin") {
 134                 fatal("print service: $service is not installed\n");
 135         }
 136
 137         my $executable = "/usr/lib/$service/bin/$command";
 138         # CUPS has it's own names for enable and disable
 139         ($command =~ /(en|dis)able/) && ($service eq 'cups') &&
 140                 (! -x $executable) &&
 141                 ($executable = "/usr/lib/$service/bin/$service$command");
 142
 143         if (! -x $executable) {
 144                 fatal("$command is not available from $service print service\n");
 145         }
 146
 147         exec($executable, @ARGV);
 148 }
 149
 150 sub export_print_queues {
 151         local ($path) = @_;
 152         my $service = print_service();
 153
 154         if ($service eq 'lp') {
 155                 my $FH, $DFH;
 156
 157                 open($FH, ">$path");
 158                 open($DFH, "$LPSTAT -v|");
 159                 while (<$DFH>) {
 160                         if (/device for (.+): (.+)/) {
 161                                 my $EFH;
 162
 163                                 print $FH "<Printer $1>\nDeviceURI $2\n";
 164                                 open($EFH, "$LPSTAT -p $1 -l |");
 165                                 while (<$EFH>) {
 166                                         (/Description: (.+)/) &&
 167                                                 print $FH "Info $1\n";
 168                                 }
 169                                 close($EFH);
 170                                 print $FH "</Printer>\n";
 171                         }
 172                 }
 173                 close($DFH);
 174                 close($FH);
 175         } else {
 176                 copy('/etc/cups/printers.conf', $path);
 177         }
 178 }
 179
 180 sub psystem {
 181         print "  @_\n";
 182         system(@_);
 183 }
 184
 185 sub import_print_queues {
 186         local ($path) = @_;
 187         my $service = print_service();
 188         my $FH, %printer, @options;
 189
 190         # store queue info in the 'active' print service
 191         open($FH, "<$path");
 192         while (<$FH>) {
 193                 if (/<Printer (.+)>/) {
 194                         $printer{'Printer'} = $1;
 195                         @options = ();
 196                         push(@options, "-p", $1);
 197                 } elsif (/([^\s]+)\s(.+)/) {
 198                         $printer{$1} = $2;
 199                         my $value = $2;
 200                         ($1 eq 'DeviceURI') &&
 201                                 push(@options, "-v", $value);
 202                         ($1 eq 'Info') &&
 203                                 push(@options, "-D", $value);
 204                 } elsif (/<\/Printer>/) {
 205                         ($service eq 'lp') &&
 206                                 push(@options, "-m", "uri");
 207                         print "importing $printer{'Printer'}...\n";
 208                         # create a queue
 209                         psystem($LPADMIN, @options);
 210                         psystem($ENABLE, $printer{'Printer'});
 211                         ($printer{'Accepting'} eq 'Yes') &&
 212                                 psystem($ACCEPT, $printer{'Printer'});
 213                         $printer = ();
 214                 }
 215         }
 216         close($FH);
 217 }
 218
 219 sub select_service {
 220         my ($service, $migrate) = @_;
 221         my $FH, $queues;
 222
 223         if (! -d "/usr/lib/$service/bin") {
 224                 fatal("print service: $service is not installed\n");
 225         }
 226
 227         if ($migrate == 1) {
 228                 # export old print queue configuration (if migrating)
 229                 $queues = tmpnam();
 230                 export_print_queues($queues);
 231         }
 232
 233         # enable/disable the services
 234         if ($service eq 'cups') {
 235                 (-f '/etc/printers.conf') && (! -f '/etc/lp/printers.conf') &&
 236                         rename('/etc/printers.conf', '/etc/lp/printers.conf');
 237                 print("disabling LP services...\n");
 238                 svcadm("disable", $SVC_LP_SCHEDULER, $SVC_LP_IPP, $SVC_LP_LPD,
 239                                   $SVC_LP_PPD);
 240                 print("enabling CUPS services...\n");
 241                 svcadm("enable", $SVC_CUPS_SCHEDULER, $SVC_CUPS_LPD);
 242                 svccfg("cups/scheduler:default",
 243                         "setprop general/active = boolean: true");
 244                 system("pkill -USR1 -f '/desktop-print-management-applet'");
 245         } else {
 246                 print("disabling CUPS services...\n");
 247                 svcadm("disable", $SVC_CUPS_SCHEDULER, $SVC_CUPS_LPD);
 248                 print("enabling LP services...\n");
 249                 svcadm("enable", $SVC_LP_SCHEDULER, $SVC_LP_IPP, $SVC_LP_LPD,
 250                                   $SVC_LP_PPD);
 251                 (-f '/etc/lp/printers.conf') &&
 252                         rename('/etc/lp/printers.conf', '/etc/printers.conf');
 253                 svccfg("cups/scheduler:default", "delprop general/active");
 254                 system("pkill -USR1 -f '/desktop-print-management-applet'");
 255         }
 256
 257         # import the new print queue configuration (if migrating)
 258         defined($queues) && import_print_queues($queues);
 259 }
 260
 261 sub query_service {
 262         my $service = print_service();
 263
 264         if (!defined($service)) {
 265                 fatal("failed to detect active print service: $!\n");
 266         }
 267         print "active print service: $service\n";
 268 }
 269
 270 if ($cmd eq 'print-service') {
 271         my ($import_path, $export_path, $svc_name, $query, $migrate) = ();
 272
 273         my $res = GetOptions('q|query' => \$query, 's|set=s' => \$service,
 274                 'm|migrate' => \$migrate, 'e|export=s' => \$export_path,
 275                 'i|import=s' => \$import_path);
 276
 277         ($res) || usage();
 278
 279         if (defined($import_path) && !defined($export_path) &&
 280             !defined($query) && !defined($service) && !defined($migrate)) {
 281                 import_print_queues($import_path);
 282         } elsif (!defined($import_path) && defined($export_path) &&
 283                  !defined($query) && !defined($service) && !defined($migrate)) {
 284                 export_print_queues($export_path);
 285         } elsif (!defined($import_path) && !defined($export_path) &&
 286                  defined($query) && !defined($service) && !defined($migrate)) {
 287                 query_service();
 288         } elsif (!defined($import_path) && !defined($export_path) &&
 289                  !defined($query) && defined($service)) {
 290                 select_service($service, $migrate);
 291         } else {
 292                 usage();
 293         }
 294 } else {
 295         print_command($cmd, @ARGV);
 296 }
 297
 298 exit(0);