csv.go

// package fcc
// Copyright (C) 2019 Lars Lehtonen KJ6CBE

// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.

// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.

package fcc

import (
	"strconv"
	"strings"
	"time"
)

const timeFormat = "01/02/2006 15:04:05"
const sep = "\u001e"

// ReadRecord reads a database record from boltdb and returns
// a MinimalLicense.
func ReadRecord(record []byte) MinimalLicense {
	var ml MinimalLicense
	split := strings.Split(string(record), sep)
	ml.Name = split[0]
	ml.Address = split[1]
	ml.City = split[2]
	ml.State = split[3]
	ml.ZIP = split[4]
	return ml
}

// Minimal returns a MinimalLicense.
func (l *License) Minimal() MinimalLicense {
	return MinimalLicense{
		Name:    l.LicName,
		Address: l.LicAddress,
		City:    l.LicCity,
		State:   l.LicState,
		ZIP:     l.LicZipCode,
	}
}

// DiskFormat converts a MinimalLicense to its on-disk format.
func (m *MinimalLicense) DiskFormat() []byte {
	join := strings.Join([]string{
		m.Name,
		m.Address,
		m.City,
		m.State,
		m.ZIP}, sep)
	return []byte(join)
}

// ParseLicense takes a license CSV parsed into a string and generates
// an *fcc.License{}.
func ParseLicense(line []string) (*License, error) {
	var err error
	var gd, ed, cd, lad time.Time

	lic := License{}
	lid, err := strconv.ParseUint(line[0], 10, 32)
	if err != nil {
		return &lic, err
	}
	lic.LicenseID = uint32(lid)

	lic.SourceSystem = line[1]

	lic.Callsign = line[2]
	if line[3] != "" {
		fid, err := strconv.ParseUint(line[3], 10, 32)
		if err != nil {
			return &lic, err
		}
		fid32 := uint32(fid)
		lic.FacilityID = &fid32
	}

	if line[4] != "" {
		frn, err := strconv.ParseUint(line[4], 10, 64)
		if err != nil {
			return &lic, err
		}
		lic.FRN = &frn
	}

	lic.LicName = line[5]
	lic.CommonName = line[6]
	lic.RadioServiceCode = line[7]
	lic.RadioServiceDesc = line[8]
	lic.RollupCategoryCode = line[9]
	lic.RollupCategoryDesc = line[10]

	if line[11] != "" {
		gd, err = parseTime(line[11])
		if err != nil {
			return &lic, err
		}
		lic.GrantDate = &gd
	}

	if line[12] != "" {
		ed, err = parseTime(line[12])
		if err != nil {
			return &lic, err
		}
		lic.ExpiredDate = &ed
	}

	if line[13] != "" {
		cd, err = parseTime(line[13])
		if err != nil {
			return &lic, err
		}
		lic.CancellationDate = &cd
	}

	if line[14] != "" {
		lad, err = parseTime(line[14])
		if err != nil {
			return &lic, err
		}
		lic.LastActionDate = &lad
	}
	lic.LicStatusCode = line[15]
	lic.LicStatusDesc = line[16]
	lic.RollupStatusCode = line[17]
	lic.RollupStatusDesc = line[18]
	lic.EntityTypeCode = line[19]
	lic.EntityTypeDesc = line[20]
	lic.RollupEntityCode = line[21]
	lic.RollupEntityDesc = line[22]
	lic.LicAddress = line[23]
	lic.LicCity = line[24]
	lic.LicState = line[25]
	lic.LicZipCode = line[26]

	return &lic, nil
}

func parseTime(ts string) (time.Time, error) {
	return time.Parse(timeFormat, ts)
}