diff --git a/src/lib/Sitemap.php b/src/lib/Sitemap.php
index 30427ca..0d0e32f 100644
--- a/src/lib/Sitemap.php
+++ b/src/lib/Sitemap.php
@@ -41,7 +41,7 @@ class Sitemap
*
* @var string
*/
- private $urlset_header = '
';
+ private $urlset_header = '';
/**
* @var string
*/
@@ -104,6 +104,9 @@ public function getFilePath()
*/
public function setFilePath($file_path)
{
+ if (mb_substr($file_path, -1) !== '/') {
+ $file_path .= '/';
+ }
$this->file_path = $file_path;
}
diff --git a/src/lib/SitemapGenerator.php b/src/lib/SitemapGenerator.php
index db8843a..74aed2f 100644
--- a/src/lib/SitemapGenerator.php
+++ b/src/lib/SitemapGenerator.php
@@ -34,6 +34,18 @@ class SitemapGenerator
* @var array
*/
private $url = array();
+ /**
+ * [
+ * 'loc' => 'image_url',
+ * 'title' => 'Image Title'
+ * ]
+ * @var array
+ */
+ private $url_image = array();
+ /**
+ * @var int
+ */
+ private $url_limit = 50000;
/**
* The date of last modification of the page. This date should be in W3C Datetime format.
* This format allows you to omit the time portion, if desired, and use YYYY-MM-DD.
@@ -119,6 +131,9 @@ public function add_url_to_list()
*/
public function getUrl()
{
+ if ($this->getUrlImage()) {
+ $this->url['image'] = $this->getUrlImage();
+ }
return $this->url;
}
@@ -130,6 +145,38 @@ public function setUrl($url)
$this->url = $url;
}
+ /**
+ * @return array
+ */
+ public function getUrlImage()
+ {
+ return $this->url_image;
+ }
+
+ /**
+ * @param array $url_image
+ */
+ public function setUrlImage($url_image)
+ {
+ $this->url_image = $url_image;
+ }
+
+ /**
+ * @return int
+ */
+ public function getUrlLimit()
+ {
+ return $this->url_limit;
+ }
+
+ /**
+ * @param int $url_limit
+ */
+ public function setUrlLimit($url_limit)
+ {
+ $this->url_limit = $url_limit;
+ }
+
/**
* @return string
*/
@@ -239,14 +286,38 @@ public function set_url_priority($url_priority)
}
/**
+ * @param $url_image_loc
* @return void
*/
- public function set_urlset_body()
+ public function set_url_image_loc($url_image_loc)
{
- $url_list = $this->getUrllist();
+ if (strpos($url_image_loc, $this->getSitemap()->getDomain()) == false) {
+ $url_image_loc = $this->getSitemap()->getDomain().'/'.$url_image_loc;
+ }
+ $this->url_image['loc'] = $url_image_loc;
+ }
+
+ /**
+ * @param $url_image_title
+ * @return void
+ */
+ public function set_url_image_title($url_image_title)
+ {
+ $this->url_image['title'] = $url_image_title;
+ }
+
+ /**
+ * @param $url_list
+ * @return void
+ */
+ public function set_urlset_body($url_list = array())
+ {
+ if (empty($url_list)) {
+ $url_list = $this->getUrllist();
+ }
$data = '';
if (!empty($url_list)) {
- foreach ($this->url_list as $url) {
+ foreach ($url_list as $url) {
$item = (object) $url;
$data .= '';
if (isset($item->loc)) {
@@ -258,6 +329,16 @@ public function set_urlset_body()
if (isset($item->priority)) {
$data .= ''.$item->priority.'';
}
+ if (isset($item->image)) {
+ $data .= '';
+ if (isset($item->image['loc'])) {
+ $data .= ''.$item->image['loc'].'';
+ }
+ if (isset($item->image['title'])) {
+ $data .= ''.$item->image['title'].'';
+ }
+ $data .= '';
+ }
$data .= '';
}
}
@@ -266,7 +347,7 @@ public function set_urlset_body()
/**
* @param $path
- * @return Response|true
+ * @return Response
*/
public function create_file_path($path)
{
@@ -285,6 +366,39 @@ public function create_file_path($path)
return $this->response;
}
+ /**
+ * @param $file_path
+ * @param $file_name
+ * @param $file_ext
+ * @param $index_dir
+ * @return Response
+ */
+ public function create_sitemap_index($file_path, $file_name, $file_ext, $index_dir)
+ {
+ $sitemap_list = scandir($index_dir);
+ if (!empty($sitemap_list) && count($sitemap_list) > 2) {
+ $sitemap_index_header = '';
+ $sitemap_index_footer = '';
+ $sitemap_index_content = '';
+ foreach ($sitemap_list as $sitemap) {
+ if ($sitemap === '.' || $sitemap === '..') {
+ continue;
+ }
+ $file_url = $this->base_url.str_replace($_SERVER["DOCUMENT_ROOT"], '', $index_dir).$sitemap;
+ $sitemap_index_content .= '
+ '.$file_url.'
+ '.date('Y-m-d', filectime($index_dir.$sitemap)).'
+ ';
+ }
+ $file_data = $sitemap_index_header.$sitemap_index_content.$sitemap_index_footer;
+ $this->response = $this->write($file_name, $file_path, $file_ext, $file_data);
+ } else {
+ $this->response->setStatus(false);
+ $this->response->setMessage('Sitemap index files not found.
Date: '.$this->response->getDate().', Sitemap index dir: '.$index_dir.'');
+ }
+ return $this->response;
+ }
+
/**
* @param $file_name
* @param $file_path
@@ -306,7 +420,7 @@ public function write($file_name, $file_path, $file_ext, $file_data)
$this->response->setStatus(true);
$this->response->setMessage('Sitemap file created successfully.
Date: '.$this->response->getDate().', File path: '.$full_path.'');
} else {
- $this->response->setMessage('Sitemap file could not write.
Date: '.$this->response->getDate().'');
+ $this->response->setMessage('Sitemap file could not write.
Date: '.$this->response->getDate().', File path: '.$full_path.'');
}
} else {
$this->response = $create_file_path;
@@ -322,8 +436,42 @@ public function generate()
$file_path = $this->sitemap->getFilePath();
$file_name = $this->sitemap->getFileName();
$file_ext = $this->sitemap->getFileExt();
- $this->set_urlset_body();
- $file_data = $this->sitemap->getHeader().$this->sitemap->getUrlsetHeader().$this->sitemap->getUrlsetBody().$this->sitemap->getUrlsetFooter();
- return $this->write($file_name, $file_path, $file_ext, $file_data);
+ $url_list = $this->getUrllist();
+ $url_limit = $this->getUrlLimit();
+ /*
+ * If url limit is not 0 (zero)
+ */
+ if (!empty($url_limit)) {
+ $url_list_chunk = array_chunk($url_list, $url_limit);
+ /*
+ * If there is more than 1 file, a sitemap index will be created
+ */
+ if (count($url_list_chunk) > 1) {
+ $file_index_path = $file_path.'index/';
+ $i = 1;
+ foreach ($url_list_chunk as $list) {
+ $this->set_urlset_body($list);
+ $file_index_data = $this->sitemap->getHeader().$this->sitemap->getUrlsetHeader().$this->sitemap->getUrlsetBody().$this->sitemap->getUrlsetFooter();
+ $file_index_name = $file_name.'-'.$i;
+ $this->response = $this->write($file_index_name, $file_index_path, $file_ext, $file_index_data);
+ if (!$this->response->isStatus()) {
+ break;
+ }
+ $i++;
+ }
+ if ($this->response->isStatus()) {
+ $this->response = $this->create_sitemap_index($file_path, $file_name, $file_ext, $file_index_path);
+ }
+ } else {
+ $this->set_urlset_body();
+ $file_data = $this->sitemap->getHeader().$this->sitemap->getUrlsetHeader().$this->sitemap->getUrlsetBody().$this->sitemap->getUrlsetFooter();
+ $this->response = $this->write($file_name, $file_path, $file_ext, $file_data);
+ }
+ } else {
+ $this->set_urlset_body();
+ $file_data = $this->sitemap->getHeader().$this->sitemap->getUrlsetHeader().$this->sitemap->getUrlsetBody().$this->sitemap->getUrlsetFooter();
+ $this->response = $this->write($file_name, $file_path, $file_ext, $file_data);
+ }
+ return $this->response;
}
}
\ No newline at end of file