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
}