#ifndef ALBUMDAO_H
#define ALBUMDAO_H

#include "Album.h"

class QSqlDatabase;
class Album;

class AlbumDao {
 public:
  AlbumDao(QSqlDatabase& database);
  /**
   * @brief init 创建 albums 表
   */
  void init() const;

  /**
   * @brief addAlbum 新增
   * @param album
   */
  void addAlbum(Album& album) const;
  /**
   * @brief updateAlbum 更新
   * @param album
   */
  void updateAlbum(const Album& album) const;
  /**
   * @brief removeAlbum 删除
   * @param id
   */
  void removeAlbum(int id) const;
  /**
   * @brief albums 查询
   * @return
   */
  std::unique_ptr<std::vector<std::unique_ptr<Album>>> albums() const;

 private:
  QSqlDatabase& mDatabase;
};

#endif  // ALBUMDAO_H