/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License (the "License").
 * You may not use this file except in compliance with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */
/*
 * ident	"%Z%%M%	%I%	%E% SMI"
 *
 * Copyright 2005 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 *
 * DoPrinterAdd class
 * Worker class for adding local and remote printers.
 */

package com.sun.admin.pm.server;

import java.io.*;
import java.util.*;

public class  DoPrinterAdd {

    public static void main(String[] args) {
	//
	// Set attributes for testing
	//
	NameService ns = new NameService();

	String file_contents[] = new String[1];
	file_contents[0] = "any";
	String user_allow_list[] = new String[1];
	user_allow_list[0] = "allow1";
	String user_deny_list[] = new String[1];
	user_deny_list[0] = "deny1";

	Printer p = new Printer(ns);
	p.setPrinterName("javatest");
	p.setPrinterType("PS");
	p.setPrintServer("zelkova");
	p.setComment("This is a comment");
	p.setDevice("/var/tmp/test");
	p.setNotify("none");
	p.setProtocol("bsd");
	p.setDestination(null);
	p.setFileContents(file_contents);
	p.setUserAllowList(user_allow_list);
	p.setIsDefaultPrinter(false);
	p.setBanner("never");
	p.setEnable(true);
	p.setAccept(true);
	p.setLocale(null);

	try {
		add(p, ns);
	}
	catch (Exception e)
	{
		System.out.println(e);
		System.exit(1);
	}
	System.out.println("Commands:\n" + p.getCmdLog());
	System.out.println("Errors:\n" + p.getErrorLog());
	System.out.println("Warnings:\n" + p.getWarnLog());
	System.exit(0);
    }

    //
    // Interface to Printer object.
    //
    public static void add(
	Printer p,
	NameService ns) throws Exception
    {
	Debug.message("SVR: DoPrinterAdd.add()");

	String device = p.getDevice();
	if (device == null) {
		addRemote(p, ns);
	} else {
		addLocal(p, ns);
	}
	return;
    }

    //
    // Do the work of adding a local printer.
    //
    private static void addLocal(
	Printer p,
	NameService ns) throws Exception
    {
	Debug.message("SVR: DoPrinterAdd.addLocal()");

	int exitvalue = 0;
	String err = null;
	String cmd = null;
	SysCommand syscmd = null;

	// Since it's local set extensions.
	// Eventually the gui should do this.
	p.setExtensions("Solaris");

	String printername = p.getPrinterName();
	String printertype = p.getPrinterType();
	String printserver = p.getPrintServer();
	String comment = p.getComment();
	String device = p.getDevice();
	String make = p.getMake();
	String model = p.getModel();
	String ppd = p.getPPD();
	String notify = p.getNotify();
	String protocol = p.getProtocol();
	String destination = p.getDestination();
	String[] file_contents = p.getFileContents();
	String[] user_allow_list = p.getUserAllowList();
	String[] user_deny_list = p.getUserDenyList();
	boolean default_printer = p.getIsDefaultPrinter();
	String banner = p.getBanner();
	boolean enable = p.getEnable();
	boolean accept = p.getAccept();

	String nameservice = ns.getNameService();
	String ppdfile = null;

	//
	// "uri" is a pseudo protocol and means that the device is
	// specified in the destination.
	//
	if ((protocol != null) && (protocol.equals("uri"))) {
		device = destination;
		destination = null;
		protocol = null;
	}

	if (ppd != null) {
		ppdfile = new String(
			DoPrinterUtil.getPPDFile(make, model, ppd));
	}

	cmd = "/usr/sbin/lpadmin -p " + printername;
	if (printserver != null)
		cmd = cmd.concat(" -s " + printserver);
	if (device != null) {
		cmd = cmd.concat(" -v " + device);

		if (device.indexOf("://") != -1)
				cmd = cmd.concat(" -m uri");
		else {
			if (destination != null)
				cmd = cmd.concat(" -m netstandard");
			else
				cmd = cmd.concat(" -m standard");
			if (ppd != null)
				cmd = cmd.concat("_foomatic");
		}
	}
	if (printertype != null)
		cmd = cmd.concat(" -T " + printertype);
	if (notify != null)
		cmd = cmd.concat(" -A " + notify);
	if (ppdfile != null) {
		cmd = cmd.concat(" -n " + ppdfile);
	}


	if (destination != null)
		cmd = cmd.concat(" -o dest=" + destination);
	if (protocol != null)
		cmd = cmd.concat(" -o protocol=" + protocol);
	if (banner != null)
		cmd = cmd.concat(" -o banner=" + banner);

	if ((file_contents != null) && (file_contents.length != 0)) {
		String tmpstr = file_contents[0];
		for (int i = 1; i < file_contents.length; i++) {
			tmpstr = tmpstr.concat("," + file_contents[i]);
		}
		cmd = cmd.concat(" -I " + tmpstr);
	} else {
		cmd = cmd.concat(" -I postscript");
	}
	if ((user_allow_list != null) && (user_allow_list.length != 0)) {
		String tmpstr = user_allow_list[0];
		for (int i = 1; i < user_allow_list.length; i++) {
			tmpstr = tmpstr.concat("," + user_allow_list[i]);
		}
		cmd = cmd.concat(" -u allow:" + tmpstr);
	}

	p.setCmdLog(cmd);
	syscmd = new SysCommand();
	syscmd.exec(cmd);

	err = syscmd.getError();
	if (syscmd.getExitValue() != 0) {
		syscmd = null;
		p.setErrorLog(err);
		throw new pmCmdFailedException(err);
	} else {
		p.setWarnLog(err);
	}
	syscmd = null;

	//
	// lpadmin won't take allow and deny lists together
	// so do the deny seperately.
	//
	if ((user_deny_list != null) && (user_deny_list.length != 0)) {
		String tmpstr = user_deny_list[0];
		for (int i = 1; i < user_deny_list.length; i++) {
			tmpstr = tmpstr.concat("," + user_deny_list[i]);
		}
		cmd = "/usr/sbin/lpadmin -p " + printername +
		    " -u deny:" + tmpstr;
		p.setCmdLog(cmd);
		syscmd = new SysCommand();
		syscmd.exec(cmd);
		err = syscmd.getError();
		if (err != null) {
			p.setWarnLog(err);
		}
		syscmd = null;
	}

	if ((comment != null) && (!comment.equals(""))) {
		//
		// Have to use a command array here since
		// exec(String) doesn't parse quoted strings.
		//
		String cmd_array[] =
			{ "/usr/sbin/lpadmin", "-p", printername,
			"-D", comment };
		cmd = "/usr/sbin/lpadmin -p " + printername + " -D " +
			"\"" + comment + "\"";
		p.setCmdLog(cmd);

		syscmd = new SysCommand();
		syscmd.exec(cmd_array);
		err = syscmd.getError();
		if (err != null) {
			p.setWarnLog(err);
		}
		syscmd = null;
	}

	// If this is the default printer set it.
	// If it fails warn user.
	if (default_printer) {
		cmd = "/usr/sbin/lpadmin -d " + printername;
		p.setCmdLog(cmd);
		syscmd = new SysCommand();
		syscmd.exec(cmd);
		err = syscmd.getError();
		if (err != null) {
			p.setWarnLog(err);
		}
		syscmd = null;
	}

	// Check to see if we should enable it.
	// If it fails warn user.
	if (enable) {
		cmd = "/usr/bin/enable " + printername;
		p.setCmdLog(cmd);
		syscmd = new SysCommand();
		syscmd.exec(cmd);
		err = syscmd.getError();
		if (err != null) {
			p.setWarnLog(err);
		}
		syscmd = null;
	}
	// Check to see if we should accept it.
	// If it fails warn user.
	if (accept) {
		cmd = "/usr/sbin/accept " + printername;
		p.setCmdLog(cmd);
		syscmd = new SysCommand();
		syscmd.exec(cmd);
		err = syscmd.getError();
		if (err != null) {
			p.setWarnLog(err);
		}
		syscmd = null;
	}

	doFilters(p);

	//
	// Take care of name service now.
	//
	if (!nameservice.equals("system")) {
		try {
			DoPrinterNS.set("add", p, ns);
		}
		catch (Exception e) {
			p.clearLogs();
			NameService localns = new NameService();
			//
			// Back out the local printer.
			//
			try {
				DoPrinterDelete.delete(p, localns);
			}
			catch (Exception e2) {
				Debug.message("SVR:" + e2.getMessage());
			}
			p.clearLogs();
			throw (e);
		}
	}
	return;
    }

    //
    // Do the work of adding a remote printer.
    //
    private static void addRemote(
	Printer p,
	NameService ns) throws Exception
    {
	Debug.message("SVR: DoPrinterAdd.addRemote()");

	int exitvalue = 0;
	String err = null;
	String cmd = "";
	String cmd_array[] = new String[7];
	SysCommand syscmd = null;

	String printername = p.getPrinterName();
	String printserver = p.getPrintServer();
	String comment = p.getComment();
	boolean default_printer = p.getIsDefaultPrinter();
	String nameservice = ns.getNameService();

	boolean isnismaster = false;
	if (nameservice.equals("nis")) {
		//
		// Find out if we are the nis master
		//
		String nshost = ns.getNameServiceHost();
		Host h = new Host();
		String lh = h.getLocalHostName();
		if (lh.equals(nshost))
			isnismaster = true;
		h = null;
	}

	//
	// If the name service is not system and we are
	// not the nis master then do the name service
	// update and return.
	//
	if ((!nameservice.equals("system")) && (!isnismaster)) {
		DoPrinterNS.set("add", p, ns);
		return;
	}

	cmd_array[0] = "/usr/sbin/lpadmin";
	cmd_array[1] = "-p";
	cmd_array[2] = printername;
	cmd_array[3] = "-s";
	cmd_array[4] = printserver;

	if ((comment != null) && (!comment.equals(""))) {
		cmd_array[5] = "-D";
		cmd_array[6] = comment;
	}

	//
	// Fix up cmd so we can log it.
	//
	for (int i = 0; i < cmd_array.length; i++) {
		if (cmd_array[i] == null)
			continue;
		if (i == 6) {
			cmd = cmd.concat("\"" + comment + "\"");
			continue;
		}
		cmd = cmd.concat(cmd_array[i] + " ");
	}

	p.setCmdLog(cmd);
	syscmd = new SysCommand();
	syscmd.exec(cmd_array);
	err = syscmd.getError();
	if (syscmd.getExitValue() != 0) {
		p.setErrorLog(err);
		syscmd = null;
		throw new pmCmdFailedException(err);
	}
	if (err != null) {
		p.setWarnLog(err);
	}
	syscmd = null;

	// If this is the default printer set it.
	// If it fails warn user.
	if (default_printer) {
		cmd = "/usr/sbin/lpadmin -d " + printername;
		p.setCmdLog(cmd);
		syscmd = new SysCommand();
		syscmd.exec(cmd);
		err = syscmd.getError();
		if (err != null) {
			p.setWarnLog(err);
		}
		syscmd = null;
	}

	//
	// If it's nis and we are here then we are the nis
	// master. This call will do the make for us.
	//
	if (nameservice.equals("nis")) {
		try {
			DoPrinterNS.set("add", p, ns);
		}
		catch (Exception e) {
			p.clearLogs();
			try {
				//
				// Back out the local printer.
				//
				DoPrinterDelete.delete(p, ns);
			}
			catch (Exception e2)
			{
				Debug.message("SVR:" + e2.getMessage());
			}
			p.clearLogs();
			throw e;
		}
	}
	return;
    }


    //
    // Configure filters
    // Look in /etc/lp/fd and configure each filter if it hasn't
    // already been configured.  We'll add warning messages if
    // there are problems but don't consider anything here fatal.
    //
    private static void doFilters(Printer p) throws Exception
    {
	Debug.message("SVR: DoPrinterAdd.doFilters()");

	int i = 0;
	int j = 0;
	String o = null;
	String err = null;
	String cmd = null;
	String psfilters[] = null;
	SysCommand syscmd = null;

	//
	// Get list of potential filters
	//
	cmd = "/usr/bin/ls /etc/lp/fd";
	syscmd = new SysCommand();
	syscmd.exec(cmd);
	if (syscmd.getExitValue() != 0) {
		syscmd = null;
		return;
	}
	o = syscmd.getOutput();
	syscmd = null;
	if (o == null) {
		return;
	}

	StringTokenizer st = new StringTokenizer(o);
	if (st.countTokens() == 0) {
		return;
	}
	psfilters = new String[st.countTokens()];
	for (i = 0; st.hasMoreTokens(); i++) {
		psfilters[i] = st.nextToken();
	}
	//
	// Remove .fd suffix and empty slots that aren't filters.
	//
	for (i = 0; i < psfilters.length; i++) {
		if (psfilters[i].endsWith(".fd")) {
			j = psfilters[i].indexOf(".fd");
			psfilters[i] = psfilters[i].substring(0, j);
		} else {
			psfilters[i] = "";
		}
	}

	// Get list of currently configured filters
	cmd = "/usr/sbin/lpfilter -l -f all";
	syscmd = new SysCommand();
	syscmd.exec(cmd);

	o = null;
	if (syscmd.getExitValue() != 0) {
		err = syscmd.getError();
		if (err != null) {
			p.setWarnLog(err);
		}
		syscmd = null;
		return;
	} else {
		o = syscmd.getOutput();
	}

	for (i = 0; i < psfilters.length; i++) {
		if (psfilters[i].equals(""))
			continue;

		// If we have filters see if this one is
		// already configured.
		if (o != null) {
			if (o.indexOf("\"" + psfilters[i] + "\"") > -1)
				continue;
		}

		// Add the filter
		cmd = "/usr/sbin/lpfilter -f " + psfilters[i] +
			" -F /etc/lp/fd/" + psfilters[i] + ".fd";
		p.setCmdLog(cmd);
		syscmd = new SysCommand();
		syscmd.exec(cmd);

		if (syscmd.getExitValue() != 0) {
			err = syscmd.getError();
			if (err != null) {
				p.setWarnLog(err);
			}
		}
		syscmd = null;
	}
    }
}