gerrit.o-ran-sc Code Review - smo/teiv.git/blob

   1 /*
   2  *  ============LICENSE_START=======================================================
   3  *  Copyright (C) 2024 Ericsson
   4  *  Modifications Copyright (C) 2024 OpenInfra Foundation Europe
   5  *  ================================================================================
   6  *  Licensed under the Apache License, Version 2.0 (the "License");
   7  *  you may not use this file except in compliance with the License.
   8  *  You may obtain a copy of the License at
   9  *
  10  *        http://www.apache.org/licenses/LICENSE-2.0
  11  *
  12  *  Unless required by applicable law or agreed to in writing, software
  13  *  distributed under the License is distributed on an "AS IS" BASIS,
  14  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15  *  See the License for the specific language governing permissions and
  16  *  limitations under the License.
  17  *
  18  *  SPDX-License-Identifier: Apache-2.0
  19  *  ============LICENSE_END=========================================================
  20  */
  21 package org.oran.smo.yangtools.parser.input;
  22
  23 import java.io.File;
  24 import java.io.FileInputStream;
  25 import java.io.InputStream;
  26 import java.util.Objects;
  27
  28 /**
  29  * An input based on a file
  30  *
  31  * @author Mark Hollmann
  32  */
  33 public class FileBasedYangInput implements YangInput {
  34
  35     private final File file;
  36
  37     public FileBasedYangInput(final File file) {
  38         this.file = Objects.requireNonNull(file);
  39         if (!file.isFile() || !file.exists()) {
  40             throw new RuntimeException("File not found: " + file.getAbsolutePath());
  41         }
  42     }
  43
  44     @Override
  45     public String getName() {
  46         return file.getName();
  47     }
  48
  49     @Override
  50     public File getFile() {
  51         return file;
  52     }
  53
  54     @Override
  55     public InputStream getInputStream() {
  56         try {
  57             return new FileInputStream(file);
  58         } catch (final Exception ex) {
  59             throw new RuntimeException("Cannot open input stream: " + ex.getMessage(), ex);
  60         }
  61     }
  62
  63     @Override
  64     public String getMediaType() {
  65         if (file.getName().endsWith(".json")) {
  66             return MEDIA_TYPE_YANG_DATA_JSON;
  67         } else if (file.getName().endsWith(".xml")) {
  68             return MEDIA_TYPE_YANG_DATA_XML;
  69         }
  70
  71         return MEDIA_TYPE_YANG;
  72     }
  73
  74     @Override
  75     public int hashCode() {
  76         return file.getName().hashCode();
  77     }
  78
  79     @Override
  80     public boolean equals(final Object other) {
  81         if (other == this) {
  82             return true;
  83         }
  84         if (other == null) {
  85             return false;
  86         }
  87         if (this.getClass() == other.getClass()) {
  88             return this.file.equals(((FileBasedYangInput) other).file);
  89         }
  90
  91         return false;
  92     }
  93
  94     @Override
  95     public String toString() {
  96         return getName();
  97     }
  98 }