story-details.go

package main

import (
	"database/sql"
	"fmt"
	"net/url"
	"strings"
	"time"

	"github.com/weppos/publicsuffix-go/publicsuffix"

	humanize "github.com/dustin/go-humanize"
)

type Story struct {
	ID                        int
	By                        string
	Title                     string
	URL                       string
	SubmissionTime            int64
	OriginalSubmissionTime    int64
	AgeApprox                 int64
	Score                     int
	Comments                  int
	CumulativeUpvotes         int
	CumulativeExpectedUpvotes float64
	UpvoteRate                float64
	TopRank                   sql.NullInt32
	QNRank                    sql.NullInt32
	RawRank                   sql.NullInt32
	Job                       bool
	Flagged                   bool
	Dupe                      bool
	Archived                  bool
}

type PageFlags struct {
	IsHNTopPage          bool
	IsFairPage           bool
	IsUpvoteratePage     bool
	IsBestUpvoteratePage bool
	IsStatsPage          bool
	IsPenaltiesPage      bool
	IsBoostsPage         bool
	IsResubmissionsPage  bool
	IsRawPage            bool
}

// StoryTemplateData combines a Story with page context for use in templates
type StoryTemplateData struct {
	Story // embed Story instead of having it as a named field
	PageFlags
}

// Page-specific methods
func (s PageFlags) IsAlternativeFrontPage() bool {
	return s.IsHNTopPage || s.IsRawPage || s.IsPenaltiesPage || s.IsBoostsPage || s.IsResubmissionsPage || s.IsFairPage || s.IsUpvoteratePage || s.IsBestUpvoteratePage
}

func (s Story) AgeString() string {
	// return humanize.Time(time.Unix(int64(time.Now().Unix()-s.AgeApprox), 0))
	return humanize.Time(time.Unix(int64(time.Now().Unix()-s.AgeApprox), 0))
}

func (s Story) OriginalAgeString() string {
	return humanize.Time(time.Unix(s.OriginalSubmissionTime, 0))
}

func (s Story) IsResubmitted() bool {
	return s.SubmissionTime != s.OriginalSubmissionTime
}

func (s Story) UpvoteRateString() string {
	return fmt.Sprintf("%.2f", s.UpvoteRate)
}

func (s Story) RankDiff() int32 {
	if !s.RawRank.Valid {
		return 0
	}
	rawRank := s.RawRank.Int32
	topRank := s.TopRank.Int32

	if !s.TopRank.Valid {
		topRank = 91
	}

	return rawRank - topRank
}

func abs(a int32) int32 {
	if a >= 0 {
		return a
	}
	return -a
}

func (s Story) RankDiffAbs() int32 {
	return abs(s.RankDiff())
}

func (s Story) OverRanked() bool {
	return s.RankDiff() > 0
}

func (s Story) UnderRanked() bool {
	return s.RankDiff() < 0
}

func (s Story) Domain() string {
	u, err := url.Parse(s.URL)
	if err != nil {
		return ""
	}

	domain, err := publicsuffix.Domain(u.Host)
	if err != nil {
		return ""
	}

	// some domains are treated specially:

	// twitter.com/x
	// github.com/x
	// x.substack.com
	// x.notion.site
	// x.dreamhosters.com

	if u.Host == "news.ycombinator.com" {
		return ""
	}
	if domain == "twitter.com" || domain == "github.com" {
		// keep first part of path
		return domain + "/" + strings.Split(u.Path, "/")[1]
	}

	if domain == "substack.com" || domain == "notion.site" || domain == "dreamhosters.com" {
		// keep subdomain
		return strings.Split(u.Host, ".")[0] + "." + domain
	}

	return domain
}

func (s Story) ISOTimestamp() string {
	return time.Unix(s.SubmissionTime, 0).UTC().Format("2006-01-02T15:04:05")
}

func (s Story) OriginalISOTimestamp() string {
	return time.Unix(s.OriginalSubmissionTime, 0).UTC().Format("2006-01-02T15:04:05")
}