308 lines
9.3 KiB
C++
308 lines
9.3 KiB
C++
/*
|
|
* This program source code file is part of KiCad, a free EDA CAD application.
|
|
*
|
|
* Copyright (C) 2012 Brian Sidebotham <brian.sidebotham@gmail.com>
|
|
* Copyright (C) 1992-2020 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 2
|
|
* 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, you may find one here:
|
|
* http://www.gnu.org/licenses/old-licenses/gpl-2.0.html
|
|
* or you may search the http://www.gnu.org website for the version 2 license,
|
|
* or you may write to the Free Software Foundation, Inc.,
|
|
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
|
|
*/
|
|
|
|
|
|
#include <wx/bitmap.h>
|
|
#include <wx/dir.h>
|
|
#include <wx/txtstrm.h>
|
|
#include <wx/wfstream.h>
|
|
#include <wx/log.h>
|
|
|
|
#include <wildcards_and_files_ext.h>
|
|
#include "project_template.h"
|
|
|
|
|
|
#define SEP wxFileName::GetPathSeparator()
|
|
|
|
|
|
PROJECT_TEMPLATE::PROJECT_TEMPLATE( const wxString& aPath )
|
|
{
|
|
m_basePath = wxFileName::DirName( aPath );
|
|
m_metaPath = wxFileName::DirName( aPath + SEP + METADIR );
|
|
m_metaHtmlFile = wxFileName::FileName( aPath + SEP + METADIR + SEP + METAFILE_INFO_HTML );
|
|
m_metaIconFile = wxFileName::FileName( aPath + SEP + METADIR + SEP + METAFILE_ICON );
|
|
|
|
m_title = wxEmptyString;
|
|
|
|
// Test the project template requirements to make sure aPath is a valid template structure.
|
|
if( !wxFileName::DirExists( m_basePath.GetPath() ) )
|
|
{
|
|
// Error, the path doesn't exist!
|
|
m_title = _( "Could open the template path!" ) + wxS( " " ) + aPath;
|
|
}
|
|
else if( !wxFileName::DirExists( m_metaPath.GetPath() ) )
|
|
{
|
|
// Error, the meta information directory doesn't exist!
|
|
m_title = _( "Couldn't open the meta information directory for this template!" ) + wxS( " " ) +
|
|
m_metaPath.GetPath();
|
|
}
|
|
else if( !wxFileName::FileExists( m_metaHtmlFile.GetFullPath() ) )
|
|
{
|
|
// Error, the meta information directory doesn't contain the informational html file!
|
|
m_title = _( "Cound't find the meta HTML information file for this template!" );
|
|
}
|
|
|
|
// Try to load an icon
|
|
if( !wxFileName::FileExists( m_metaIconFile.GetFullPath() ) )
|
|
m_metaIcon = &wxNullBitmap;
|
|
else
|
|
m_metaIcon = new wxBitmap( m_metaIconFile.GetFullPath(), wxBITMAP_TYPE_PNG );
|
|
}
|
|
|
|
|
|
std::vector<wxFileName> PROJECT_TEMPLATE::GetFileList()
|
|
{
|
|
std::vector<wxFileName> files;
|
|
wxString f = m_basePath.GetPath();
|
|
wxArrayString allfiles;
|
|
wxFileName p;
|
|
|
|
wxDir::GetAllFiles( f, &allfiles );
|
|
|
|
// Create the vector and ignore all of the meta data files!
|
|
for( size_t i=0; i < allfiles.size(); i++ )
|
|
{
|
|
p = allfiles[i];
|
|
|
|
// Files that are in the meta directory must not be included
|
|
if( !p.GetPath().StartsWith( m_metaPath.GetPath() ) )
|
|
files.emplace_back(allfiles[i] );
|
|
}
|
|
|
|
return files;
|
|
}
|
|
|
|
|
|
wxString PROJECT_TEMPLATE::GetPrjDirName()
|
|
{
|
|
return m_basePath.GetDirs()[ m_basePath.GetDirCount() - 1 ];
|
|
}
|
|
|
|
|
|
PROJECT_TEMPLATE::~PROJECT_TEMPLATE()
|
|
{
|
|
|
|
}
|
|
|
|
|
|
wxFileName PROJECT_TEMPLATE::GetHtmlFile()
|
|
{
|
|
return m_metaHtmlFile;
|
|
}
|
|
|
|
|
|
wxBitmap* PROJECT_TEMPLATE::GetIcon()
|
|
{
|
|
return m_metaIcon;
|
|
}
|
|
|
|
|
|
size_t PROJECT_TEMPLATE::GetDestinationFiles( const wxFileName& aNewProjectPath,
|
|
std::vector< wxFileName >& aDestFiles )
|
|
{
|
|
std::vector< wxFileName > srcFiles = GetFileList();
|
|
|
|
// Find the template file name base. this is the name of the .pro template file
|
|
wxString basename;
|
|
int projectCount = 0;
|
|
|
|
for( wxFileName& file : srcFiles )
|
|
{
|
|
if( file.GetExt() == ProjectFileExtension || file.GetExt() == LegacyProjectFileExtension )
|
|
{
|
|
basename = file.GetName();
|
|
projectCount++;
|
|
}
|
|
}
|
|
|
|
if( projectCount != 1 )
|
|
basename = GetPrjDirName();
|
|
|
|
for( wxFileName& srcFile : srcFiles )
|
|
{
|
|
// Replace the template path
|
|
wxFileName destFile = srcFile;
|
|
|
|
// Replace the template filename with the project filename for the new project creation
|
|
wxString name = destFile.GetName();
|
|
name.Replace( basename, aNewProjectPath.GetName() );
|
|
destFile.SetName( name );
|
|
|
|
// Replace the template path with the project path.
|
|
wxString path = destFile.GetPathWithSep();
|
|
path.Replace( m_basePath.GetPathWithSep(), aNewProjectPath.GetPathWithSep() );
|
|
destFile.SetPath( path );
|
|
|
|
aDestFiles.push_back( destFile );
|
|
}
|
|
|
|
return aDestFiles.size();
|
|
}
|
|
|
|
|
|
bool PROJECT_TEMPLATE::CreateProject( wxFileName& aNewProjectPath, wxString* aErrorMsg )
|
|
{
|
|
// CreateProject copy the files from template to the new project folder
|
|
// and rename files which have the same name as the template .pro file
|
|
bool result = true;
|
|
|
|
std::vector<wxFileName> srcFiles = GetFileList();
|
|
|
|
// Find the template file name base. this is the name of the .pro template file
|
|
wxString basename;
|
|
int projectCount = 0;
|
|
|
|
for( wxFileName& file : srcFiles )
|
|
{
|
|
if( file.GetExt() == ProjectFileExtension || file.GetExt() == LegacyProjectFileExtension )
|
|
{
|
|
basename = file.GetName();
|
|
projectCount++;
|
|
}
|
|
}
|
|
|
|
if( projectCount != 1 )
|
|
basename = GetPrjDirName();
|
|
|
|
for( wxFileName& srcFile : srcFiles )
|
|
{
|
|
// Replace the template path
|
|
wxFileName destFile = srcFile;
|
|
|
|
// Replace the template filename with the project filename for the new project creation
|
|
wxString currname = destFile.GetName();
|
|
|
|
// Do not rename project specific symbol libraries. This will break the symbol library
|
|
// table which will cause broken symbol library links in the schematic.
|
|
if( !( destFile.GetExt() == "dcm"
|
|
|| ( destFile.GetExt() == "lib" && !destFile.GetName().EndsWith( "-cache" ) ) ) )
|
|
{
|
|
currname.Replace( basename, aNewProjectPath.GetName() );
|
|
}
|
|
|
|
destFile.SetName( currname );
|
|
|
|
// Replace the template path with the project path for the new project creation
|
|
// but keep the sub directory name, if exists
|
|
wxString destpath = destFile.GetPathWithSep();
|
|
destpath.Replace( m_basePath.GetPathWithSep(), aNewProjectPath.GetPathWithSep() );
|
|
|
|
// Check to see if the path already exists, if not attempt to create it here. Don't worry
|
|
// about error checking, if the path isn't created the file copy will fail anyway
|
|
if( !wxFileName::DirExists( destpath ) )
|
|
{
|
|
if( !wxFileName::Mkdir( destpath, 0777, wxPATH_MKDIR_FULL ) )
|
|
{
|
|
if( aErrorMsg )
|
|
{
|
|
if( !aErrorMsg->empty() )
|
|
*aErrorMsg += "\n";
|
|
|
|
wxString msg;
|
|
|
|
msg.Printf( _( "Cannot create folder \"%s\"." ), destpath );
|
|
*aErrorMsg += msg;
|
|
}
|
|
|
|
continue;
|
|
}
|
|
}
|
|
|
|
destFile.SetPath( destpath );
|
|
|
|
if( !wxCopyFile( srcFile.GetFullPath(), destFile.GetFullPath() ) )
|
|
{
|
|
if( aErrorMsg )
|
|
{
|
|
if( !aErrorMsg->empty() )
|
|
*aErrorMsg += "\n";
|
|
|
|
wxString msg;
|
|
|
|
msg.Printf( _( "Cannot copy file \"%s\"." ), destFile.GetFullPath() );
|
|
*aErrorMsg += msg;
|
|
}
|
|
|
|
result = false;
|
|
}
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
|
|
wxString* PROJECT_TEMPLATE::GetTitle()
|
|
{
|
|
wxFileInputStream input( GetHtmlFile().GetFullPath() );
|
|
wxString separator( wxT( "\x9" ) );
|
|
wxTextInputStream text( input, separator, wxConvUTF8 );
|
|
|
|
/* Open HTML file and get the text between the title tags */
|
|
if( m_title == wxEmptyString )
|
|
{
|
|
int start = 0;
|
|
int finish = 0;
|
|
bool done = false;
|
|
|
|
while( input.IsOk() && !input.Eof() && !done )
|
|
{
|
|
wxString line = text.ReadLine();
|
|
wxString upperline = line.Clone().Upper();
|
|
|
|
start = upperline.Find( wxT( "<TITLE>" ) );
|
|
finish = upperline.Find( wxT( "</TITLE>" ) );
|
|
int length = finish - start - 7;
|
|
|
|
// find the opening tag
|
|
if( start != wxNOT_FOUND )
|
|
{
|
|
if( finish != wxNOT_FOUND )
|
|
{
|
|
m_title = line( start + 7, length );
|
|
}
|
|
else
|
|
{
|
|
m_title = line.Mid( start + 7 );
|
|
}
|
|
|
|
done = true;
|
|
}
|
|
else
|
|
{
|
|
if( finish != wxNOT_FOUND )
|
|
{
|
|
m_title += line.SubString( 0, finish );
|
|
done = true;
|
|
}
|
|
}
|
|
|
|
// Remove line endings
|
|
m_title.Replace( wxT( "\r" ), wxT( " " ) );
|
|
m_title.Replace( wxT( "\n" ), wxT( " " ) );
|
|
}
|
|
}
|
|
|
|
return &m_title;
|
|
}
|