YUMScriptImpl.cc

Go to the documentation of this file.

/*---------------------------------------------------------------------\
|                          ____ _   __ __ ___                          |
|                         |__  / \ / / . \ . \                         |
|                           / / \ V /|  _/  _/                         |
|                          / /__ | | | | | |                           |
|                         /_____||_| |_| |_|                           |
|                                                                      |
\---------------------------------------------------------------------*/
#include "zypp/source/yum/YUMScriptImpl.h"
#include "zypp/Arch.h"
#include "zypp/Edition.h"
#include "zypp/base/Gettext.h"
#include "zypp/ZYppFactory.h"

#include "zypp/source/yum/YUMSourceImpl.h"

#include <fstream>


using namespace std;
using namespace zypp::detail;

namespace zypp
{ 

namespace source
{ 
namespace yum
{

//
//        CLASS NAME : YUMScriptImpl
//

YUMScriptImpl::YUMScriptImpl(
  Source_Ref source_r,
  const zypp::parser::yum::YUMPatchScript & parsed
)
    : _do_script(parsed.do_script)
    , _undo_script(parsed.undo_script)
    , _do_location(parsed.do_location)
    , _undo_location(parsed.undo_location)
    , _do_media(1)
    , _undo_media(1)
    , _do_checksum(parsed.do_checksum_type, parsed.do_checksum)
    , _undo_checksum(parsed.undo_checksum_type, parsed.undo_checksum)
    , _source(source_r)
{
  unsigned do_media = strtol(parsed.do_media.c_str(), 0, 10);
  if (do_media > 0)
    _do_media = do_media;
  unsigned undo_media = strtol(parsed.undo_media.c_str(), 0, 10);
  if (undo_media > 0)
    _undo_media = undo_media;
}

Pathname YUMScriptImpl::do_script() const
{
  if (_do_script != "")
  {
    if ( !_tmp_do_script )
      _tmp_do_script.reset(new filesystem::TmpFile(getZYpp()->tmpPath(), "zypp-yum-do-script-"));

    Pathname pth = _tmp_do_script->path();
    // FIXME check success
    ofstream st(pth.asString().c_str());

    if ( !st )
    {
      ZYPP_THROW(Exception(N_("Can't write the patch script to a temporary file.")));
    }

    st << _do_script << endl;
    return pth;
  }
  else if (_do_location != "" && _do_location != "/")
  {
    Pathname script = source().provideFile(_do_location, _do_media);
    if (! filesystem::is_checksum(script, _do_checksum))
    {
      ZYPP_THROW(Exception(N_("Failed check for the script file check sum")));
    }
    return script;
  }
  else
  {
    return Pathname();
  }
}
Pathname YUMScriptImpl::undo_script() const
{
  if (_undo_script != "")
  {
    if ( !_tmp_undo_script )
      _tmp_undo_script.reset(new filesystem::TmpFile(getZYpp()->tmpPath(), "zypp-yum-undo-script-"));

    Pathname pth = _tmp_undo_script->path();
    // FIXME check success
    ofstream st(pth.asString().c_str());

    if ( !st )
    {
      ZYPP_THROW(Exception(N_("Can't write the patch script to a temporary file.")));
    }

    st << _undo_script << endl;
    return pth;
  }
  else if (_undo_location != "" && _undo_location != "/")
  {
    Pathname script = source().provideFile(_undo_location, _undo_media);
    if (! filesystem::is_checksum(script, _undo_checksum) )
    {
      ZYPP_THROW(Exception(N_("Failed check for the script file check sum")));
    }
    return script;
  }
  else return Pathname();
}
bool YUMScriptImpl::undo_available() const
{
  return _undo_script != ""
         || (_undo_location != "" && _undo_location != "/");
}

Source_Ref YUMScriptImpl::source() const
{
  return _source;
}



} // namespace yum
} // namespace source
} // namespace zypp

---