podcatch/main/podcatch.go

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
}