#ifndef PICTUREDAO_H
#define PICTUREDAO_H

#include <QSqlDatabase>
#include <QVector>

#include "Picture.h"

class QSqlDatabase;
class Picture;

class PictureDao {
 public:
  PictureDao(QSqlDatabase& database);
  void init() const;

  void addPictureInAlbum(int albumId, Picture& picture) const;
  void removePicture(int id) const;
  void removePictureForAlbum(int albumId) const;
  std::unique_ptr<std::vector<std::unique_ptr<Picture>>> picturesForAlbum(
      int albumId) const;

 private:
  QSqlDatabase& mDatabase;
  const QString mTableName = "pictures";
};

#endif  // PICTUREDAO_H