hostsmanager.go

package main

import (
	"errors"
	"strings"

	"github.com/openatx/atx-server/proto"
)

func deviceQueryToUdid(query string) (udid string, err error) {
	if strings.HasPrefix(query, "ip:") {
		infos := db.DeviceFindAll(proto.DeviceInfo{IP: query[3:], Present: newBool(true)})
		return extractUdidFromInfos(infos)
	}
	return query, nil
}

func extractUdidFromInfos(infos []proto.DeviceInfo) (udid string, err error) {
	if len(infos) == 0 {
		return "", errors.New("not found")
	}
	if len(infos) > 1 {
		return "", errors.New("too many matches")
	}
	return infos[0].Udid, nil
}

// // TODO: need to delete bellow
// type HostsManager struct {
// 	maps map[string]*proto.DeviceInfo
// 	mu   sync.RWMutex
// }

// func NewHostsManager() *HostsManager {
// 	return &HostsManager{
// 		maps: make(map[string]*proto.DeviceInfo),
// 	}
// }

// func (t *HostsManager) Lookup(query string) *proto.DeviceInfo {
// 	if strings.HasPrefix(query, "ip:") {
// 		return t.FromIP(query[3:])
// 	}
// 	return t.FromUdid(query)
// }

// // A return value of nil indicates not found
// func (t *HostsManager) FromIP(ip string) *proto.DeviceInfo {
// 	t.mu.Lock()
// 	defer t.mu.Unlock()
// 	for _, info := range t.maps {
// 		if info.IP == ip {
// 			return info
// 		}
// 	}
// 	return nil
// }

// // A return value of nil indicates not found
// func (t *HostsManager) FromUdid(udid string) *proto.DeviceInfo {
// 	t.mu.Lock()
// 	defer t.mu.Unlock()
// 	return t.maps[udid]
// }

// func (t *HostsManager) AddFromDeviceInfo(devInfo *proto.DeviceInfo) {
// 	t.mu.Lock()
// 	defer t.mu.Unlock()
// 	udid := devInfo.Udid
// 	if info, ok := t.maps[udid]; ok {
// 		info.IP = devInfo.IP
// 		info.ConnectionCount++
// 	} else {
// 		devInfo.ConnectionCount = 1
// 		t.maps[udid] = devInfo
// 	}
// }

// func (t *HostsManager) Remove(udid string) {
// 	t.mu.Lock()
// 	defer t.mu.Unlock()
// 	if info, ok := t.maps[udid]; ok {
// 		info.ConnectionCount--
// 		if info.ConnectionCount <= 0 {
// 			delete(t.maps, udid)
// 		}
// 	}
// }

// func (t *HostsManager) Acquire(query string) error {
// 	info := t.Lookup(query)
// 	if info == nil {
// 		return errors.New("device not found")
// 	}
// 	if info.Reserved != "" {
// 		return errors.New("device already reserved")
// 	}
// 	info.Reserved = "hzsunshx"
// 	return nil
// }

// func (t *HostsManager) Release(query string) error {
// 	info := t.Lookup(query)
// 	if info == nil {
// 		return errors.New("device not found")
// 	}
// 	info.Reserved = ""
// 	return nil
// }

// func (t *HostsManager) Random() (devInfo *proto.DeviceInfo, err error) {
// 	t.mu.Lock()
// 	defer t.mu.Unlock()
// 	for _, info := range t.maps {
// 		if info.Ready != nil && *info.Ready == true && info.Reserved == "" {
// 			info.Reserved = "random"
// 			return info, nil
// 		}
// 	}
// 	return nil, errors.New("no devices avaliable")
// }