File: json_generator.py

package info (click to toggle)
chromium 120.0.6099.224-1~deb11u1
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 6,112,112 kB
  • sloc: cpp: 32,907,025; ansic: 8,148,123; javascript: 3,679,536; python: 2,031,248; asm: 959,718; java: 804,675; xml: 617,256; sh: 111,417; objc: 100,835; perl: 88,443; cs: 53,032; makefile: 29,579; fortran: 24,137; php: 21,162; tcl: 21,147; sql: 20,809; ruby: 17,735; pascal: 12,864; yacc: 8,045; lisp: 3,388; lex: 1,323; ada: 727; awk: 329; jsp: 267; csh: 117; exp: 43; sed: 37
file content (77 lines) | stat: -rw-r--r-- 2,558 bytes parent folder | download | duplicates (6)
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
# Copyright 2022 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import json

from style_variable_generator.base_generator import Color, BaseGenerator
from style_variable_generator.model import Modes, VariableType
from .css_generator import CSSStyleGenerator

SYS_SUBSTR = "sys"
REF_SUBSTR = "ref"
TOKENS_SUBSTR = "tokens"


def TokenArrayPropertyTitle(property):
    return "%s_%s" % (property, TOKENS_SUBSTR)


# This generator is used to provide a raw JSON dump of token names, css
# variables and, in the case of sys tokens, a formula of how they were
# calculated.
class JSONStyleGenerator(CSSStyleGenerator):
    @staticmethod
    def GetName():
        return 'json'

    def __init__(self):
        super().__init__()
        # We want the output JSON to include the formula for how a color was
        # calculated, which means we don't want to resolve blended colors down
        # to RGBA values.
        self.resolve_blended_colors = False

    def GetParameters(self):
        color_model = self.model.colors
        ref_tokens = []
        sys_tokens = []
        tokens = {}
        for name, mode_values in color_model.items():
            color_item = {
                "token_name": name,
                "css_variable": self.ToCSSVarName(name),
            }
            # Ref tokens should not contain a dark mode entry, as we default to
            # light mode. Sys tokens that do not have generate_per_mode will
            # have an entry for both light and dark, and we want to include the
            # formula for these.
            if SYS_SUBSTR in name:
                color_item["mode_values"] = {
                    Modes.LIGHT:
                    color_model.Resolve(name, Modes.LIGHT).GetFormula(),
                    Modes.DARK:
                    color_model.Resolve(name, Modes.DARK).GetFormula()
                }
                sys_tokens.append(color_item)
            else:
                ref_tokens.append(color_item)
        return {
            "tokens":
            json.dumps(
                {
                    TokenArrayPropertyTitle(REF_SUBSTR): ref_tokens,
                    TokenArrayPropertyTitle(SYS_SUBSTR): sys_tokens,
                },
                indent=4)
        }

    def GetGlobals(self):
        return {}

    def GetFilters(self):
        return {}

    def Render(self):
        return self.ApplyTemplate(self, "templates/json_generator.tmpl",
                                  self.GetParameters())