pcm.h

Go to the documentation of this file.

/*
 * This program source code file is part of KiCad, a free EDA CAD application.
 *
 * Copyright (C) 2021 Andrew Lutsenko, anlutsenko at gmail dot com
 * Copyright (C) 1992-2021 KiCad Developers, see AUTHORS.txt for contributors.
 *
 * This program 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.
 *
 * 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
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program. If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef PCM_H_
#define PCM_H_

#include "core/wx_stl_compat.h"
#include "pcm_data.h"
#include "widgets/wx_progress_reporters.h"
#include <iostream>
#include <map>
#include <nlohmann/json-schema.hpp>
#include <tuple>
#include <unordered_map>
#include <unordered_set>
#include <vector>
#include <wx/wx.h>


const std::unordered_set<wxString> PCM_PACKAGE_DIRECTORIES( {
 wxT( "plugins" ),
 wxT( "footprints" ),
 wxT( "3dmodels" ),
 wxT( "symbols" ),
 wxT( "resources" ),
 wxT( "colors" ),
} );


enum PCM_PACKAGE_STATE
{
 PPS_AVAILABLE = 0,
 PPS_UNAVAILABLE = 1,
 PPS_INSTALLED = 2,
 PPS_PENDING_INSTALL = 3,
 PPS_PENDING_UNINSTALL = 4
};


enum PCM_PACKAGE_ACTION
{
 PPA_INSTALL = 0,
 PPA_UNINSTALL = 1
};


typedef std::vector<std::pair<wxString, wxString>> STRING_PAIR_LIST;
typedef std::vector<std::tuple<wxString, wxString, wxString>> STRING_TUPLE_LIST;


class PLUGIN_CONTENT_MANAGER
{
public:
 PLUGIN_CONTENT_MANAGER( wxWindow* aParent );
 ~PLUGIN_CONTENT_MANAGER();

 bool FetchRepository( const wxString& aUrl, PCM_REPOSITORY& aRepository,
 WX_PROGRESS_REPORTER* aReporter );

 void ValidateJson( const nlohmann::json& aJson,
 const nlohmann::json_uri& aUri = nlohmann::json_uri( "#" ) ) const;

 bool VerifyHash( std::istream& aStream, const wxString& aHash ) const;

 void SetRepositoryList( const STRING_PAIR_LIST& aRepositories );

 void DiscardRepositoryCache( const wxString& aRepositoryId );

 void MarkInstalled( const PCM_PACKAGE& aPackage, const wxString& aVersion,
 const wxString& aRepositoryId );

 void MarkUninstalled( const PCM_PACKAGE& aPackage );

 const STRING_TUPLE_LIST& GetRepositoryList() const { return m_repository_list; }

 const bool CacheRepository( const wxString& aRepositoryId );

 const std::vector<PCM_PACKAGE>& GetRepositoryPackages( const wxString& aRepositoryId ) const;

 const std::vector<PCM_INSTALLATION_ENTRY> GetInstalledPackages() const;

 const wxString& GetInstalledPackageVersion( const wxString& aPackageId ) const;

 const wxString& Get3rdPartyPath() const { return m_3rdparty_path; };

 PCM_PACKAGE_STATE GetPackageState( const wxString& aRepositoryId, const wxString& aPackageId );

 bool DownloadToStream( const wxString& aUrl, std::ostream* aOutput,
 WX_PROGRESS_REPORTER* aReporter,
  const size_t aSizeLimit = DEFAULT_DOWNLOAD_MEM_LIMIT );

 int GetPackageSearchRank( const PCM_PACKAGE& aPackage, const wxString& aSearchTerm );

 std::unordered_map<wxString, wxBitmap>
 GetRepositoryPackageBitmaps( const wxString& aRepositoryId );

 std::unordered_map<wxString, wxBitmap> GetInstalledPackageBitmaps();

private:
 static constexpr size_t DEFAULT_DOWNLOAD_MEM_LIMIT = 10 * 1024 * 1024;

 bool fetchPackages( const wxString& aUrl, const boost::optional<wxString>& aHash,
 std::vector<PCM_PACKAGE>& aPackages, WX_PROGRESS_REPORTER* aReporter );

 const PCM_REPOSITORY& getCachedRepository( const wxString& aRepositoryId ) const;

 static void preparePackage( PCM_PACKAGE& aPackage );

 time_t getCurrentTimestamp() const;

 wxWindow* m_dialog;
 nlohmann::json_schema::json_validator m_schema_validator;
 wxString m_3rdparty_path;
 wxString m_cache_path;
 std::unordered_map<wxString, PCM_REPOSITORY> m_repository_cache;
 STRING_TUPLE_LIST m_repository_list; // (id, name, url) tuples
 // Using sorted map to keep order of entries in installed list stable
 std::map<wxString, PCM_INSTALLATION_ENTRY> m_installed;
 const static std::tuple<int, int> m_kicad_version;
};

#endif // PCM_H_