package pwd

import (
	"github.com/play-with-docker/play-with-docker/pwd/types"
	"github.com/play-with-docker/play-with-docker/storage"
)

func (p *pwd) UserNewLoginRequest(providerName string) (*types.LoginRequest, error) {
	req := &types.LoginRequest{Id: p.generator.NewId(), Provider: providerName}
	if err := p.storage.LoginRequestPut(req); err != nil {
		return nil, err
	}
	return req, nil
}

func (p *pwd) UserGetLoginRequest(id string) (*types.LoginRequest, error) {
	if req, err := p.storage.LoginRequestGet(id); err != nil {
		return nil, err
	} else {
		return req, nil
	}
}

func (p *pwd) UserLogin(loginRequest *types.LoginRequest, user *types.User) (*types.User, error) {
	if err := p.storage.LoginRequestDelete(loginRequest.Id); err != nil {
		return nil, err
	}
	u, err := p.storage.UserFindByProvider(user.Provider, user.ProviderUserId)
	if err != nil {
		if storage.NotFound(err) {
			user.Id = p.generator.NewId()
		} else {
			return nil, err
		}
	} else {
		user.Id = u.Id
	}
	if err := p.storage.UserPut(user); err != nil {
		return nil, err
	}

	return user, nil
}
func (p *pwd) UserGet(id string) (*types.User, error) {
	if user, err := p.storage.UserGet(id); err != nil {
		return nil, err
	} else {
		return user, nil
	}
}