import java.io.DataOutputStream;
import java.io.IOException;
import java.net.Authenticator;
import java.net.PasswordAuthentication;
import java.net.URL;
import java.util.Map;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.xml.ws.BindingProvider;
import javax.xml.ws.WebServiceFeature;
import javax.xml.ws.soap.MTOMFeature;

import org.apache.ws.axis2.Bizstream;
import org.apache.ws.axis2.BizstreamPortType;

import com.brainsellers.schemas.webservices.bizstream._2006_10.DataType;
import com.brainsellers.schemas.webservices.bizstream._2006_10.Excel;
import com.brainsellers.schemas.webservices.bizstream._2006_10.GenerateDataRequest;
import com.brainsellers.schemas.webservices.bizstream._2006_10.LayoutData;
import com.brainsellers.schemas.webservices.bizstream._2006_10.LayoutDefinition;
import com.brainsellers.schemas.webservices.common._2006_10.FileData;



public class WSS_JAXWS_Sample6 extends HttpServlet {

    // Webサービスサーバのホスト名
    private static final String WSS_HOST_NAME = "wsssrv";

    // Webサービスサーバ上のサンプルディレクトリの場所
    private static final String WSS_BIZSTREAM_SAMPLE_DIR = "/bs/sample";

    // Webサービスのエンドポイントプリフィックス
    private static final String WSS_SERVICE_LOCATION = "http://" + WSS_HOST_NAME + ":8080/axis2/services";

    // BASIC認証のユーザ名
    private static final String USER_NAME = "bizuser1";

    // BASIC認証のパスワード
    private static final String PASSWORD = "bizuser1";

    // 監査ロギング用のユーザ定義
    private static final String USER_DEF = "AAA";

    // Webサービスサーバ上の入力レイアウト定義ファイルパス
    private static final String LAYOUT_FILE = WSS_BIZSTREAM_SAMPLE_DIR + "/xml/Sample6.xml";

    // ダウンロードファイル名
    private static final String DOWNLOAD_FILE_NAME = "WSS_JAXWS_Sample6.xls";


    /**
     * @see HttpServlet#HttpServlet()
     */
    public WSS_JAXWS_Sample6() {
        super();
    }

    /**
     * @see HttpServlet#doGet(HttpServletRequest request, HttpServletResponse
     *      response)
     */
    protected void doGet(HttpServletRequest httpRequest, HttpServletResponse httpResponse) throws ServletException, IOException {
        try {
            // 認証の設定
            setAuth();

            // データの生成
            byte[] data = generate();

            // HTTPヘッダを設定する
            httpResponse.setCharacterEncoding("Windows-31J");
            httpResponse.setContentType("application/excel");
            httpResponse.setContentLength(data.length);
            httpResponse.addHeader("Accept-Ranges", "bytes");
            httpResponse.addHeader("Content-Disposition", "attachment; filename=\"" + DOWNLOAD_FILE_NAME + "\"");

            // PDFをストリームに書き出す
            DataOutputStream out_res = new DataOutputStream(httpResponse.getOutputStream());
            out_res.write(data);
            out_res.close();
        } catch (Exception e) {
            e.printStackTrace();
            throw new ServletException(e);
        }
    }

    /**
     * 認証の設定
     */
    private void setAuth() {
        Authenticator.setDefault(new Authenticator() {
            protected PasswordAuthentication getPasswordAuthentication() {
                return new PasswordAuthentication(USER_NAME, PASSWORD.toCharArray());
            }
        });
    }

    /**
     * アプリケーションでデータソースを設定してPDFを生成しダウンロードする
     *
     * @param
     * @return byte[]
     */
    private byte[] generate() throws Exception {
        // クライアントを生成
        Bizstream service = new Bizstream(new URL(WSS_SERVICE_LOCATION + "/bizstream?wsdl"));

        // MTOMを有効化
        WebServiceFeature feature = new MTOMFeature(true);
        BizstreamPortType type = service.getBizstreamSOAP11Port(feature);
        BindingProvider bp = (BindingProvider) type;
        Map<String, Object> reqContext = bp.getRequestContext();
        reqContext.put(BindingProvider.ENDPOINT_ADDRESS_PROPERTY,
                WSS_SERVICE_LOCATION + "/bizstream?UserDef=" + USER_DEF);

        // リクエストメッセージを作成
        GenerateDataRequest request = new GenerateDataRequest();

        // 出力ファイルの種類を生成
        DataType dataType = new DataType();
        Excel excel = new Excel();
//        excel.setMultisheet("true"); // 1ページ毎にシートを分ける
        dataType.setExcel(excel);
        request.setDataType(dataType);

        // レイアウト定義を設定
        LayoutData layoutData = new LayoutData();
        LayoutDefinition layout = new LayoutDefinition();
        layout.setUri(LAYOUT_FILE);
        layoutData.getLayoutDefinition().add(layout);
        request.setLayoutData(layoutData);

        // リクエストを送信
        FileData data = type.generateData(request);

        return data.getContentData().getValue();
    }
}