-
Notifications
You must be signed in to change notification settings - Fork 114
/
scorecardcompat.go
78 lines (68 loc) · 2.36 KB
/
scorecardcompat.go
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
77
78
// Copyright 2023 Criticality Score Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package localworker
import (
"context"
"fmt"
"time"
"github.com/ossf/scorecard/v4/clients"
"github.com/ossf/scorecard/v4/cron/data"
"github.com/ossf/scorecard/v4/cron/worker"
"google.golang.org/protobuf/encoding/protojson"
"google.golang.org/protobuf/proto"
"google.golang.org/protobuf/types/known/timestamppb"
"github.com/ossf/criticality_score/v2/cmd/collect_signals/vcs"
)
var headSHA = clients.HeadSHA
func writeMetadata(ctx context.Context, bucketURL string, lastShard int32, t time.Time) error {
// Populate `.shard_metadata` file.
metadata := data.ShardMetadata{
NumShard: new(int32),
ShardLoc: new(string),
CommitSha: new(string),
}
*metadata.NumShard = lastShard + 1
*metadata.ShardLoc = bucketURL + "/" + data.GetBlobFilename("", t)
*metadata.CommitSha = vcs.CommitID()
metadataJSON, err := protojson.Marshal(&metadata)
if err != nil {
return fmt.Errorf("protojson marshal: %w", err)
}
if err := data.WriteToBlobStore(ctx, bucketURL, data.GetShardMetadataFilename(t), metadataJSON); err != nil {
return fmt.Errorf("writing to blob store: %w", err)
}
return nil
}
func makeRequest(repos []string, shardNum int32, t time.Time) *data.ScorecardBatchRequest {
req := &data.ScorecardBatchRequest{
JobTime: timestamppb.New(t),
ShardNum: proto.Int32(shardNum),
}
for _, repo := range repos {
localRepo := repo
req.Repos = append(req.GetRepos(), &data.Repo{
Url: proto.String(localRepo),
Commit: proto.String(headSHA),
Metadata: []string{},
})
}
return req
}
func resultExists(ctx context.Context, sbr *data.ScorecardBatchRequest, bucketURL string) (bool, error) {
exists, err := data.BlobExists(ctx, bucketURL, worker.ResultFilename(sbr))
if err != nil {
return false, fmt.Errorf("blob exists check: %w", err)
}
return exists, nil
}