source/packagemanager.cpp

   1 #include <cstdlib>
   2 #include <msp/fs/dir.h>
   3 #include <msp/fs/stat.h>
   4 #include <msp/fs/utils.h>
   5 #include <msp/io/print.h>
   6 #include <msp/strings/format.h>
   7 #include <msp/strings/utils.h>
   8 #include <msp/time/units.h>
   9 #include <msp/time/utils.h>
  10 #include "binarypackage.h"
  11 #include "builder.h"
  12 #include "externaltask.h"
  13 #include "package.h"
  14 #include "packagemanager.h"
  15
  16 using namespace std;
  17 using namespace Msp;
  18
  19 PackageManager::PackageManager(Builder &b):
  20         builder(b),
  21         no_externals(false),
  22         env_set(false)
  23 {
  24         pkg_path.push_back(builder.get_work_directory()/".");
  25         pkg_path.push_back(builder.get_work_directory()/"..");
  26 }
  27
  28 PackageManager::~PackageManager()
  29 {
  30         for(PackageMap::iterator i=packages.begin(); i!=packages.end(); ++i)
  31                 delete i->second;
  32 }
  33
  34 void PackageManager::set_no_externals(bool x)
  35 {
  36         no_externals = x;
  37 }
  38
  39 void PackageManager::add_package(Package *pkg)
  40 {
  41         PackageMap::iterator i = packages.find(pkg->get_name());
  42         if(i!=packages.end())
  43         {
  44                 if(i->second!=pkg)
  45                         throw invalid_argument("Package name conflict");
  46                 else
  47                         throw logic_error("Package is already managed");
  48         }
  49
  50         packages.insert(PackageMap::value_type(pkg->get_name(), pkg));
  51 }
  52
  53 Package *PackageManager::find_package(const string &name)
  54 {
  55         PackageMap::iterator i = packages.find(format("%s/%s", name, builder.get_current_arch().get_system()));
  56         if(i==packages.end())
  57                 i = packages.find(name);
  58         if(i!=packages.end())
  59                 return i->second;
  60
  61         if(!no_externals)
  62         {
  63                 FS::Path path = get_package_location(name);
  64                 if(!path.empty() && !builder.load_build_file(path/"Build"))
  65                 {
  66                         i = packages.find(name);
  67                         if(i!=packages.end())
  68                                 return i->second;
  69                 }
  70         }
  71
  72         Package *pkg = 0;
  73         try
  74         {
  75                 // Package source not found - create a binary package
  76                 string flags_str = run_pkgconfig(name, "flags");
  77                 vector<string> flags = split(flags_str);
  78                 pkg = BinaryPackage::from_flags(builder, name, flags);
  79         }
  80         catch(...)
  81         {
  82                 builder.problem(name, "not found");
  83         }
  84
  85         packages.insert(PackageMap::value_type(name, pkg));
  86
  87         return pkg;
  88 }
  89
  90 string PackageManager::run_pkgconfig(const string &pkg, const string &what)
  91 {
  92         if(!env_set)
  93         {
  94                 const FS::Path &prefix = builder.get_prefix();
  95                 if(prefix.str()!="/usr")
  96                 {
  97                         FS::Path pcdir = prefix/"lib/pkgconfig";
  98                         if(const char *pcp = getenv("PKG_CONFIG_PATH"))
  99                         {
 100                                 vector<string> path = split(pcp, ':');
 101                                 bool found = false;
 102                                 for(vector<string>::const_iterator i=path.begin(); (!found && i!=path.end()); ++i)
 103                                         found = (*i==pcdir.str());
 104                                 if(!found)
 105                                 {
 106                                         path.push_back(pcdir.str());
 107                                         setenv("PKG_CONFIG_PATH", join(path.begin(), path.end(), ":").c_str(), true);
 108                                 }
 109                         }
 110                         else
 111                                 setenv("PKG_CONFIG_PATH", pcdir.str().c_str(), true);
 112                 }
 113         }
 114
 115         ExternalTask::Arguments argv;
 116         argv.push_back("pkg-config");
 117         if(what=="cflags" || what=="libs")
 118                 argv.push_back("--"+what);
 119         else if(what=="flags")
 120         {
 121                 argv.push_back("--cflags");
 122                 argv.push_back("--libs");
 123         }
 124         else
 125                 argv.push_back("--variable="+what);
 126         argv.push_back(pkg);
 127
 128         builder.get_logger().log("auxcommands", format("Running %s", join(argv.begin(), argv.end())));
 129
 130         return ExternalTask::run_and_capture_output(argv);
 131 }
 132
 133 FS::Path PackageManager::get_package_location(const string &name)
 134 {
 135         builder.get_logger().log("packagemgr", format("Looking for package %s", name));
 136
 137         try
 138         {
 139                 // Try to get source directory with pkgconfig
 140                 string srcdir = strip(run_pkgconfig(name, "source"));
 141                 if(!srcdir.empty())
 142                         return srcdir;
 143         }
 144         catch(...)
 145         { }
 146
 147         if(pkg_dirs.empty())
 148         {
 149                 for(list<FS::Path>::const_iterator i=pkg_path.begin(); i!=pkg_path.end(); ++i)
 150                 {
 151                         builder.get_logger().log("files", format("Traversing %s", *i));
 152                         list<string> files = list_files(*i);
 153                         for(list<string>::const_iterator j=files.begin(); j!=files.end(); ++j)
 154                         {
 155                                 FS::Path full = *i / *j;
 156                                 if(FS::exists(full/"Build"))
 157                                         pkg_dirs.push_back(full);
 158                         }
 159                 }
 160                 builder.get_logger().log("packagemgr", format("%d packages found in path", pkg_dirs.size()));
 161         }
 162
 163         bool msp = !name.compare(0, 3, "msp");
 164         for(list<FS::Path>::const_iterator i=pkg_dirs.begin(); i!=pkg_dirs.end(); ++i)
 165         {
 166                 string base = basename(*i);
 167                 unsigned dash = base.rfind('-');
 168
 169                 if(!base.compare(0, dash, name))
 170                         return *i;
 171                 else if(msp && !base.compare(0, dash-3, name, 3, string::npos))
 172                         return *i;
 173         }
 174
 175         return FS::Path();
 176 }