modules/tomcat-lite/java/org/apache/tomcat/lite/proxy/StaticContentService.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.apache.tomcat.lite.proxy;

import java.io.FileInputStream;
import java.io.IOException;
import java.util.logging.Logger;

import org.apache.tomcat.lite.http.HttpRequest;
import org.apache.tomcat.lite.http.HttpResponse;
import org.apache.tomcat.lite.http.HttpChannel.HttpService;
import org.apache.tomcat.lite.io.BBuffer;
import org.apache.tomcat.lite.io.BBucket;
import org.apache.tomcat.lite.io.IOBuffer;

/*
 *
 * Serve static content, from memory.
 */
public class StaticContentService implements HttpService  {
    protected Logger log = Logger.getLogger("coyote.static");
    protected BBucket mb;

    protected boolean chunked = false;
    int code = 200;

    protected String contentType = "text/plain";


    public StaticContentService() {
    }

    /**
     * Used for testing chunked encoding.
     * @return
     */
    public StaticContentService chunked() {
      chunked = true;
      return this;
    }

    public StaticContentService setData(byte[] data) {
        mb = BBuffer.wrapper(data, 0, data.length);
        return this;
    }

    public StaticContentService setStatus(int status) {
        this.code = status;
        return this;
    }

    public StaticContentService withLen(int len) {
        byte[] data = new byte[len];
        for (int i = 0; i < len; i++) {
          data[i] = 'A';
        }
        mb = BBuffer.wrapper(data, 0, data.length);
        return this;
      }


    public StaticContentService setData(CharSequence data) {
      try {
          IOBuffer tmp = new IOBuffer(null);
          tmp.append(data);
          mb = tmp.readAll(null);
      } catch (IOException e) {
      }
      return this;
    }

    public StaticContentService setContentType(String ct) {
      this.contentType = ct;
      return this;
    }

    public void setFile(String path) {
      try {
        FileInputStream fis = new FileInputStream(path);
        BBuffer bc = BBuffer.allocate(4096);

        byte b[] = new byte[4096];
        int rd = 0;
        while ((rd = fis.read(b)) > 0) {
            bc.append(b, 0, rd);
        }
        mb = bc;
      } catch (IOException e) {
        throw new RuntimeException(e);
      }
    }

    @Override
    public void service(HttpRequest httpReq, HttpResponse res) throws IOException {

        res.setStatus(code);

          if (!chunked) {
            res.setContentLength(mb.remaining());
          }
          res.setContentType(contentType);

          int len = mb.remaining();
          int first = 0;

          if (chunked) {
              first = len / 2;
              res.getBody()
                  .queue(BBuffer.wrapper(mb, 0, first));
              res.flush();
          }

          res.getBody().queue(BBuffer.wrapper(mb, 0, len - first));
    }
}