1 files changed, 72 insertions, 0 deletions
diff --git a/src/main/java/org/traccar/database/MediaManager.java b/src/main/java/org/traccar/database/MediaManager.java
new file mode 100644
index 000000000..edade5766
--- /dev/null
+++ b/src/main/java/org/traccar/database/MediaManager.java
@@ -0,0 +1,72 @@
+/*
+ * Copyright 2017 - 2018 Anton Tananaev (anton@traccar.org)
+ *
+ * Licensed 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.traccar.database;
+
+import io.netty.buffer.ByteBuf;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.io.File;
+import java.io.FileOutputStream;
+import java.io.IOException;
+import java.nio.ByteBuffer;
+import java.nio.channels.FileChannel;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+import java.text.SimpleDateFormat;
+import java.util.Date;
+
+public class MediaManager {
+
+    private static final Logger LOGGER = LoggerFactory.getLogger(MediaManager.class);
+
+    private String path;
+
+    public MediaManager(String path) {
+        this.path = path;
+    }
+
+    private File createFile(String uniqueId, String name) throws IOException {
+        Path filePath = Paths.get(path, uniqueId, name);
+        Path directoryPath = filePath.getParent();
+        if (directoryPath != null) {
+            Files.createDirectories(directoryPath);
+        }
+        return filePath.toFile();
+    }
+
+    public String writeFile(String uniqueId, ByteBuf buf, String extension) {
+        if (path != null) {
+            int size = buf.readableBytes();
+            String name = new SimpleDateFormat("yyyyMMddHHmmss").format(new Date()) + "." + extension;
+            try (FileOutputStream output = new FileOutputStream(createFile(uniqueId, name));
+                    FileChannel fileChannel = output.getChannel()) {
+                    ByteBuffer byteBuffer = buf.nioBuffer();
+                int written = 0;
+                while (written < size) {
+                    written += fileChannel.write(byteBuffer);
+                }
+                fileChannel.force(false);
+                return name;
+            } catch (IOException e) {
+                LOGGER.warn("Save media file error", e);
+            }
+        }
+        return null;
+    }
+
+}