示例#1
0
// Produces a good looking HTML for the given newsletter and stories
func Produce(newsletter *newsletters.Newsletter, all_stories []*stories.Story) string {
	var output bytes.Buffer

	var everything all_information
	everything.Title = newsletter.GetTitle()
	everything.NumStories = len(all_stories)

	// Group stories based on source type
	everything.TextStories = make([]*stories.Story, 0, 10)
	for _, story := range all_stories {
		everything.TextStories = append(everything.TextStories, story)
	}

	tmpl, err := template.ParseFiles("./src/munch/producer/email.html")
	if err != nil {
		fmt.Printf("There was an error loading the template file: %s\n", err)
	}
	tmpl.Execute(&output, everything)

	b := output.Bytes()
	len_total := len(b)

	minified, err := htmlmin.Minify(b, &htmlmin.Options{MinifyScripts: true, MinifyStyles: true, UnquoteAttrs: true})

	len_min := len(minified)

	if err != nil {
		fmt.Printf("There was an error in minifying... using the unminified version\n")
		return output.String()
	}
	fmt.Printf("Total saving of %d bytes\n", (len_total - len_min))
	return string(minified)
}
// WARNING: Mutation of newsletter here
func (db *Db) FetchNewsletterSources(newsletter *newsletters.Newsletter) {
	output, err := db.connection.Query("SELECT source_id, source_lastchecked FROM broccoli_newsletters_sources WHERE newsletter_id = ?", newsletter.GetId())
	if err != nil {
		fmt.Printf("There was an error fetching all the source IDs\n%s\n", err)
	}

	for output.Next() {
		var source_id int
		var source_lastchecked int64
		output.Scan(&source_id, &source_lastchecked)
		fmt.Printf("Source last checked = %d\n", source_lastchecked)
		source := db.FetchSource(source_id)
		newsletter.AddSource(source, source_lastchecked)
	}
}
func (db *Db) PersistNewsletter(newsletter *newsletters.Newsletter) {
	stmt, err := db.connection.Prepare("UPDATE broccoli_newsletters SET title=?, pubdate=? WHERE id = ?")
	if err != nil {
		fmt.Printf("There was an error\n")
	}
	result, err := stmt.Exec(newsletter.GetTitle(), newsletter.GetPubDate(), newsletter.GetId())
	rows_affected, err := result.RowsAffected()
	if rows_affected == 0 {
		// Use INSERT
		stmt, err = db.connection.Prepare("INSERT INTO broccoli_newsletters(title, pubdate) VALUES (?, ?)")
		_, err = stmt.Exec(newsletter.GetTitle(), newsletter.GetPubDate())
	}

	// Now, store the newsletter sources
	sources := newsletter.GetSources()
	sources_lastchecked := newsletter.GetSourcesLastChecked()
	newsletter_id := newsletter.GetId()
	for i, src := range sources {
		src_id := (*src).GetId()

		// Use update
		stmt, err = db.connection.Prepare("UPDATE broccoli_newsletters_sources SET newsletter_id=?, source_id=?, source_lastchecked=?")
		result, err = stmt.Exec(newsletter_id, src_id, sources_lastchecked[i])
		rows_affected, err = result.RowsAffected()
		if rows_affected == 0 {
			// Use insert
			stmt, err = db.connection.Prepare("INSERT broccoli_newsletters_sources(newsletter_id, source_id, source_lastchecked) VALUES (?, ?, ?)")
			_, err = stmt.Exec(newsletter.GetId(), newsletter_id, src_id, sources_lastchecked[i])
		}
	}
}