renamed podcatch folder

main.go

@@ -1,6 +1,6 @@
 package main
 import (
-  "podcatch/podcatch"
+  "podcatch/main"
 )
 func main(){
   podcatch.Start()

main/podcatch.go

@@ -0,0 +1,280 @@
+package podcatch
+import (
+  "fmt"
+  . "podcatch/structs"
+  "github.com/pelletier/go-toml"
+  "encoding/xml"
+  // "io"
+  "io/ioutil"
+  "net/http"
+  "regexp"
+  "os"
+  "log"
+  "strings"
+  id3 "github.com/mikkyang/id3-go"
+  v2 "github.com/mikkyang/id3-go/v2"
+)
+var Version string = "0.4"
+var Settings Settings
+var Podcasts map[string]Podcast = make(map[string]Podcast)
+var donefile string
+var podcatchdir string
+var homedir string
+var dbdir string
+func Start(){
+  fmt.Printf("Starting PodCatch Version : %s...\r\n", Version )
+  getHomeDirs()
+  getSettings()
+  getPodcasts()
+}
+func getHomeDirs(){
+  h, err := os.UserHomeDir()
+  if err != nil {
+    log.Fatal( err )
+  }
+  homedir = h
+  podcatchdir = h + "/.podcatch/"
+}
+func getSettings(){
+  settings := podcatchdir + "settings.toml"
+  if !checkFileExists(settings){
+    fmt.Println("Creating default settings.toml in user dir.")
+    ok := createDefaultFile("settings",settings)
+    if ok {
+      fmt.Println("Copied.")
+    }
+  }
+  content, err := ioutil.ReadFile(settings)
+  if err != nil {
+    log.Fatal(err)
+  }
+  e := toml.Unmarshal(content,&Settings)
+  if e != nil {
+    log.Fatal(err)
+  }
+  Settings.Directory = strings.Replace(Settings.Directory,"~",homedir,1)
+  dbdir = Settings.Directory + ".db/"
+  os.Mkdir(dbdir,0755)
+
+}
+func getPodcasts(){
+  if len(Podcasts) == 0 {
+    getPodcastFiles()
+  }
+  for shortname,podcast := range Podcasts {
+    podcast.Directory = shortname
+    fmt.Printf("Checking RSS for %s...\r\n", podcast.Name)
+    podcast.RSS = getRSS(podcast)
+    downloadCasts(podcast)
+  }
+}
+func getPodcastFiles() {
+  pcs := podcatchdir + "podcasts.toml"
+  if !checkFileExists(pcs){
+    fmt.Println("Creating default podcasts.toml in user dir.")
+    ok := createDefaultFile("podcasts",pcs)
+    if ok {
+      fmt.Println("Copied.")
+      fmt.Println("Please Edit the ~/.podcatch/*.toml files as required and run again")
+       os.Exit(0)
+    }
+  }
+  content, err := ioutil.ReadFile(podcatchdir + "podcasts.toml")
+  if err != nil {
+    log.Fatal(err)
+  }
+  e := toml.Unmarshal(content,&Podcasts)
+  if e != nil {
+    log.Fatal(err)
+  }
+  fmt.Printf("Found %d podcasts.\r\n",len(Podcasts))
+}
+func getRSS(podcast Podcast) Rss {
+  resp, err := http.Get(podcast.URL)
+  if err != nil {
+    log.Fatal(err)
+  }
+  defer resp.Body.Close()
+  html, err := ioutil.ReadAll(resp.Body)
+	if err != nil {
+		log.Fatal(err)
+	}
+  return parseRSS(podcast,html)
+}
+func parseRSS(podcast Podcast, rssxml []byte) Rss {
+  var rss Rss
+  e := xml.Unmarshal(rssxml,&rss)
+  if e != nil {
+    log.Fatal(e)
+  }
+  return rss
+}
+func downloadCasts(podcast Podcast) {
+  count := 0
+  for _,item := range podcast.RSS.Channel.Items {
+    if count >= Settings.Limit {
+      break
+    }
+    if !podcastDownloaded(item){
+      fmt.Printf("Downloading '%s %s' from : %s.\r\n", item.Episode, item.Title, item.Media.URL)
+      re := regexp.MustCompile(`[^0-9a-zA-Z-_]+`)
+      filename := item.Episode + re.ReplaceAllString(item.Title,"_") + ".mp3"
+      dir := Settings.Directory + podcast.Directory
+      err := os.Mkdir(dir, 0777)
+      if err != nil && err.Error() != fmt.Sprintf("mkdir %s: file exists",dir){
+        log.Fatal(err)
+      }
+      ok := downloadMp3(item.Media.URL, dir + "/" + filename)
+      if ok {
+        // createNFO(item, strings.Replace(dir + "/" + filename,".mp3",".nfo",1))
+        addId3(podcast.Name, item,dir + "/" + filename)
+        markAsReceived(item)
+      } else {
+        markAsErrored(item)
+      }
+    } else {
+      fmt.Printf("Skipping '%s' - already downloaded\r\n", item.Title)
+    }
+    count = count + 1
+  }
+}
+func podcastDownloaded(item Item) bool {
+  db := dbdir + "complete"
+  if checkCreate(db) {
+    if len(donefile) < 1 {
+      content, err := ioutil.ReadFile(db)
+      if err != nil {
+        log.Fatal(err)
+      }
+      donefile = string(content)
+    }
+    if strings.Contains(donefile,item.Title){
+      return true
+    }
+    if strings.Contains(donefile,item.Media.URL){
+      return true
+    }
+  }
+  return false
+}
+func downloadMp3(url string, file string) bool {
+  ok := false
+  resp, err := http.Get(url)
+  if err != nil {
+    log.Fatal(err)
+  }
+  defer resp.Body.Close()
+  data, err := ioutil.ReadAll(resp.Body)
+  if err != nil {
+    log.Fatal(err)
+  }
+  err = ioutil.WriteFile(file, data, 0775)
+  if err != nil {
+    log.Fatal(err)
+  }
+  ok = true
+  return ok
+}
+func addId3(artist string, item Item, file string) {
+  fmt.Printf("Saving ID3 to %s\r\n",file)
+  mp3File, err := id3.Open(file)
+  if err != nil {
+    log.Fatal(err)
+  }
+  defer mp3File.Close()
+  if mp3File.Artist() == "" {
+    mp3File.SetArtist(artist)
+  }
+  if mp3File.Album() == "" {
+    mp3File.SetAlbum(artist)
+  }
+  if mp3File.Title() == "" {
+    mp3File.SetTitle(item.Title)
+  }
+  if len(mp3File.Comments()) == 0 {
+    ft := v2.V23FrameTypeMap["COMM"]
+    textFrame := v2.NewTextFrame(ft, item.Description)
+    mp3File.AddFrames(textFrame)
+  }
+}
+func markAsReceived(item Item)  {
+  db := dbdir + "complete"
+  checkCreate(db)
+  file, err := os.OpenFile(db, os.O_APPEND|os.O_WRONLY, 0755)
+  if err != nil {
+    log.Println(err)
+  }
+  defer file.Close()
+  content := fmt.Sprintf("%s - %s\r\n",item.Title, item.Media.URL)
+  if _, err := file.WriteString(content); err != nil {
+    log.Fatal(err)
+  }
+}
+func markAsErrored(item Item)  {
+  db := dbdir + "error"
+  checkCreate(db)
+  file, err := os.OpenFile(db, os.O_APPEND|os.O_WRONLY, 0755)
+  if err != nil {
+    log.Println(err)
+  }
+  defer file.Close()
+  content := fmt.Sprintf("%s\r\n%s\r\n",item.Title, item.Media.URL)
+  if _, err := file.WriteString(content); err != nil {
+    log.Fatal(err)
+  }
+}
+func checkFileExists(file string) bool {
+  if _, err := os.Stat(file); err == nil {
+    // fmt.Println("Exists")
+    // exists
+    return true
+  } else if os.IsNotExist(err) {
+      // fmt.Println("Not Exists")
+    // not exists
+    return false
+  } else {
+    // fmt.Println("Maybe Exists, Maybe Not")
+    return false
+  // Schrodinger: file may or may not exist. See err for details.
+  // Therefore, do *NOT* use !os.IsNotExist(err) to test for file existence
+  }
+  return false
+}
+func checkCreate(file string) bool {
+  if checkFileExists(file) {
+    return true
+  } else {
+    if createFile(file) {
+      return true
+    }
+  }
+  return false
+}
+func createFile(file string) bool {
+  f, err := os.Create(file)
+  if err != nil {
+    log.Fatal(err)
+    return false
+  }
+  defer f.Close()
+  return true
+}
+func createDefaultFile(template string, file string) bool {
+  if !checkFileExists(file) {
+    createFile(file)
+  }
+  var data []byte
+  switch template {
+	case "podcasts":
+    data = []byte(DefaultPodcasts)
+	case "settings":
+		data = []byte(DefaultSettings)
+	default:
+		fmt.Printf("Unknown : %s.\n", template)
+	}
+  err := ioutil.WriteFile(file, data, 0775)
+  if err != nil {
+    log.Fatal(err)
+  }
+  return true
+}