1 files changed, 31 insertions, 0 deletions
diff --git a/src/server/database/image.go b/src/server/database/image.go
new file mode 100644
index 0000000..d85063f
--- /dev/null
+++ b/src/server/database/image.go
@@ -0,0 +1,31 @@
+package database
+
+import (
+	"database/sql"
+)
+
+func getThumbnailOrImage(thumbnail bool) string {
+	if thumbnail {
+		return "Posts.thumbnail "
+	}
+	return "Posts.image "
+}
+
+func GetImage(uid int, thumbnail bool) ([]byte, error) {
+	var db = GetDb()
+
+	var image []byte
+	row := db.QueryRow("SELECT "+getThumbnailOrImage(thumbnail)+
+		"FROM Posts "+
+		"WHERE Posts.uid = ?;", uid)
+	if err := row.Scan(&image); err != nil {
+
+		if err == sql.ErrNoRows {
+			return nil, nil
+		}
+
+		return nil, err
+	}
+
+	return image, nil
+}