Kicad: unzip project: fix issues:

- files with no extension were skipped.
- subdirs were ignored.
This commit is contained in:
jean-pierre charras 2020-03-21 14:02:54 +01:00
parent 103b0c18de
commit e0e4ea6223
1 changed files with 93 additions and 39 deletions

View File

@ -61,6 +61,44 @@ void KICAD_MANAGER_FRAME::OnClearFileHistory( wxCommandEvent& aEvent )
} }
bool CopyStreamData( wxInputStream& inputStream, wxOutputStream& outputStream, wxFileOffset size )
{
wxChar buf[128 * 1024];
int readSize = 128 * 1024;
wxFileOffset copiedData = 0;
for (;;)
{
if (size != -1 && copiedData + readSize > size)
readSize = size - copiedData;
inputStream.Read( buf, readSize );
size_t actuallyRead = inputStream.LastRead();
outputStream.Write( buf, actuallyRead );
if( outputStream.LastWrite() != actuallyRead )
{
wxLogError( "Failed to output data" );
//return false;
}
if (size == -1)
{
if (inputStream.Eof())
break;
}
else
{
copiedData += actuallyRead;
if( copiedData >= size )
break;
}
}
return true;
}
void KICAD_MANAGER_FRAME::OnUnarchiveFiles( wxCommandEvent& event ) void KICAD_MANAGER_FRAME::OnUnarchiveFiles( wxCommandEvent& event )
{ {
wxFileName fn = GetProjectFileName(); wxFileName fn = GetProjectFileName();
@ -74,7 +112,7 @@ void KICAD_MANAGER_FRAME::OnUnarchiveFiles( wxCommandEvent& event )
if( zipfiledlg.ShowModal() == wxID_CANCEL ) if( zipfiledlg.ShowModal() == wxID_CANCEL )
return; return;
wxString msg = wxString::Format( _( "\nOpen \"%s\"\n" ), GetChars( zipfiledlg.GetPath() ) ); wxString msg = wxString::Format( _( "\nOpen \"%s\"\n" ), zipfiledlg.GetPath() );
PrintMsg( msg ); PrintMsg( msg );
wxDirDialog dirDlg( this, _( "Target Directory" ), fn.GetPath(), wxDirDialog dirDlg( this, _( "Target Directory" ), fn.GetPath(),
@ -84,49 +122,65 @@ void KICAD_MANAGER_FRAME::OnUnarchiveFiles( wxCommandEvent& event )
return; return;
wxString unzipDir = dirDlg.GetPath() + wxT( "/" ); wxString unzipDir = dirDlg.GetPath() + wxT( "/" );
msg.Printf( _( "Unzipping project in \"%s\"\n" ), GetChars( unzipDir ) ); msg.Printf( _( "Unzipping project in \"%s\"\n" ), unzipDir );
PrintMsg( msg ); PrintMsg( msg );
wxFileSystem zipfilesys; wxString archiveFileName = zipfiledlg.GetPath();
wxFileInputStream fileInputStream( archiveFileName );
zipfilesys.AddHandler( new wxZipFSHandler ); if (!fileInputStream.IsOk())
auto path = wxURI( zipfiledlg.GetPath() + wxT( "#zip:" ) ).BuildURI(); return;
zipfilesys.ChangePathTo( path, true );
wxFSFile* zipfile = NULL; const wxArchiveClassFactory* archiveClassFactory =
wxString localfilename = zipfilesys.FindFirst( wxFileSelectorDefaultWildcardStr, wxFILE ); wxArchiveClassFactory::Find( archiveFileName, wxSTREAM_FILEEXT );
while( !localfilename.IsEmpty() ) if( archiveClassFactory )
{ {
zipfile = zipfilesys.OpenFile( localfilename ); wxScopedPtr<wxArchiveInputStream> archiveStream( archiveClassFactory->NewStream(fileInputStream) );
if( !zipfile )
for (wxArchiveEntry* entry = archiveStream->GetNextEntry(); entry;
entry = archiveStream->GetNextEntry())
{ {
DisplayError( this, wxT( "Zip file read error" ) ); PrintMsg( wxString::Format( _( "Extract file \"%s\"" ), entry->GetName() ) );
break;
wxString fullname = unzipDir + entry->GetName();
// Ensure the target directory exists and created it if not
wxString t_path = wxPathOnly( fullname );
if( !wxDirExists( t_path ) )
{
// To create t_path, we need to create all subdirs from unzipDir
// to t_path.
wxFileName pathToCreate;
pathToCreate.AssignDir( t_path );
pathToCreate.MakeRelativeTo( unzipDir );
// Create the list of subdirs candidates
wxArrayString subDirs;
subDirs = pathToCreate.GetDirs();
pathToCreate.AssignDir( unzipDir );
for( size_t ii = 0; ii < subDirs.Count(); ii++ )
{
pathToCreate.AppendDir( subDirs[ii] );
wxString currPath = pathToCreate.GetPath();
if( !wxDirExists( currPath ) )
wxMkdir( currPath );
}
} }
wxFileName uzfn = localfilename.AfterLast( ':' ); wxTempFileOutputStream outputFileStream( fullname );
uzfn.MakeAbsolute( unzipDir );
wxString unzipfilename = uzfn.GetFullPath();
msg.Printf( _( "Extract file \"%s\"" ), GetChars( unzipfilename ) ); if( CopyStreamData( *archiveStream, outputFileStream, entry->GetSize() ) )
PrintMsg( msg );
wxInputStream* stream = zipfile->GetStream();
wxFFileOutputStream* ofile = new wxFFileOutputStream( unzipfilename );
if( ofile->Ok() )
{ {
ofile->Write( *stream ); outputFileStream.Commit();
PrintMsg( _( " OK\n" ) ); PrintMsg( _( " OK\n" ) );
} }
else else
PrintMsg( _( " *ERROR*\n" ) ); PrintMsg( _( " *ERROR*\n" ) );
}
delete ofile;
delete zipfile;
localfilename = zipfilesys.FindNext();
} }
PrintMsg( wxT( "** end **\n" ) ); PrintMsg( wxT( "** end **\n" ) );
@ -217,7 +271,7 @@ void KICAD_MANAGER_FRAME::OnArchiveFiles( wxCommandEvent& event )
curr_fn.MakeRelativeTo( currdirname ); curr_fn.MakeRelativeTo( currdirname );
currFilename = curr_fn.GetFullPath(); currFilename = curr_fn.GetFullPath();
msg.Printf( _( "Archive file \"%s\"" ), GetChars( currFilename ) ); msg.Printf( _( "Archive file \"%s\"" ), currFilename );
PrintMsg( msg ); PrintMsg( msg );
// Read input file and add it to the zip file: // Read input file and add it to the zip file:
@ -245,14 +299,14 @@ void KICAD_MANAGER_FRAME::OnArchiveFiles( wxCommandEvent& event )
if( zipstream.Close() ) if( zipstream.Close() )
{ {
msg.Printf( _( "\nZip archive \"%s\" created (%d bytes)" ), msg.Printf( _( "\nZip archive \"%s\" created (%d bytes)" ),
GetChars( zipfilename ), zipBytesCnt ); zipfilename, zipBytesCnt );
PrintMsg( msg ); PrintMsg( msg );
PrintMsg( wxT( "\n** end **\n" ) ); PrintMsg( wxT( "\n** end **\n" ) );
} }
else else
{ {
msg.Printf( wxT( "Unable to create archive \"%s\", abort\n" ), msg.Printf( wxT( "Unable to create archive \"%s\", abort\n" ),
GetChars( zipfilename ) ); zipfilename );
PrintMsg( msg ); PrintMsg( msg );
} }