74 lines
1.8 KiB
Go
74 lines
1.8 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import "context"
|
||
|
|
import sw "./go-client"
|
||
|
|
|
||
|
|
type State struct {
|
||
|
|
HomeId string
|
||
|
|
BridgeId string
|
||
|
|
Api *sw.DefaultApiService
|
||
|
|
Auth context.Context
|
||
|
|
NameForRoom map[string]string
|
||
|
|
RoomForName map[string]string
|
||
|
|
RoomForModule map[string]string
|
||
|
|
ModulesForRoom map[string][]string
|
||
|
|
NameForModule map[string]string
|
||
|
|
ModuleForName map[string]string
|
||
|
|
ModuleStatus map[string]sw.ModuleStatus
|
||
|
|
RoomStatus map[string]sw.RoomStatus
|
||
|
|
}
|
||
|
|
|
||
|
|
func fetchData(tokenFile string) *State {
|
||
|
|
token := refreshToken(tokenFile)
|
||
|
|
|
||
|
|
var state = &State{
|
||
|
|
Api: sw.NewAPIClient(sw.NewConfiguration()).DefaultApi,
|
||
|
|
BridgeId: BridgeId,
|
||
|
|
Auth: context.WithValue(context.Background(), sw.ContextAccessToken, token.AccessToken),
|
||
|
|
NameForRoom: make(map[string]string),
|
||
|
|
RoomForName: make(map[string]string),
|
||
|
|
RoomForModule: make(map[string]string),
|
||
|
|
ModulesForRoom: make(map[string][]string),
|
||
|
|
NameForModule: make(map[string]string),
|
||
|
|
ModuleForName: make(map[string]string),
|
||
|
|
ModuleStatus: make(map[string]sw.ModuleStatus),
|
||
|
|
RoomStatus: make(map[string]sw.RoomStatus),
|
||
|
|
}
|
||
|
|
|
||
|
|
r, _, err := state.Api.HomesData(state.Auth)
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
state.HomeId = r.Body.Homes[0].Id
|
||
|
|
|
||
|
|
for _, r := range r.Body.Homes[0].Rooms {
|
||
|
|
state.NameForRoom[r.Id] = r.Name
|
||
|
|
state.NameForRoom[r.Name] = r.Id
|
||
|
|
for _, m := range r.Modules {
|
||
|
|
state.RoomForModule[m] = r.Id
|
||
|
|
}
|
||
|
|
state.ModulesForRoom[r.Id] = r.Modules
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, m := range r.Body.Homes[0].Modules {
|
||
|
|
state.NameForModule[m.Id] = m.Name
|
||
|
|
state.ModuleForName[m.Name] = m.Id
|
||
|
|
}
|
||
|
|
|
||
|
|
r2, _, err := state.Api.HomeStatus(state.Auth, sw.Body{HomeId: state.HomeId})
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, m := range r2.Body.Home.Modules {
|
||
|
|
state.ModuleStatus[m.Id] = m
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, r := range r2.Body.Home.Rooms {
|
||
|
|
state.RoomStatus[r.Id] = r
|
||
|
|
}
|
||
|
|
|
||
|
|
return state
|
||
|
|
}
|