cpp-study/MovieDatabase.h at main · HonourBoundGameStudios/cpp-study · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
#ifndef MOVIE_DATABASE_H
#define MOVIE_DATABASE_H

#include <list>

#include "Movie.h"
#include "json/single_include/nlohmann/json.hpp"

using MovieList = std::list<Movie>;

class MovieDatabase
{
public:
    MovieDatabase() = default;

    void PopulateWithFakeData();

    const MovieList &GetMovies() const
    {
        return movies;
    }

    MovieList GetMoviesSortedByTitle() const
    {
        auto sortedMovies = movies;

        sortedMovies.sort([](const Movie &a, const Movie &b) {
            return a.GetTitle() < b.GetTitle();
        });

        return sortedMovies;
    }

    MovieList GetMoviesSortedByRating() const
    {
        auto sortedMovies = movies;

        sortedMovies.sort([](const Movie &a, const Movie &b) {
            return a.GetRating() > b.GetRating();
        });

        return sortedMovies;
    }

    void AddMovie(const std::string& title, float rating);

private:

    MovieList movies;
};

#endif