Lots of things, mainly models and CSV parser
This commit is contained in:
parent
d63d1a4513
commit
901b42d801
4 changed files with 147 additions and 1 deletions
53
model/birthdays.go
Normal file
53
model/birthdays.go
Normal file
|
@ -0,0 +1,53 @@
|
||||||
|
package model
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Birthday struct {
|
||||||
|
Name string
|
||||||
|
Email string
|
||||||
|
Phone string
|
||||||
|
YearOfBirth int
|
||||||
|
MonthOfBirth int
|
||||||
|
DayOfBirth int
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewBirthdayFromRecord(record []string) (*Birthday, error) {
|
||||||
|
if len(record) != 4 {
|
||||||
|
return nil, fmt.Errorf("invalid length %d for record", len(record))
|
||||||
|
}
|
||||||
|
|
||||||
|
dateComponents := strings.Split(strings.TrimSpace(record[3]), "/")
|
||||||
|
if len(dateComponents) != 3 {
|
||||||
|
return nil, fmt.Errorf("invalid date format in %q", strings.TrimSpace(record[3]))
|
||||||
|
}
|
||||||
|
|
||||||
|
dayOfBirth, err := strconv.Atoi(dateComponents[0])
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("error parsing day of birth from %q: %w", dateComponents[0], err)
|
||||||
|
}
|
||||||
|
|
||||||
|
monthOfBirth, err := strconv.Atoi(dateComponents[1])
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("error parsing month of birth from %q: %w", dateComponents[1], err)
|
||||||
|
}
|
||||||
|
|
||||||
|
yearOfBirth, err := strconv.Atoi(dateComponents[2])
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("error parsing year of birth from %q: %w", dateComponents[2], err)
|
||||||
|
}
|
||||||
|
|
||||||
|
b := &Birthday{
|
||||||
|
Name: strings.TrimSpace(record[0]),
|
||||||
|
Email: strings.TrimSpace(record[1]),
|
||||||
|
Phone: strings.TrimSpace(record[2]),
|
||||||
|
YearOfBirth: yearOfBirth,
|
||||||
|
MonthOfBirth: monthOfBirth,
|
||||||
|
DayOfBirth: dayOfBirth,
|
||||||
|
}
|
||||||
|
|
||||||
|
return b, nil
|
||||||
|
}
|
44
model/birthdays_test.go
Normal file
44
model/birthdays_test.go
Normal file
|
@ -0,0 +1,44 @@
|
||||||
|
package model
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/stretchr/testify/require"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestNewBirthdayFromRecord(t *testing.T) {
|
||||||
|
t.Run("should correctly parse different records", func(t *testing.T) {
|
||||||
|
testCases := []struct {
|
||||||
|
Record []string
|
||||||
|
ExpectedName string
|
||||||
|
ExpectedEmail string
|
||||||
|
ExpectedPhone string
|
||||||
|
ExpectedYearOfBirth int
|
||||||
|
ExpectedMonthOfBirth int
|
||||||
|
ExpectedDayOfBirth int
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
Record: []string{"John Doe ", "john@doe.com", " 123456789", "17/04/2192"},
|
||||||
|
ExpectedName: "John Doe",
|
||||||
|
ExpectedEmail: "john@doe.com",
|
||||||
|
ExpectedPhone: "123456789",
|
||||||
|
ExpectedYearOfBirth: 2192,
|
||||||
|
ExpectedMonthOfBirth: 4,
|
||||||
|
ExpectedDayOfBirth: 17,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tc := range testCases {
|
||||||
|
t.Run(tc.ExpectedName, func(t *testing.T) {
|
||||||
|
b, err := NewBirthdayFromRecord(tc.Record)
|
||||||
|
require.NoError(t, err)
|
||||||
|
require.Equal(t, tc.ExpectedName, b.Name)
|
||||||
|
require.Equal(t, tc.ExpectedEmail, b.Email)
|
||||||
|
require.Equal(t, tc.ExpectedPhone, b.Phone)
|
||||||
|
require.Equal(t, tc.ExpectedYearOfBirth, b.YearOfBirth)
|
||||||
|
require.Equal(t, tc.ExpectedMonthOfBirth, b.MonthOfBirth)
|
||||||
|
require.Equal(t, tc.ExpectedDayOfBirth, b.DayOfBirth)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
42
parser/csv_parser.go
Normal file
42
parser/csv_parser.go
Normal file
|
@ -0,0 +1,42 @@
|
||||||
|
package parser
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/csv"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"os"
|
||||||
|
|
||||||
|
"git.ctrlz.es/mgdelacroix/birthdaybot/model"
|
||||||
|
)
|
||||||
|
|
||||||
|
func ParseCSV(path string) ([]*model.Birthday, error) {
|
||||||
|
f, err := os.Open(path)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer f.Close()
|
||||||
|
|
||||||
|
reader := csv.NewReader(f)
|
||||||
|
|
||||||
|
birthdays := []*model.Birthday{}
|
||||||
|
lineno := 0
|
||||||
|
for {
|
||||||
|
lineno++
|
||||||
|
record, err := reader.Read()
|
||||||
|
if errors.Is(err, io.EOF) {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("error reading lineno %d: %w", lineno, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
birthday, err := model.NewBirthdayFromRecord(record)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("error parsing record for line %d into birthday: %w", lineno, err)
|
||||||
|
}
|
||||||
|
birthdays = append(birthdays, birthday)
|
||||||
|
}
|
||||||
|
|
||||||
|
return birthdays, nil
|
||||||
|
}
|
|
@ -1,4 +1,11 @@
|
||||||
package server
|
package server
|
||||||
|
|
||||||
type Server struct {
|
import (
|
||||||
|
"git.ctrlz.es/mgdelacroix/birthdaybot/model"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Server struct{}
|
||||||
|
|
||||||
|
func New(config *model.Config) (*Server, error) {
|
||||||
|
return nil, nil
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue