view src/getexr/getexr.go @ 13:4fb189ae0a46

getexr: version up.
author pyon@macmini
date Wed, 03 Jan 2018 08:10:34 +0900
parents aea69f84ed33
children 001e2aa380ad
line wrap: on
line source

package main

import (
    "fmt"
    "flag"
    "io/ioutil"
    "log"
    "math"
    "net/http"
    "net/smtp"
    "os"
    "strconv"
    "strings"
    "time"

    "golang.org/x/net/html"
)

var ver = "0.1.1"

func main() {

    version := flag.Bool( "v", false, "print version" )
    verbose := flag.Bool( "V", false, "verbose" )
    file := flag.String( "f", "usd.dat", "data file" )
    flag.Parse()

    if ( *version ) {
        fmt.Fprintf( os.Stderr, "getexr - Get Exchange Rate  [ version = %s ]\n",ver )
        os.Exit( 0 )
    }

    // Get USD Rate
    PrintVerbose( *verbose, "> get rate from yahoo..." )
    url := "http://finance.yahoo.co.jp/"
    resp, err := http.Get( url )
    if err != nil {
        log.Fatal( err )
    }
    PrintVerbose( *verbose, "done\n" )

    z := html.NewTokenizer( resp.Body )
    usd, err := ParseHtml( z )
    if err != nil {
        log.Fatal( err )
    }
    defer resp.Body.Close()

    // Compare Past Rate
    content, err := ioutil.ReadFile( *file )
    if err != nil {
        log.Fatal( err )
    }

    buf1 := strings.Split( string( content ), "\n" )
    buf2 := strings.Split( buf1[ len( buf1 ) - 2 ], "," )
    usd0, _ := strconv.ParseFloat( buf2[1], 32 )

    if ( math.Abs( usd0 - usd ) > 1.0 ) {
        PrintVerbose( *verbose, "> rate changed.\n" )
        t := time.Now()
        now := t.Format( "2006-01-02 15:04:05" )

        // Save Rate
        buf := fmt.Sprintf( "%s,%.2f\n", now, usd )
        content = append( content,  buf... )

        err = ioutil.WriteFile( *file, content, 0777 )
        if err != nil {
            log.Fatal( err )
        }

        // Post E-mail
        PrintVerbose( *verbose, "> sending mail.\n" )
        err = SendMail( now, usd, buf2[0], usd0 )
        if err != nil {
            log.Fatal( err )
        }
    } else {
        PrintVerbose( *verbose, "> no change.\n" )
    }

    PrintVerbose( *verbose, "> finish.\n" )
}

func ParseHtml( z *html.Tokenizer ) ( float64, error ) {
    for {
        tt := z.Next()
        switch tt {
        case html.ErrorToken:
            return 0.0, nil

        case html.StartTagToken:
            tag, _ := z.TagName()
            key, val, _ := z.TagAttr()
            if string( tag ) == "strong" && string( key ) == "class" && string( val ) == "bkLine" {
                z.Next()
                //fmt.Printf( "%s %s %s\n", key, val, z.Token() )
                return strconv.ParseFloat( z.Token().String(), 32 )
            }
        }
    }
}

func SendMail( now string, usd float64, old string, usd0 float64 ) error {
    address := "muty@willcom.com"
    hostname := "sdm.sakura.ne.jp"
    auth := smtp.PlainAuth( "", "bad_user@sdm.sakura.ne.jp", "hogehoge3", hostname )
    to := []string{ address }
    msg := []byte( "To: " + address + "\r\n" +
                "Subject: Exchange-Mail\r\n" +
                "\r\n" +
                "rate changed\r\n" +
                now + "\r\n" +
                fmt.Sprintf( "[ USD/JPY : %.2f ].", usd ) + "\r\n\r\n" +
                old + "\r\n" +
                fmt.Sprintf( "[ USD/JPY : %.2f ].", usd0 ) + "\r\n" ) // mail body
    err := smtp.SendMail( hostname + ":587", auth, "bad_user@sdm.sakura.ne.jp", to, msg )
    if err != nil {
        return err
    }
    return nil
}

func PrintVerbose( y bool, msg string ) {
    if y {
        fmt.Fprintf( os.Stderr, msg )
    }
}