RepoInfo.cc

Go to the documentation of this file.

/*---------------------------------------------------------------------\
|                          ____ _   __ __ ___                          |
|                         |__  / \ / / . \ . \                         |
|                           / / \ V /|  _/  _/                         |
|                          / /__ | | | | | |                           |
|                         /_____||_| |_| |_|                           |
|                                                                      |
\---------------------------------------------------------------------*/
#include <iostream>
#include <vector>

#include "zypp/base/Logger.h"
#include "zypp/base/DefaultIntegral.h"
#include "zypp/parser/xml/XmlEscape.h"

#include "zypp/RepoInfo.h"
#include "zypp/repo/RepoInfoBaseImpl.h"
#include "zypp/repo/RepoMirrorList.h"
#include "zypp/ExternalProgram.h"
#include "zypp/media/MediaAccess.h"

using namespace std;
using zypp::xml::escape;

namespace zypp
{ 

  //
  //    CLASS NAME : RepoInfo::Impl
  //
  struct RepoInfo::Impl : public repo::RepoInfoBase::Impl
  {
    Impl()
      : repo::RepoInfoBase::Impl()
      , gpgcheck(indeterminate)
      , keeppackages(indeterminate)
      , type(repo::RepoType::NONE_e)
      , emptybaseurls(false)
    {}

    ~Impl()
    {}

  public:
    static const unsigned defaultPriority = 99;

    void setProbedType( const repo::RepoType & t ) const
    {
      if ( type == repo::RepoType::NONE
           && t != repo::RepoType::NONE )
      {
        // lazy init!
        const_cast<Impl*>(this)->type = t;
      }
    }

  public:
    Pathname licenseTgz() const
    { return metadatapath.empty() ? Pathname() : metadatapath / path / "license.tar.gz"; }

    Url getmirrorListUrl() const
    {
      return replacer(mirrorlist_url);
    }

    Url &setmirrorListUrl()
    {
      return mirrorlist_url;
    }

    const std::set<Url> &baseUrls() const
    {
      if ( _baseUrls.empty() && ! (getmirrorListUrl().asString().empty()) )
      {
        emptybaseurls = true;
        repo::RepoMirrorList *rmirrorlist = NULL;

        DBG << "MetadataPath: " << metadatapath << endl;
        if( metadatapath.empty() )
          rmirrorlist = new repo::RepoMirrorList (getmirrorListUrl() );
        else
          rmirrorlist = new repo::RepoMirrorList (getmirrorListUrl(), metadatapath );

        std::vector<Url> rmurls = rmirrorlist->getUrls();
        delete rmirrorlist;
        rmirrorlist = NULL;
        _baseUrls.insert(rmurls.begin(), rmurls.end());
      }
      return _baseUrls;
    }

    std::set<Url> &baseUrls()
    {
      return _baseUrls;
    }

    bool baseurl2dump() const
    {
      return !emptybaseurls && !_baseUrls.empty();
    }

    bool keepPackagesDefault() const
    {
      if (indeterminate(keeppackages))
      {
        if (_baseUrls.empty())
          return mirrorlist_url.schemeIsDownloading();
        else
          return _baseUrls.begin()->schemeIsDownloading();
      }
      return (bool) keeppackages;
    }

  public:
    TriBool gpgcheck;
    TriBool keeppackages;
    Url gpgkey_url;
    repo::RepoType type;
    Pathname path;
    std::string service;
    std::string targetDistro;
    Pathname metadatapath;
    Pathname packagespath;
    DefaultIntegral<unsigned,defaultPriority> priority;
    mutable bool emptybaseurls;
    repo::RepoVariablesUrlReplacer replacer;

  private:
    Url mirrorlist_url;
    mutable std::set<Url> _baseUrls;

    friend Impl * rwcowClone<Impl>( const Impl * rhs );
    Impl * clone() const
    { return new Impl( *this ); }
  };

  inline std::ostream & operator<<( std::ostream & str, const RepoInfo::Impl & obj )
  {
    return str << "RepoInfo::Impl";
  }

  //
  //    CLASS NAME : RepoInfo
  //

  const RepoInfo RepoInfo::noRepo;

  //
  //    METHOD NAME : RepoInfo::RepoInfo
  //    METHOD TYPE : Ctor
  //
  RepoInfo::RepoInfo()
  : _pimpl( new Impl() )
  {}

  //
  //    METHOD NAME : RepoInfo::~RepoInfo
  //    METHOD TYPE : Dtor
  //
  RepoInfo::~RepoInfo()
  {
    //MIL << std::endl;
  }

  unsigned RepoInfo::priority() const
  { return _pimpl->priority; }
  unsigned RepoInfo::defaultPriority()
  { return Impl::defaultPriority; }
  void RepoInfo::setPriority( unsigned newval_r )
  {
    _pimpl->priority = newval_r ? newval_r : Impl::defaultPriority;
  }

  void RepoInfo::setGpgCheck( bool check )
  {
    _pimpl->gpgcheck = check;
  }

  void RepoInfo::setMirrorListUrl( const Url &url )
  {
    _pimpl->setmirrorListUrl() = url;
  }

  void RepoInfo::setGpgKeyUrl( const Url &url )
  {
    _pimpl->gpgkey_url = url;
  }

  void RepoInfo::addBaseUrl( const Url &url )
  {
    _pimpl->baseUrls().insert(url);
  }

  void RepoInfo::setBaseUrl( const Url &url )
  {
    _pimpl->baseUrls().clear();
    addBaseUrl(url);
  }

  void RepoInfo::setPath( const Pathname &path )
  {
    _pimpl->path = path;
  }

  void RepoInfo::setType( const repo::RepoType &t )
  {
    _pimpl->type = t;
  }

  void RepoInfo::setProbedType( const repo::RepoType &t ) const
  { _pimpl->setProbedType( t ); }


  void RepoInfo::setMetadataPath( const Pathname &path )
  {
    _pimpl->metadatapath = path;
  }

  void RepoInfo::setPackagesPath( const Pathname &path )
  {
    _pimpl->packagespath = path;
  }

  void RepoInfo::setKeepPackages( bool keep )
  {
    _pimpl->keeppackages = keep;
  }

  void RepoInfo::setService( const std::string& name )
  {
    _pimpl->service = name;
  }

  void RepoInfo::setTargetDistribution(
      const std::string & targetDistribution)
  {
    _pimpl->targetDistro = targetDistribution;
  }

  bool RepoInfo::gpgCheck() const
  { return indeterminate(_pimpl->gpgcheck) ? true : (bool) _pimpl->gpgcheck; }

  Pathname RepoInfo::metadataPath() const
  { return _pimpl->metadatapath; }

  Pathname RepoInfo::packagesPath() const
  { return _pimpl->packagespath; }

  repo::RepoType RepoInfo::type() const
  { return _pimpl->type; }

  Url RepoInfo::mirrorListUrl() const
  {
    return _pimpl->getmirrorListUrl();
  }

  Url RepoInfo::gpgKeyUrl() const
  { return _pimpl->gpgkey_url; }

  std::set<Url> RepoInfo::baseUrls() const
  {
    RepoInfo::url_set replaced_urls;
    for ( url_set::const_iterator it = _pimpl->baseUrls().begin();
          it != _pimpl->baseUrls().end();
          ++it )
    {
      replaced_urls.insert(_pimpl->replacer(*it));
    }
    return replaced_urls;
  }

  Pathname RepoInfo::path() const
  { return _pimpl->path; }

  std::string RepoInfo::service() const
  { return _pimpl->service; }

  std::string RepoInfo::targetDistribution() const
  { return _pimpl->targetDistro; }

  RepoInfo::urls_const_iterator RepoInfo::baseUrlsBegin() const
  {
    return make_transform_iterator( _pimpl->baseUrls().begin(),
                                    _pimpl->replacer );
    //return _pimpl->baseUrls.begin();
  }

  RepoInfo::urls_const_iterator RepoInfo::baseUrlsEnd() const
  {
    //return _pimpl->baseUrls.end();
    return make_transform_iterator( _pimpl->baseUrls().end(),
                                    _pimpl->replacer );
  }

  RepoInfo::urls_size_type RepoInfo::baseUrlsSize() const
  { return _pimpl->baseUrls().size(); }

  bool RepoInfo::baseUrlsEmpty() const
  { return _pimpl->baseUrls().empty(); }

  bool RepoInfo::baseUrlSet() const
  { return _pimpl->baseurl2dump(); }

  // false by default (if not set by setKeepPackages)
  bool RepoInfo::keepPackages() const
  {
    return _pimpl->keepPackagesDefault();
  }


  bool RepoInfo::hasLicense() const
  {
    Pathname licenseTgz( _pimpl->licenseTgz() );
    SEC << licenseTgz << endl;
    SEC << PathInfo(licenseTgz) << endl;

    return ! licenseTgz.empty() &&  PathInfo(licenseTgz).isFile();
  }

  std::string RepoInfo::getLicense( const Locale & lang_r )
  {
    LocaleSet avlocales( getLicenseLocales() );
    if ( avlocales.empty() )
      return std::string();

    Locale getLang( Locale::bestMatch( avlocales, lang_r ) );
    if ( getLang == Locale::noCode
         && avlocales.find( Locale::noCode ) == avlocales.end() )
    {
      WAR << "License.tar.gz contains no fallback text! " << *this << endl;
      // Using the fist locale instead of returning no text at all.
      // So the user might recognize that there is a license, even if he
      // can't read it.
      getLang = *avlocales.begin();
    }

    // now extract the license file.
    static const std::string licenseFileFallback( "license.txt" );
    std::string licenseFile( getLang == Locale::noCode
                             ? licenseFileFallback
                             : str::form( "license.%s.txt", getLang.code().c_str() ) );

    ExternalProgram::Arguments cmd;
    cmd.push_back( "tar" );
    cmd.push_back( "-x" );
    cmd.push_back( "-z" );
    cmd.push_back( "-O" );
    cmd.push_back( "-f" );
    cmd.push_back( _pimpl->licenseTgz().asString() ); // if it not exists, avlocales was empty.
    cmd.push_back( licenseFile );

    std::string ret;
    ExternalProgram prog( cmd, ExternalProgram::Discard_Stderr );
    for ( std::string output( prog.receiveLine() ); output.length(); output = prog.receiveLine() )
    {
      ret += output;
    }
    prog.close();
    return ret;
  }

  LocaleSet RepoInfo::getLicenseLocales() const
  {
    Pathname licenseTgz( _pimpl->licenseTgz() );
    if ( licenseTgz.empty() || ! PathInfo( licenseTgz ).isFile() )
      return LocaleSet();

    ExternalProgram::Arguments cmd;
    cmd.push_back( "tar" );
    cmd.push_back( "-t" );
    cmd.push_back( "-z" );
    cmd.push_back( "-f" );
    cmd.push_back( licenseTgz.asString() );

    LocaleSet ret;
    ExternalProgram prog( cmd, ExternalProgram::Stderr_To_Stdout );
    for ( std::string output( prog.receiveLine() ); output.length(); output = prog.receiveLine() )
    {
      static const C_Str license( "license." );
      static const C_Str dotTxt( ".txt\n" );
      if ( str::hasPrefix( output, license ) && str::hasSuffix( output, dotTxt ) )
      {
        if ( output.size() <= license.size() +  dotTxt.size() ) // license.txt
          ret.insert( Locale() );
        else
          ret.insert( Locale( std::string( output.c_str()+license.size(), output.size()- license.size() - dotTxt.size() ) ) );
      }
      else
      {
        WAR << "  " << output;
      }
    }
    prog.close();
    return ret;
  }


  std::ostream & RepoInfo::dumpOn( std::ostream & str ) const
  {
    RepoInfoBase::dumpOn(str);
    if ( _pimpl->baseurl2dump() )
    {
      for ( urls_const_iterator it = baseUrlsBegin();
            it != baseUrlsEnd();
            ++it )
      {
        str << "- url         : " << *it << std::endl;
      }
    }
    if ( ! (_pimpl->getmirrorListUrl().asString().empty())  )
    {
      str << "- mirrorlist  : " << _pimpl->getmirrorListUrl() << std::endl;
    }
    str << "- path        : " << path() << std::endl;
    str << "- type        : " << type() << std::endl;
    str << "- priority    : " << priority() << std::endl;

    str << "- gpgcheck    : " << gpgCheck() << std::endl;
    str << "- gpgkey      : " << gpgKeyUrl() << std::endl;

    if (!indeterminate(_pimpl->keeppackages))
      str << "- keeppackages: " << keepPackages() << std::endl;

    str << "- service     : " << service() << std::endl;

    if (!targetDistribution().empty())
      str << "- targetdistro: " << targetDistribution() << std::endl;

    if (!metadataPath().empty())
      str << "- metadataPath: " << metadataPath() << std::endl;

    if (!packagesPath().empty())
      str << "- packagesPath: " << packagesPath() << std::endl;

    return str;
  }

  std::ostream & RepoInfo::dumpAsIniOn( std::ostream & str ) const
  {
    RepoInfoBase::dumpAsIniOn(str);

    if ( _pimpl->baseurl2dump() )
    {
      str << "baseurl=";
      for ( url_set::const_iterator it = _pimpl->baseUrls().begin();
            it != _pimpl->baseUrls().end();
            ++it )
      {
        str << *it << endl;
      }
    }

    if ( ! _pimpl->path.empty() )
      str << "path="<< path() << endl;

    if ( ! (_pimpl->getmirrorListUrl().asString().empty()) )
      str << "mirrorlist=" << _pimpl->getmirrorListUrl() << endl;

    str << "type=" << type().asString() << endl;

    if ( priority() != defaultPriority() )
      str << "priority=" << priority() << endl;

    if (!indeterminate(_pimpl->gpgcheck))
      str << "gpgcheck=" << (gpgCheck() ? "1" : "0") << endl;
    if ( ! (gpgKeyUrl().asString().empty()) )
      str << "gpgkey=" <<gpgKeyUrl() << endl;

    if (!indeterminate(_pimpl->keeppackages))
      str << "keeppackages=" << keepPackages() << endl;

    if( ! service().empty() )
      str << "service=" << service() << endl;

    return str;
  }

  std::ostream & RepoInfo::dumpAsXMLOn( std::ostream & str) const
  { return dumpAsXMLOn(str, ""); }

  std::ostream & RepoInfo::dumpAsXMLOn( std::ostream & str, const std::string & content) const
  {
    string tmpstr;
    str
      << "<repo"
      << " alias=\"" << escape(alias()) << "\""
      << " name=\"" << escape(name()) << "\"";
    if (type() != repo::RepoType::NONE)
      str << " type=\"" << type().asString() << "\"";
    str
      << " enabled=\"" << enabled() << "\""
      << " autorefresh=\"" << autorefresh() << "\""
      << " gpgcheck=\"" << gpgCheck() << "\"";
    if (!(tmpstr = gpgKeyUrl().asString()).empty())
      str << " gpgkey=\"" << escape(tmpstr) << "\"";
    if (!(tmpstr = mirrorListUrl().asString()).empty())
      str << " mirrorlist=\"" << escape(tmpstr) << "\"";
    str << ">" << endl;

    if ( _pimpl->baseurl2dump() )
    {
      for (RepoInfo::urls_const_iterator urlit = baseUrlsBegin();
           urlit != baseUrlsEnd(); ++urlit)
        str << "<url>" << escape(urlit->asString()) << "</url>" << endl;
    }

    str << "</repo>" << endl;
    return str;
  }


  std::ostream & operator<<( std::ostream & str, const RepoInfo & obj )
  {
    return obj.dumpOn(str);
  }


} // namespace zypp