semaphore/services/server/inventory_svc.go at 2a0dc9d091882d05657b43e5a7157b0b8d00d7e4 · semaphoreui/semaphore · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
package server

import "github.com/semaphoreui/semaphore/db"

type InventoryService interface {
	GetInventory(projectID int, inventoryID int) (inventory db.Inventory, err error)
}

func NewInventoryService(
	accessKeyRepo db.AccessKeyManager,
	repositoryRepo db.RepositoryManager,
	inventoryRepo db.InventoryManager,
	encryptionService AccessKeyEncryptionService,
) InventoryService {
	return &InventoryServiceImpl{
		accessKeyRepo:     accessKeyRepo,
		repositoryRepo:    repositoryRepo,
		encryptionService: encryptionService,
		inventoryRepo:     inventoryRepo,
	}
}

type InventoryServiceImpl struct {
	accessKeyRepo     db.AccessKeyManager
	repositoryRepo    db.RepositoryManager
	encryptionService AccessKeyEncryptionService
	inventoryRepo     db.InventoryManager
}

func (s *InventoryServiceImpl) GetInventory(projectID int, inventoryID int) (inventory db.Inventory, err error) {
	inventory, err = s.inventoryRepo.GetInventory(projectID, inventoryID)
	if err != nil {
		return
	}

	err = s.fillInventory(&inventory)
	return
}

func (s *InventoryServiceImpl) fillInventory(inventory *db.Inventory) (err error) {
	if inventory.SSHKeyID != nil {
		inventory.SSHKey, err = s.accessKeyRepo.GetAccessKey(inventory.ProjectID, *inventory.SSHKeyID)
	}

	if err != nil {
		return
	}

	if inventory.BecomeKeyID != nil {
		inventory.BecomeKey, err = s.accessKeyRepo.GetAccessKey(inventory.ProjectID, *inventory.BecomeKeyID)
	}

	if err != nil {
		return
	}

	if inventory.RepositoryID != nil {
		var repo db.Repository
		repo, err = s.repositoryRepo.GetRepository(inventory.ProjectID, *inventory.RepositoryID)
		if err != nil {
			return
		}

		if repo.SSHKeyID != nil {
			err = s.encryptionService.DeserializeSecret(repo.SSHKey)
		}

		if err != nil {
			return
		}

		inventory.Repository = &repo
	}

	return
}