summaryrefslogtreecommitdiff
path: root/libraries/collectors-helper/app/src/com/android/helpers/AppVersionHelper.java
blob: f8d2678ee3915bdf8aa6042ce046d9afc13b5817 (plain)
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
79
80
81
82
83
84
85
86
87
88
/*
 * Copyright (C) 2021 The Android Open Source Project
 *
 * 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
 *
 *      http://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 com.android.helpers;

import static com.android.helpers.MetricUtility.constructKey;

import android.content.Context;
import android.content.pm.PackageInfo;
import android.content.pm.PackageManager;
import android.util.Log;

import androidx.test.core.app.ApplicationProvider;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * This is a collector helper to use PackageManager to get app package versions.
 */
public class AppVersionHelper implements ICollectorHelper<Long> {

    private static final String TAG = AppVersionHelper.class.getSimpleName();
    private static final String METRIC_PREFIX = "app-version-code-long";

    private Context context;
    private PackageManager packageManager;

    private String[] mPackageNames = {};

    public void setUp(String... packageNames) {
        context = ApplicationProvider.getApplicationContext().getApplicationContext();
        packageManager = context.getPackageManager();
        mPackageNames = packageNames;
    }

    @Override
    public boolean startCollecting() {
        return true;
    }

    @Override
    public Map<String, Long> getMetrics() {
        Map<String, Long> metrics = new HashMap<>();
        if (mPackageNames == null || mPackageNames.length == 0 || mPackageNames[0].length() == 0) {
            // If no package specified, collects all.
            // Additional option flags is set to 0, which is simply default.
            List<PackageInfo> pkgInfos = packageManager.getInstalledPackages(0);
            for (PackageInfo pkgInfo : pkgInfos) {
                metrics.put(constructKey(METRIC_PREFIX, pkgInfo.packageName),
                        pkgInfo.getLongVersionCode());
            }
        } else {
            for (String pkg : mPackageNames) {
                try {
                    // Additional option flags is set to 0, which is simply default.
                    PackageInfo pkgInfo = packageManager.getPackageInfo(pkg, 0);
                    metrics.put(constructKey(METRIC_PREFIX, pkg),
                            pkgInfo.getLongVersionCode());
                    Log.d(TAG, "Found app version for package name " + pkg);
                } catch (PackageManager.NameNotFoundException exception) {
                    Log.e(TAG, "Can't find package name " + pkg);
                    continue;
                }
            }
        }
        return metrics;
    }

    @Override
    public boolean stopCollecting() {
        return true;
    }
}