View Javadoc

    1   /*
    2    * Licensed to the Apache Software Foundation (ASF) under one or more
    3    * contributor license agreements.  See the NOTICE file distributed with
    4    * this work for additional information regarding copyright ownership.
    5    * The ASF licenses this file to You under the Apache License, Version 2.0
    6    * (the "License"); you may not use this file except in compliance with
    7    * the License.  You may obtain a copy of the License at
    8    *
    9    *  http://www.apache.org/licenses/LICENSE-2.0
    10   *
    11   * Unless required by applicable law or agreed to in writing, software
    12   * distributed under the License is distributed on an "AS IS" BASIS,
    13   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    14   * See the License for the specific language governing permissions and
    15   * limitations under the License.
    16   */
    17
    18  package org.apache.any23.source;
    19
    20  import java.io.ByteArrayInputStream;
    21  import java.io.IOException;
    22  import java.io.InputStream;
    23  import java.nio.charset.StandardCharsets;
    24
    25  /**
    26   * String implementation of {@link DocumentSource}.
    27   */
    28  public class StringDocumentSource implements DocumentSource {
    29
    30      private final String in;
    31
    32      private final String contentType;
    33
    34      private final String encoding;
    35
    36      private final String uri;
    37
    38      public StringDocumentSource(String in, String uri) {
    39          this(in, uri, null, null);
    40      }
    41
    42      public StringDocumentSource(String in, String uri, String contentType) {
    43          this(in, uri, contentType, null);
    44      }
    45
    46      public StringDocumentSource(String in, String uri, String contentType, String encoding) {
    47          this.in = in;
    48          this.uri = uri;
    49          this.contentType = contentType;
    50          this.encoding = encoding;
    51      }
    52
    53      public InputStream openInputStream() throws IOException {
    54          if (encoding == null) {
    55              return new ByteArrayInputStream(in.getBytes(StandardCharsets.UTF_8));
    56          }
    57          return new ByteArrayInputStream(in.getBytes(encoding));
    58      }
    59
    60      public long getContentLength() {
    61          return in.length();
    62      }
    63
    64      public String getDocumentIRI() {
    65          return uri;
    66      }
    67
    68      public String getContentType() {
    69          return contentType;
    70      }
    71
    72      public boolean isLocal() {
    73          return true;
    74      }
    75
    76  }