766d7deff094829a527c315d3f8b61bdbc8f8640
[sonitus.git] / src / main / java / net / pterodactylus / sonitus / data / source / FileSource.java
1 /*
2  * Sonitus - FileSource.java - Copyright © 2013 David Roden
3  *
4  * This program is free software: you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License as published by
6  * the Free Software Foundation, either version 3 of the License, or
7  * (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
16  */
17
18 package net.pterodactylus.sonitus.data.source;
19
20 import static com.google.common.base.Preconditions.checkNotNull;
21
22 import java.io.EOFException;
23 import java.io.FileInputStream;
24 import java.io.IOException;
25 import java.io.InputStream;
26 import java.util.Arrays;
27 import java.util.Collections;
28 import java.util.List;
29
30 import net.pterodactylus.sonitus.data.Controller;
31 import net.pterodactylus.sonitus.data.Metadata;
32 import net.pterodactylus.sonitus.data.Source;
33 import net.pterodactylus.sonitus.io.IdentifyingInputStream;
34
35 import com.google.common.base.Optional;
36
37 /**
38  * A {@link net.pterodactylus.sonitus.data.Source} that is read from the local
39  * file system.
40  *
41  * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
42  */
43 public class FileSource implements Source {
44
45         /** The path of the file. */
46         private final String path;
47
48         /** The identified metadata of the file. */
49         private final Metadata metadata;
50
51         /** The input stream. */
52         private InputStream fileInputStream;
53
54         /**
55          * Creates a new file source.
56          *
57          * @param path
58          *              The path of the file
59          * @throws java.io.IOException
60          *              if the file can not be found, or an I/O error occurs
61          */
62         public FileSource(String path) throws IOException {
63                 this.path = checkNotNull(path, "path must not be null");
64                 fileInputStream = new FileInputStream(path);
65
66                 /* identify file type. */
67                 Optional<IdentifyingInputStream> identifyingInputStream = IdentifyingInputStream.create(new FileInputStream(path));
68                 if (identifyingInputStream.isPresent()) {
69                         metadata = identifyingInputStream.get().metadata();
70                 } else {
71                         /* fallback. */
72                         metadata = new Metadata().name(path);
73                 }
74         }
75
76         //
77         // CONTROLLED METHODS
78         //
79
80         @Override
81         public String name() {
82                 return path;
83         }
84
85         @Override
86         public List<Controller<?>> controllers() {
87                 return Collections.emptyList();
88         }
89
90         //
91         // SOURCE METHODS
92         //
93
94         @Override
95         public byte[] get(int bufferSize) throws IOException {
96                 byte[] buffer = new byte[bufferSize];
97                 int read = fileInputStream.read(buffer);
98                 if (read == -1) {
99                         throw new EOFException();
100                 }
101                 return Arrays.copyOf(buffer, read);
102         }
103
104         @Override
105         public Metadata metadata() {
106                 return metadata;
107         }
108
109         //
110         // OBJECT METHODS
111         //
112
113         @Override
114         public String toString() {
115                 return String.format("%s (%s)", path, metadata);
116         }
117
118 }