/*****************************************************************************
 * Media Library
 *****************************************************************************
 * Copyright (C) 2015-2019 Hugo Beauzée-Luyssen, Videolabs, VideoLAN
 *
 * Authors: Hugo Beauzée-Luyssen <hugo@beauzee.fr>
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation; either version 2.1 of the License, or
 * (at your option) any later version.
 *
 * This program 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 Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
 *****************************************************************************/

#if HAVE_CONFIG_H
# include "config.h"
#endif

#include "Tester.h"
#include "medialibrary/filesystem/IDirectory.h"
#include "medialibrary/filesystem/IFile.h"
#include "utils/Filename.h"
#include "utils/Directory.h"
#include "Playlist.h"
#include "Media.h"
#include "File.h"
#include "Subscription.h"

#include <vlcpp/vlc.hpp>

static void Parse( Tests* T )
{
    T->m_cb->waitForParsingComplete();

    T->runChecks();
}

static void ParseTwice( Tests* T )
{
    T->m_cb->waitForParsingComplete();

    T->runChecks();

    if ( T->input.IsArray() )
    {
        T->m_cb->prepareForRemoval( T->input.Size() );
        for ( auto i = 0u; i < T->input.Size(); ++i )
        {
            auto samplesDir = Tests::Directory + "samples/" + T->input[i].GetString();
            samplesDir = utils::fs::toAbsolute( samplesDir );
            T->m_ml->removeRoot( utils::file::toMrl( samplesDir ) );
        }
        ASSERT_TRUE( T->m_cb->waitForRemovalComplete() );
        T->m_cb->reinit();
        for ( auto i = 0u; i < T->input.Size(); ++i )
        {
            auto samplesDir = Tests::Directory + "samples/" + T->input[i].GetString();
            samplesDir = utils::fs::toAbsolute( samplesDir );
            T->m_ml->discover( utils::file::toMrl( samplesDir ) );
        }
    }
    else

    if ( T->subscriptions.IsArray() )
    {
        auto service = T->m_ml->service( IService::Type::Podcast );
        auto subscriptions = service->subscriptions( nullptr )->all();
        for ( const auto& c : subscriptions )
            T->m_ml->removeSubscription( c->id() );
        /* Ensure we don't wait for a discovery that will not come */
        T->m_cb->onDiscoveryCompleted();
        for ( auto i = 0u; i < T->subscriptions.Size(); ++i )
        {
            auto& sub = T->subscriptions[i];
            ASSERT_TRUE( sub.HasMember( "service" ) && sub.HasMember( "mrl" ) );
            T->addSubscription( static_cast<IService::Type>( sub["service"].GetUint() ),
                    sub["mrl"].GetString() );
        }
    }

    T->m_cb->waitForParsingComplete();

    T->runChecks();
}

static void RunResumeTests( ResumeTests* T )
{
    T->m_cb->waitForDiscoveryComplete();
    auto testMl = static_cast<MediaLibraryResumeTest*>( T->m_ml.get() );
    testMl->forceParserStart();
    T->m_cb->waitForParsingComplete();

    T->runChecks();
}

static void Rescan( ResumeTests* T )
{
    T->m_cb->waitForDiscoveryComplete();
    auto testMl = static_cast<MediaLibraryResumeTest*>( T->m_ml.get() );
    testMl->forceParserStart();
    T->m_cb->waitForParsingComplete();

    T->m_cb->reinit();
    T->m_ml->forceRescan();
    T->m_cb->waitForParsingComplete();

    T->runChecks();
}

static void RunRefreshTests( RefreshTests* T )
{
    T->m_cb->waitForDiscoveryComplete();
    T->m_cb->waitForParsingComplete();

    T->runChecks();

    T->m_cb->reinit();
    T->forceRefresh();

    T->m_cb->waitForParsingComplete();

    T->runChecks();
}

static void ReplaceVlcInstance( Tests* T )
{
    VLC::Instance inst{ 0, nullptr };
    T->m_ml->setExternalLibvlcInstance( inst.get() );
    /* Replacing the instance will stop the discoverer so let's resume it */
    T->m_ml->reload();
    T->m_cb->waitForParsingComplete();

    T->runChecks();
}

static void RunBackupRestorePlaylist( BackupRestorePlaylistTests* T )
{
    auto samplesFolder = std::string{ SRC_DIR "/test/samples/samples/playlist/tracks" };
    ASSERT_TRUE( utils::fs::isDirectory( samplesFolder ) );
    samplesFolder = utils::fs::toAbsolute( samplesFolder );
    T->m_ml->discover( utils::file::toMrl( samplesFolder ) );
    T->m_cb->waitForParsingComplete();
    // Now we should have discovered some media

    auto media = T->m_ml->audioFiles( nullptr )->all();
    ASSERT_EQ( 3u, media.size() );

    auto pl1 = std::static_pointer_cast<Playlist>( T->m_ml->createPlaylist( "Exported Playlist 1" ) );
    auto m1 = media[0];
    auto m2 = media[1];
    auto m3 = T->m_ml->addExternalMedia( "http://example.org/sea&ottér.avi", -1 );
    pl1->append( *m1 );
    pl1->append( *m2 );
    pl1->append( *m3 );

    auto pl2 = std::static_pointer_cast<Playlist>( T->m_ml->createPlaylist( "Exported Playlist <2>" ) );
    pl2->append( *m3 );
    pl2->append( *m2 );
    pl2->append( *m1 );

    auto backup = Playlist::backupPlaylists( static_cast<MediaLibrary*>( T->m_ml.get() ),
                                             Settings::DbModelVersion );
    ASSERT_TRUE( std::get<0>( backup ) );

    T->m_cb->prepareForPlaylistReload();
    T->m_ml->clearDatabase( true );

    T->m_cb->waitForPlaylistReload();

    auto playlists = T->m_ml->playlists( PlaylistType::All, nullptr )->all();
    ASSERT_EQ( 2u, playlists.size() );
    auto playlist1 = playlists[0];
    media = playlist1->media( nullptr )->all();
    ASSERT_EQ( 3u, media.size() );
    ASSERT_EQ( "Exported Playlist 1", playlist1->name() );

    auto playlist2 = playlists[1];
    media = playlist2->media( nullptr )->all();
    ASSERT_EQ( 3u, media.size() );
    ASSERT_EQ( "Exported Playlist <2>", playlist2->name() );

    /*
     * Since the folder isn't discovered yet, the media won't be preparsed and
     * won't have their duration or title.
     * However if we discover those media again, the media should be analyzed and
     * converted back to internal media, meaning they'll recover their titles
     * and duration among other information.
     */
    T->m_cb->reinit();
    T->m_ml->discover( utils::file::toMrl( samplesFolder ) );
    T->m_cb->waitForParsingComplete();
    media = playlist1->media( nullptr )->all();
    ASSERT_EQ( m1->title(), media[0]->title() );
    ASSERT_EQ( m2->title(), media[1]->title() );
    ASSERT_EQ( m3->title(), media[2]->title() );

    media = playlist2->media( nullptr )->all();
    ASSERT_EQ( m3->title(), media[0]->title() );
    ASSERT_EQ( m2->title(), media[1]->title() );
    ASSERT_EQ( m1->title(), media[2]->title() );
}

static void ReplaceExternalMediaByPlaylist( ReplaceExternalMediaByPlaylistTests* T )
{
    /*
     * This test was initialized with a playlist inserted as an external media.
     * From there, just run the same test as the ParseTwice suite to check if we
     * can recover properly and insert the playlist as an actual playlist
     */
    ParseTwice( T );
    /*
     * Initially the playlist was inserted with the same MRL as the media, ensure
     * this is not the case anymore
     */
    auto mlPtr = static_cast<MediaLibrary*>( T->m_ml.get() );
    auto file = File::fromMrl( mlPtr, T->playlistMrl );
    ASSERT_NON_NULL( file );
    ASSERT_EQ( IFile::Type::Playlist, file->type() );
    auto res = file->destroy();
    ASSERT_TRUE( res );
    /* Ensure there's no other file with the same MRL */
    file = File::fromMrl( mlPtr, T->playlistMrl );
    ASSERT_EQ( nullptr, file );
    file = File::fromExternalMrl( mlPtr, T->playlistMrl );
    ASSERT_EQ( nullptr, file );
}

#define RUN_TEST( Type, Func ) \
    try \
    { \
        auto T = std::make_unique<Type>(); \
        T->SetUp( testType, testName ); \
        Func( T.get() ); \
        T->TearDown(); \
        return 0; \
    } catch ( const TestFailed& tf ) { \
        fprintf( stderr, "Test %s.%s failed: %s\n", testType.c_str(), \
                 testName.c_str(), tf.what() ); \
        return 2; \
    }

int main(int ac, char** av)
{
    std::string testType;
    std::string testName;
    if ( ac != 3 )
    {
        fprintf( stderr, "usage: %s <test type> <test name>\n", av[0] );
        return 1;
    }
    testType = av[1];
    testName = av[2];
    if ( testType == "Parse" )
    {
        RUN_TEST( Tests, Parse );
    }
    else if ( testType == "ParseTwice" )
    {
        RUN_TEST( Tests, ParseTwice );
    }
    else if ( testType == "Resume" )
    {
        RUN_TEST( ResumeTests, RunResumeTests );
    }
    else if ( testType == "Rescan" )
    {
        RUN_TEST( ResumeTests, Rescan );
    }
    else if ( testType == "Refresh" )
    {
        RUN_TEST( RefreshTests, RunRefreshTests );
    }
    else if ( testType == "ReplaceVlcInstance" )
    {
        RUN_TEST( Tests, ReplaceVlcInstance );
    }
    else if ( testType == "BackupRestorePlaylist" )
    {
        RUN_TEST( BackupRestorePlaylistTests, RunBackupRestorePlaylist );
    }
    else if ( testType == "ReplaceExternalMediaByPlaylist" )
    {
        RUN_TEST( ReplaceExternalMediaByPlaylistTests, ReplaceExternalMediaByPlaylist );
    }
    else
    {
        assert( !"Invalid test type" );
        return 1;
    }
    return 1;
}