Added release country support

src/libs/database/impl/Migration.cpp

@@ -35,7 +35,7 @@ namespace lms::db
 {
     namespace
     {
-        static constexpr Version LMS_DATABASE_VERSION{ 81 };
+        static constexpr Version LMS_DATABASE_VERSION{ 82 };
     }
 
     VersionInfo::VersionInfo()
@@ -1072,6 +1072,30 @@ FROM tracklist)");
         utils::executeCommand(*session.getDboSession(), "UPDATE scan_settings SET scan_version = scan_version + 1");
     }
 
+    void migrateFromV81(Session& session)
+    {
+        // Add country + release country
+        utils::executeCommand(*session.getDboSession(), R"(CREATE TABLE IF NOT EXISTS "country" (
+  "id" integer primary key autoincrement,
+  "version" integer not null,
+  "name" text not null
+    ))");
+
+        utils::executeCommand(*session.getDboSession(), R"(CREATE TABLE IF NOT EXISTS "release_country" (
+        "country_id" bigint,
+        "release_id" bigint,
+        primary key ("country_id", "release_id"),
+        constraint "fk_release_country_key1" foreign key ("country_id") references "country" ("id") on delete cascade deferrable initially deferred,
+        constraint "fk_release_country_key2" foreign key ("release_id") references "release" ("id") on delete cascade deferrable initially deferred
+    ))");
+
+        utils::executeCommand(*session.getDboSession(), R"(CREATE INDEX "release_country_country" on "release_country" ("country_id"))");
+        utils::executeCommand(*session.getDboSession(), R"(CREATE INDEX "release_country_release" on "release_country" ("release_id"))");
+
+        // Just increment the scan version of the settings to make the next scan rescan everything
+        utils::executeCommand(*session.getDboSession(), "UPDATE scan_settings SET scan_version = scan_version + 1");
+    }
+
     bool doDbMigration(Session& session)
     {
         constexpr std::string_view outdatedMsg{ "Outdated database, please rebuild it (delete the .db file and restart)" };
@@ -1129,6 +1153,7 @@ FROM tracklist)");
             { 78, migrateFromV78 },
             { 79, migrateFromV79 },
             { 80, migrateFromV80 },
+            { 81, migrateFromV81 },
         };
 
         bool migrationPerformed{};

src/libs/database/impl/Release.cpp

@@ -253,6 +253,50 @@ namespace lms::db
 
     } // namespace
 
+    Country::Country(std::string_view name)
+        : _name{ name }
+    {
+        // As we use the name to uniquely identoify release type, we must throw (and not truncate)
+        if (name.size() > _maxNameLength)
+            throw Exception{ "Country name is too long: " + std::string{ name } + "'" };
+    }
+
+    Country::pointer Country::create(Session& session, std::string_view name)
+    {
+        return session.getDboSession()->add(std::unique_ptr<Country>{ new Country{ name } });
+    }
+
+    std::size_t Country::getCount(Session& session)
+    {
+        return utils::fetchQuerySingleResult(session.getDboSession()->query<int>("SELECT COUNT(*) FROM country"));
+    }
+
+    Country::pointer Country::find(Session& session, CountryId id)
+    {
+        session.checkReadTransaction();
+
+        return utils::fetchQuerySingleResult(session.getDboSession()->query<Wt::Dbo::ptr<Country>>("SELECT c from country c").where("c.id = ?").bind(id));
+    }
+
+    Country::pointer Country::find(Session& session, std::string_view name)
+    {
+        session.checkReadTransaction();
+
+        if (name.size() > _maxNameLength)
+            throw Exception{ "Requeted Country name is too long: " + std::string{ name } + "'" };
+
+        return utils::fetchQuerySingleResult(session.getDboSession()->query<Wt::Dbo::ptr<Country>>("SELECT c from country c").where("c.name = ?").bind(name));
+    }
+
+    RangeResults<CountryId> Country::findOrphanIds(Session& session, std::optional<Range> range)
+    {
+        session.checkReadTransaction();
+
+        // select the labels that have no releases
+        auto query{ session.getDboSession()->query<CountryId>("select c.id from country c LEFT OUTER JOIN release_country r_c ON c.id = r_c.country_id WHERE r_c.release_id IS NULL") };
+        return utils::execRangeQuery<CountryId>(query, range);
+    }
+
     Label::Label(std::string_view name)
         : _name{ name }
     {
@@ -612,6 +656,11 @@ namespace lms::db
         _labels.clear();
     }
 
+    void Release::clearCountries()
+    {
+        _countries.clear();
+    }
+
     void Release::clearReleaseTypes()
     {
         _releaseTypes.clear();
@@ -622,6 +671,11 @@ namespace lms::db
         _labels.insert(getDboPtr(label));
     }
 
+    void Release::addCountry(ObjectPtr<Country> country)
+    {
+        _countries.insert(getDboPtr(country));
+    }
+
     void Release::addReleaseType(ObjectPtr<ReleaseType> releaseType)
     {
         _releaseTypes.insert(getDboPtr(releaseType));
@@ -659,8 +713,22 @@ namespace lms::db
     {
         std::vector<std::string> res;
 
-        for (const auto& label : _labels)
+        auto query{ _labels.find() };
+        utils::forEachQueryResult(query, [&](const Label::pointer& label) {
             res.push_back(std::string{ label->getName() });
+        });
+
+        return res;
+    }
+
+    std::vector<std::string> Release::getCountryNames() const
+    {
+        std::vector<std::string> res;
+
+        auto query{ _countries.find() };
+        utils::forEachQueryResult(query, [&](const Country::pointer& country) {
+            res.push_back(std::string{ country->getName() });
+        });
 
         return res;
     }
@@ -669,8 +737,10 @@ namespace lms::db
     {
         std::vector<std::string> res;
 
-        for (const auto& releaseType : _releaseTypes)
+        auto query{ _releaseTypes.find() };
+        utils::forEachQueryResult(query, [&](const ReleaseType::pointer& releaseType) {
             res.push_back(std::string{ releaseType->getName() });
+        });
 
         return res;
     }

src/libs/database/impl/Session.cpp

@@ -102,6 +102,7 @@ namespace lms::db
         _session.mapClass<AuthToken>("auth_token");
         _session.mapClass<Cluster>("cluster");
         _session.mapClass<ClusterType>("cluster_type");
+        _session.mapClass<Country>("country");
         _session.mapClass<Directory>("directory");
         _session.mapClass<Image>("image");
         _session.mapClass<Label>("label");
@@ -199,6 +200,9 @@ namespace lms::db
             utils::executeCommand(_session, "CREATE INDEX IF NOT EXISTS cluster_cluster_type_idx ON cluster(cluster_type_id)");
             utils::executeCommand(_session, "CREATE INDEX IF NOT EXISTS cluster_type_name_idx ON cluster_type(name)");
 
+            utils::executeCommand(_session, "CREATE INDEX IF NOT EXISTS country_id_idx ON country(id)");
+            utils::executeCommand(_session, "CREATE INDEX IF NOT EXISTS country_name_idx ON country(name)");
+
             utils::executeCommand(_session, "CREATE INDEX IF NOT EXISTS directory_id_idx ON directory(id)");
             utils::executeCommand(_session, "CREATE INDEX IF NOT EXISTS directory_parent_directory_idx ON directory(parent_directory_id)");
             utils::executeCommand(_session, "CREATE INDEX IF NOT EXISTS directory_path_idx ON directory(absolute_path)");
@@ -210,6 +214,7 @@ namespace lms::db
             utils::executeCommand(_session, "CREATE INDEX IF NOT EXISTS image_path_idx ON image(absolute_file_path)");
             utils::executeCommand(_session, "CREATE INDEX IF NOT EXISTS image_stem_idx ON image(stem COLLATE NOCASE)");
 
+            utils::executeCommand(_session, "CREATE INDEX IF NOT EXISTS label_id_idx ON label(id)");
             utils::executeCommand(_session, "CREATE INDEX IF NOT EXISTS label_name_idx ON label(name)");
 
             utils::executeCommand(_session, "CREATE INDEX IF NOT EXISTS listen_backend_idx ON listen(backend)");

src/libs/database/include/database/CountryId.hpp

@@ -0,0 +1,24 @@
+/*
+ * Copyright (C) 2024 Emeric Poupon
+ *
+ * This file is part of LMS.
+ *
+ * LMS is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * LMS is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with LMS.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#pragma once
+
+#include "database/IdType.hpp"
+
+LMS_DECLARE_IDTYPE(CountryId)

src/libs/database/include/database/Release.hpp

@@ -33,6 +33,7 @@
 #include "core/UUID.hpp"
 #include "database/ArtistId.hpp"
 #include "database/ClusterId.hpp"
+#include "database/CountryId.hpp"
 #include "database/DirectoryId.hpp"
 #include "database/LabelId.hpp"
 #include "database/MediaLibraryId.hpp"
@@ -53,6 +54,37 @@ namespace lms::db
     class Track;
     class User;
 
+    class Country final : public Object<Country, CountryId>
+    {
+    public:
+        Country() = default;
+
+        static std::size_t getCount(Session& session);
+        static pointer find(Session& session, CountryId id);
+        static pointer find(Session& session, std::string_view name);
+        static RangeResults<CountryId> findOrphanIds(Session& session, std::optional<Range> range = std::nullopt);
+
+        // Accessors
+        std::string_view getName() const { return _name; }
+
+        template<class Action>
+        void persist(Action& a)
+        {
+            Wt::Dbo::field(a, _name, "name");
+            Wt::Dbo::hasMany(a, _releases, Wt::Dbo::ManyToMany, "release_country", "", Wt::Dbo::OnDeleteCascade);
+        }
+
+    private:
+        static constexpr std::size_t _maxNameLength{ 32 };
+
+        friend class Session;
+        Country(std::string_view name);
+        static pointer create(Session& session, std::string_view name);
+
+        std::string _name;
+        Wt::Dbo::collection<Wt::Dbo::ptr<Release>> _releases; // releases that match this country
+    };
+
     class Label final : public Object<Label, LabelId>
     {
     public:
@@ -251,6 +283,7 @@ namespace lms::db
         std::size_t getTrackCount() const;
         std::vector<ObjectPtr<ReleaseType>> getReleaseTypes() const;
         std::vector<std::string> getLabelNames() const;
+        std::vector<std::string> getCountryNames() const;
         std::vector<std::string> getReleaseTypeNames() const;
         void visitLabels(const std::function<void(const Label::pointer& label)>& _func) const;
         core::EnumSet<Advisory> getAdvisories() const;
@@ -267,8 +300,10 @@ namespace lms::db
         void setArtistDisplayName(std::string_view name) { _artistDisplayName = name; }
         void setCompilation(bool value) { _isCompilation = value; }
         void clearLabels();
+        void clearCountries();
         void clearReleaseTypes();
-        void addLabel(ObjectPtr<Label> releaseType);
+        void addLabel(ObjectPtr<Label> label);
+        void addCountry(ObjectPtr<Country> country);
         void addReleaseType(ObjectPtr<ReleaseType> releaseType);
         void setBarcode(std::string_view barcode) { _barcode = barcode; }
         void setComment(std::string_view comment) { _comment = comment; }
@@ -299,6 +334,7 @@ namespace lms::db
             Wt::Dbo::belongsTo(a, _image, "image", Wt::Dbo::OnDeleteSetNull);
             Wt::Dbo::hasMany(a, _labels, Wt::Dbo::ManyToMany, "release_label", "", Wt::Dbo::OnDeleteCascade);
             Wt::Dbo::hasMany(a, _releaseTypes, Wt::Dbo::ManyToMany, "release_release_type", "", Wt::Dbo::OnDeleteCascade);
+            Wt::Dbo::hasMany(a, _countries, Wt::Dbo::ManyToMany, "release_country", "", Wt::Dbo::OnDeleteCascade);
         }
 
     private:
@@ -325,6 +361,7 @@ namespace lms::db
         Wt::Dbo::collection<Wt::Dbo::ptr<Track>> _tracks;
         Wt::Dbo::collection<Wt::Dbo::ptr<Label>> _labels;
         Wt::Dbo::collection<Wt::Dbo::ptr<ReleaseType>> _releaseTypes;
+        Wt::Dbo::collection<Wt::Dbo::ptr<Country>> _countries;
     };
 
 } // namespace lms::db

src/libs/database/test/Migration.cpp

@@ -340,6 +340,7 @@ VALUES
 
             EXPECT_FALSE(Artist::find(session, ArtistId{}));
             EXPECT_FALSE(AuthToken::find(session, AuthTokenId{}));
+            EXPECT_FALSE(Country::find(session, CountryId{}));
             EXPECT_FALSE(Cluster::find(session, ClusterId{}));
             EXPECT_FALSE(ClusterType::find(session, ClusterTypeId{}));
             EXPECT_FALSE(Directory::find(session, DirectoryId{}));

src/libs/database/test/Release.cpp

@@ -26,6 +26,7 @@ namespace lms::db::tests
 {
     using ScopedImage = ScopedEntity<db::Image>;
     using ScopedLabel = ScopedEntity<db::Label>;
+    using ScopedCountry = ScopedEntity<db::Country>;
     using ScopedReleaseType = ScopedEntity<db::ReleaseType>;
 
     TEST_F(DatabaseFixture, Release)
@@ -808,6 +809,30 @@ namespace lms::db::tests
         }
     }
 
+    TEST_F(DatabaseFixture, Release_getLabelNames)
+    {
+        ScopedRelease release{ session, "MyRelease" };
+        ScopedLabel label{ session, "MyLabel" };
+
+        {
+            auto transaction{ session.createReadTransaction() };
+            const auto names{ release.get()->getLabelNames() };
+            EXPECT_EQ(names.size(), 0);
+        }
+
+        {
+            auto transaction{ session.createWriteTransaction() };
+            release.get().modify()->addLabel(label.get());
+        }
+
+        {
+            auto transaction{ session.createReadTransaction() };
+            const auto names{ release.get()->getLabelNames() };
+            ASSERT_EQ(names.size(), 1);
+            EXPECT_EQ(names[0], "MyLabel");
+        }
+    }
+
     TEST_F(DatabaseFixture, Label_orphan)
     {
         ScopedLabel label{ session, "MyLabel" };
@@ -845,6 +870,78 @@ namespace lms::db::tests
         }
     }
 
+    TEST_F(DatabaseFixture, Country)
+    {
+        {
+            auto transaction{ session.createReadTransaction() };
+            Country::pointer res{ Country::find(session, "country") };
+            EXPECT_EQ(res, Country::pointer{});
+        }
+
+        ScopedCountry country{ session, "MyCountry" };
+
+        {
+            auto transaction{ session.createReadTransaction() };
+            Country::pointer res{ Country::find(session, "MyCountry") };
+            EXPECT_EQ(res, country.get());
+        }
+    }
+
+    TEST_F(DatabaseFixture, Release_getCountryNames)
+    {
+        ScopedCountry country{ session, "MyCountry" };
+        ScopedRelease release{ session, "MyRelease" };
+
+        {
+            auto transaction{ session.createWriteTransaction() };
+            release.get().modify()->addCountry(country.get());
+        }
+
+        {
+            auto transaction{ session.createReadTransaction() };
+            const auto names{ release.get()->getCountryNames() };
+            ASSERT_EQ(names.size(), 1);
+            EXPECT_EQ(names[0], "MyCountry");
+        }
+    }
+
+    TEST_F(DatabaseFixture, Country_orphan)
+    {
+        ScopedCountry country{ session, "MyCountry" };
+
+        {
+            auto transaction{ session.createReadTransaction() };
+            auto countries{ Country::findOrphanIds(session) };
+            ASSERT_EQ(countries.results.size(), 1);
+            EXPECT_EQ(countries.results.front(), country.getId());
+        }
+
+        ScopedRelease release{ session, "MyRelease" };
+
+        {
+            auto transaction{ session.createWriteTransaction() };
+            release.get().modify()->addCountry(country.get());
+        }
+
+        {
+            auto transaction{ session.createReadTransaction() };
+            auto countries{ Country::findOrphanIds(session) };
+            EXPECT_EQ(countries.results.size(), 0);
+        }
+
+        {
+            auto transaction{ session.createWriteTransaction() };
+            release.get().modify()->clearCountries();
+        }
+
+        {
+            auto transaction{ session.createReadTransaction() };
+            auto countries{ Country::findOrphanIds(session) };
+            ASSERT_EQ(countries.results.size(), 1);
+            EXPECT_EQ(countries.results.front(), country.getId());
+        }
+    }
+
     TEST_F(DatabaseFixture, ReleaseType)
     {
         {