ohlc/main.go

package ohlc

import (
	"encoding/json"
	"fmt"
	"os"
	"time"
)

func Conv(name string) {
	text, err := os.ReadFile(name + ".json")
	if err != nil {
		fmt.Printf("%s: %s\n", name, err)
		return
	}

	var i interface{}
	err = json.Unmarshal(text, &i)
	if err != nil {
		fmt.Printf("%s: %s\n", name, err)
		return
	}

	j, ok := i.(map[string]interface{})
	if !ok { fmt.Println(name + ": Bad format"); return }

	j, ok = j["chart"].(map[string]interface{})
	if !ok { fmt.Println(name + ": Bad format (chart)"); return }

	x, ok := j["result"].([]interface{})
	if !ok { fmt.Println(name + ": Bad format (result)"); return }

	result, ok := x[0].(map[string]interface{})
	if !ok { fmt.Println(name + ": Bad format (result[0])"); return }

	timestamp, ok := result["timestamp"].([]interface{})
	if !ok { fmt.Println(name + ": Bad format (timestamp)"); return }

	indicators, ok := result["indicators"].(map[string]interface{})
	if !ok { fmt.Println(name + ": Bad format (indicators)"); return }

	events, ok := result["events"].(map[string]interface{})

	dividends, ok := events["dividends"].(map[string]interface{})

	capgains, ok := events["capitalGains"].(map[string]interface{})

	splits, ok := events["splits"].(map[string]interface{})

	x, ok = indicators["quote"].([]interface{})
	if !ok { fmt.Println(name + ": Bad format (quote)"); return }

	quote, ok := x[0].(map[string]interface{})
	if !ok { fmt.Println(name + ": Bad format (quote[0])"); return }

	x, ok = indicators["adjclose"].([]interface{})
	if !ok { fmt.Println(name + ": Bad format (adjclose)"); return }

	a, ok := x[0].(map[string]interface{})
	if !ok { fmt.Println(name + ": Bad format (adjclose[0])"); return }

	adjclose, ok := a["adjclose"].([]interface{})
	if !ok { fmt.Println(name + ": Bad format (adjclose[adjclose])"); return }

	open, ok := quote["open"].([]interface{})
	if !ok { fmt.Println(name + ": Bad format (open)"); return }

	high, ok := quote["high"].([]interface{})
	if !ok { fmt.Println(name + ": Bad format (high)"); return }

	low, ok := quote["low"].([]interface{})
	if !ok { fmt.Println(name + ": Bad format (low)"); return }

	close, ok := quote["close"].([]interface{})
	if !ok { fmt.Println(name + ": Bad format (close)"); return }

	volume, ok := quote["volume"].([]interface{})
	if !ok { fmt.Println(name + ": Bad format (volume)"); return }

	out, err := os.Create(name + "-history.csv")
	if err != nil {
		fmt.Printf("%s: %s\n", name, err)
		os.Exit(-1)
	}
	defer out.Close()

	out.WriteString("Date,Open,High,Low,Close,Adj Close,Volume\n")
	for k, v := range timestamp {
		unixTime := time.Unix(int64(v.(float64)),0)
		if open[k] == nil {
			continue
		}
		out.WriteString(fmt.Sprintf("%s,%f,%f,%f,%f,%f,%d\n",
			unixTime.Format("2006-01-02"),
			open[k],high[k],low[k],close[k],
			adjclose[k],int(volume[k].(float64))))
	}
	if dividends != nil {
		out, err = os.Create(name + "-dividend.csv")
		if err != nil {
			fmt.Printf("%s: %s\n", name, err)
			os.Exit(-1)
		}

		defer out.Close()

		out.WriteString("Date,Dividends\n")
		for _,v := range dividends {
			v2, ok := v.(map[string]interface{})
			if !ok {
				fmt.Println(name + ": Bad format (dividends)")
				return
			}
			amount, ok := v2["amount"]
			if !ok {
				fmt.Println(name + ": Bad format (dividend amount)")
				return
			}
			t, ok := v2["date"].(float64)
			if !ok {
				fmt.Println(name + ": Bad format (dividend date)")
				return
			}
			unixTime := time.Unix(int64(t),0)
			out.WriteString(fmt.Sprintf("%s,%f\n",
				unixTime.Format("2006-01-02"),
				amount.(float64)))
		}
	}

	if capgains != nil {
		out, err = os.Create(name + "-capitalGain.csv")
		if err != nil {
			fmt.Printf("%s: %s\n", name, err)
			os.Exit(-1)
		}

		defer out.Close()

		out.WriteString("Date,Capital Gains\n")
		for _,v := range capgains {
			v2, ok := v.(map[string]interface{})
			if !ok {
				fmt.Println(name + ": Bad format (capital gains)")
				return
			}
			amount, ok := v2["amount"]
			if !ok {
				fmt.Println(name + ": Bad format (capital gains amount)")
				return
			}
			t, ok := v2["date"].(float64)
			if !ok {
				fmt.Println(name + ": Bad format (capital gains date)")
				return
			}
			unixTime := time.Unix(int64(t),0)
			out.WriteString(fmt.Sprintf("%s,%f\n",
				unixTime.Format("2006-01-02"),
				amount.(float64)))
		}
	}

	if splits != nil {
		out, err = os.Create(name + "-split.csv")
		if err != nil {
			fmt.Printf("%s: %s\n", name, err)
			os.Exit(-1)
		}

		defer out.Close()

		out.WriteString("Date,Stock Splits\n")
		for _,v := range splits {
			v2, ok := v.(map[string]interface{})
			if !ok {
				fmt.Println(name + ": Bad format (splits)")
				return
			}
			splitRatio, ok := v2["splitRatio"]
			if !ok {
				fmt.Println(name + ": Bad format (split ratio)")
				return
			}
			t, ok := v2["date"].(float64)
			if !ok {
				fmt.Println(name + ": Bad format (split date)")
				return
			}
			unixTime := time.Unix(int64(t),0)
			out.WriteString(fmt.Sprintf("%s,%s\n",
				unixTime.Format("2006-01-02"),
				splitRatio))
		}
	}
}