added team update pass and get event id

Signed-off-by: Ahmet Turkmen <[email protected]>

database/query.go

@@ -39,6 +39,7 @@ var (
 
 	UpdateEventLastaccessedDate = "UPDATE team SET last_access = $2 WHERE tag = $1"
 	UpdateTeamSolvedChl         = "UPDATE team SET solved_challenges = $2 WHERE tag = $1"
+	UpdateTeamPassword          = "UPDATE team SET password = $1 WHERE tag = $2 and event_id = $3"
 
 	QuerySolvedChls = "SELECT solved_challenges FROM team WHERE tag=$1"
 	QueryEventTable = "SELECT * FROM event"

database/store.go

@@ -47,6 +47,8 @@ type Store interface {
 	SetEventStatus(*pb.SetEventStatusRequest) (int32, error)
 	UpdateTeamSolvedChallenge(*pb.UpdateTeamSolvedChallengeRequest) (string, error)
 	UpdateTeamLastAccess(*pb.UpdateTeamLastAccessRequest) (string, error)
+	UpdateTeamPassword(in *pb.UpdateTeamPassRequest) error
+	GetEventID(in *pb.GetEventIDReq) (int32, error)
 	UpdateCloseEvent(*pb.UpdateEventRequest) (string, error)
 }
 
@@ -243,6 +245,23 @@ func (s *store) UpdateTeamSolvedChallenge(in *pb.UpdateTeamSolvedChallengeReques
 	return OK, nil
 }
 
+func (s *store) UpdateTeamPassword(in *pb.UpdateTeamPassRequest) error {
+	_, err := s.db.Exec(UpdateTeamPassword, in.EncryptedPass, in.TeamID, in.EventID)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func (s *store) GetEventID(in *pb.GetEventIDReq) (int32, error) {
+
+	var eventId int32
+	if err := s.db.QueryRow(QueryEventId, in.EventTag).Scan(&eventId); err != nil && !strings.Contains(err.Error(), "no rows in result set") {
+		return -1, err
+	}
+	return eventId, nil
+}
+
 func (s *store) UpdateTeamLastAccess(in *pb.UpdateTeamLastAccessRequest) (string, error) {
 	s.m.Lock()
 	defer s.m.Unlock()

util/util.go

@@ -188,6 +188,24 @@ func (s server) UpdateTeamLastAccess(ctx context.Context, in *pb.UpdateTeamLastA
 	return &pb.UpdateResponse{Message: result}, nil
 }
 
+func (s server) UpdateTeamPassword(ctx context.Context, req *pb.UpdateTeamPassRequest) (*pb.UpdateResponse, error) {
+	if err := s.store.UpdateTeamPassword(req); err != nil {
+		log.Printf("ERR: Error Update team %s password: %s", req.TeamID, err.Error())
+		return &pb.UpdateResponse{ErrorMessage: err.Error()}, err
+	}
+
+	return &pb.UpdateResponse{}, nil
+}
+
+func (s server) GetEventID(ctx context.Context, req *pb.GetEventIDReq) (*pb.GetEventIDResp, error) {
+	id, err := s.store.GetEventID(req)
+	if err != nil {
+		return &pb.GetEventIDResp{}, err
+	}
+
+	return &pb.GetEventIDResp{EventID: id}, nil
+}
+
 func GetCreds(conf *model.Config) (credentials.TransportCredentials, error) {
 	log.Printf("Preparing credentials for RPC")