Merge pull request #221 from pixelpark/CustomRepoLocation

Added an option to set a custom repository location

Author: cyberious

README.md

@@ -92,6 +92,19 @@ class {'::mongodb::server': }->
 class {'::mongodb::client': }
 ```
 
+If you don't want to use the 10gen/MongoDB software repository or the OS packages,
+you can point the module to a custom one.
+To install MongoDB from a custom repository:
+
+```puppet
+class {'::mongodb::globals':
+  manage_package_repo => true,
+  repo_location => 'http://example.com/repo'
+}->
+class {'::mongodb::server': }->
+class {'::mongodb::client': }
+```
+
 Having a local copy of MongoDB repository (that is managed by your private modules)
 you can still enjoy the charms of `mongodb::params` that manage packages.
 To disable managing of repository, but still enable managing packages:
@@ -215,6 +228,10 @@ The version of MonogDB to install/manage. This is a simple way of providing
 a specific version such as '2.2' or '2.4' for example. If not specified,
 the module will use the default for your OS distro.
 
+#####`repo_location`
+This setting can be used to override the default MongoDB repository location.
+If not specified, the module will use the default repository for your OS distro.
+
 ####Class: mongodb::server
 
 Most of the parameters manipulate the mongod.conf file.

manifests/globals.pp

@@ -28,13 +28,15 @@
   $manage_package_repo   = undef,
   $manage_package        = undef,
 
+  $repo_location         = undef,
   $use_enterprise_repo   = undef,
 ) {
 
   # Setup of the repo only makes sense globally, so we are doing it here.
   if($manage_package_repo) {
     class { '::mongodb::repo':
-      ensure  => present,
+      ensure        => present,
+      repo_location => $repo_location,
     }
   }
 }

manifests/init.pp

@@ -38,7 +38,6 @@
   $enable_10gen    = undef,
 
   $init            = $mongodb::params::service_provider,
-  $location        = '',
   $packagename     = undef,
   $version         = undef,
   $servicename     = $mongodb::params::service_name,

manifests/repo.pp

@@ -1,11 +1,15 @@
 # PRIVATE CLASS: do not use directly
 class mongodb::repo (
-  $ensure  = $mongodb::params::ensure,
-  $version = $mongodb::params::version,
+  $ensure        = $mongodb::params::ensure,
+  $version       = $mongodb::params::version,
+  $repo_location = undef,
 ) inherits mongodb::params {
   case $::osfamily {
     'RedHat', 'Linux': {
-      if $mongodb::globals::use_enterprise_repo == true {
+      if ($repo_location != undef){
+        $location = $repo_location
+        $description = 'MongoDB Custom Repository'
+      } elsif $mongodb::globals::use_enterprise_repo == true {
         $location = 'https://repo.mongodb.com/yum/redhat/$releasever/mongodb-enterprise/stable/$basearch/'
         $description = 'MongoDB Enterprise Repository'
       }
@@ -25,14 +29,19 @@
         }
         $description = 'MongoDB/10gen Repository'
       }
+
       class { '::mongodb::repo::yum': }
     }
 
     'Debian': {
-      $location = $::operatingsystem ? {
-        'Debian' => 'http://downloads-distro.mongodb.org/repo/debian-sysvinit',
-        'Ubuntu' => 'http://downloads-distro.mongodb.org/repo/ubuntu-upstart',
-        default  => undef
+      if ($repo_location != undef){
+        $location = $repo_location
+      }else{
+        $location = $::operatingsystem ? {
+          'Debian' => 'http://downloads-distro.mongodb.org/repo/debian-sysvinit',
+          'Ubuntu' => 'http://downloads-distro.mongodb.org/repo/ubuntu-upstart',
+          default  => undef
+        }
       }
       class { '::mongodb::repo::apt': }
     }

manifests/repo/apt.pp

@@ -6,7 +6,7 @@
   include ::apt
 
   if($::mongodb::repo::ensure == 'present' or $::mongodb::repo::ensure == true) {
-    apt::source { 'downloads-distro.mongodb.org':
+    apt::source { 'mongodb':
       location    => $::mongodb::repo::location,
       release     => 'dist',
       repos       => '10gen',
@@ -15,10 +15,10 @@
       include_src => false,
     }
 
-    Apt::Source['downloads-distro.mongodb.org']->Package<|tag == 'mongodb'|>
+    Apt::Source['mongodb']->Package<|tag == 'mongodb'|>
   }
   else {
-    apt::source { 'downloads-distro.mongodb.org':
+    apt::source { 'mongodb':
       ensure => absent,
     }
   }