feat: support path

README.md

@@ -18,8 +18,8 @@ package main
 import "github.com/sosolyht/go-sitemap/sitemap"
 
 func main() {
-// Create a standard sitemap
-s := sitemap.NewSitemap()
+// Create a sitemap with path
+s := sitemap.NewSitemap().Path("util/sitemaps")
 
  links := []string{
  "https://google.com",

examples/default-sitemap/main.go

@@ -0,0 +1,15 @@
+package main
+
+import "github.com/sosolyht/go-sitemap/sitemap"
+
+func main() {
+ s := sitemap.NewSitemap().Path("sitemaps")
+
+ links := []string{
+ "https://google.com",
+ "https://naver.com",
+ }
+ for i := range links {
+ s.AddURL(links[i])
+ }
+}

main.go → examples/video-sitemap/main.go

@@ -1,21 +1,9 @@
 package main
 
-import (
- "github.com/sosolyht/go-sitemap/sitemap"
-)
+import "github.com/sosolyht/go-sitemap/sitemap"
 
 func main() {
- s := sitemap.NewSitemap()
-
- links := []string{
- "https://google.com",
- "https://naver.com",
- }
- for i := range links {
- s.AddURL(links[i])
- }
-
- vs := sitemap.NewVideoSitemap()
+ vs := sitemap.NewVideoSitemap().Path("sitemaps")
 
  videoURLs := []sitemap.VideoURL{
  {

sitemap/sitemap.go

@@ -3,8 +3,10 @@ package sitemap
 import (
  "encoding/xml"
  "io"
+ "log"
  "net/http"
  "os"
+ "path/filepath"
  "strings"
  "time"
 )
@@ -13,6 +15,7 @@ type sitemap struct {
  XMLName xml.Name `xml:"urlset"`
  Xmlns string `xml:"xmlns,attr"`
  URL []URLs `xml:"url,omitempty"`
+ path string
 }
 
 type URLs struct {
@@ -60,7 +63,7 @@ func (s *sitemap) AddURL(url string) (err error) {
  return err
  }
 
- sitemapFile, err := os.Create("sitemaps/sitemap.xml")
+ sitemapFile, err := os.Create(s.path)
  if err != nil {
  return err
  }
@@ -78,6 +81,30 @@ func (s *sitemap) AddURL(url string) (err error) {
  return
 }
 
+func (s *sitemap) Path(path string) *sitemap {
+ currentDir, err := os.Getwd()
+ if err != nil {
+ log.Fatal(err)
+ }
+
+ projectRoot := filepath.Join(currentDir, "..", "..")
+ sitemapsDir := filepath.Join(projectRoot, path)
+
+ _, err = os.Stat(sitemapsDir)
+ if err != nil {
+ if os.IsNotExist(err) {
+ err = os.MkdirAll(sitemapsDir, 0755)
+ if err != nil {
+ log.Fatal(err)
+ }
+ }
+ }
+
+ output := filepath.Join(sitemapsDir, "sitemap.xml")
+ s.path = output
+ return s
+}
+
 func (s *sitemap) createSitemapFromLinksFile() ([]string, error) {
  linkFile, err := os.Open("sitemaps/links")
  if err != nil {

sitemap/video_sitemap.go

@@ -2,7 +2,9 @@ package sitemap
 
 import (
  "encoding/xml"
+ "log"
  "os"
+ "path/filepath"
  "time"
 )
 
@@ -11,6 +13,7 @@ type videoSitemap struct {
  Xmlns string `xml:"xmlns,attr"`
  XmlnsVideo string `xml:"xmlns:video,attr"`
  URL []VideoURL
+ path string
 }
 
 type VideoURL struct {
@@ -67,7 +70,7 @@ func (v *videoSitemap) AddVideoURL(url VideoURL) (err error) {
  return err
  }
 
- sitemapFile, err := os.Create("sitemaps/sitemap_video.xml")
+ sitemapFile, err := os.Create(v.path)
  if err != nil {
  return err
  }
@@ -84,3 +87,27 @@ func (v *videoSitemap) AddVideoURL(url VideoURL) (err error) {
 
  return
 }
+
+func (v *videoSitemap) Path(path string) *videoSitemap {
+ currentDir, err := os.Getwd()
+ if err != nil {
+ log.Fatal(err)
+ }
+
+ projectRoot := filepath.Join(currentDir, "..", "..")
+ sitemapsDir := filepath.Join(projectRoot, path)
+
+ _, err = os.Stat(sitemapsDir)
+ if err != nil {
+ if os.IsNotExist(err) {
+ err = os.MkdirAll(sitemapsDir, 0755)
+ if err != nil {
+ log.Fatal(err)
+ }
+ }
+ }
+
+ output := filepath.Join(sitemapsDir, "sitemap_video.xml")
+ v.path = output
+ return v
+}

sitemaps/sitemap.xml

@@ -2,10 +2,10 @@
 <urlset xmlns="http://www.sitemaps.org/schemas/sitemap/0.9">
  <url>
  <loc>https://google.com</loc>
- <lastmod>2023-04-28</lastmod>
+ <lastmod>2023-05-08</lastmod>
  </url>
  <url>
  <loc>https://naver.com</loc>
- <lastmod>2023-04-28</lastmod>
+ <lastmod>2023-05-08</lastmod>
  </url>
 </urlset>