1#
2# Copyright 2022 The Android Open Source Project
3#
4# Licensed under the Apache License, Version 2.0 (the "License");
5# you may not use this file except in compliance with the License.
6# You may obtain a copy of the License at
7#
8#      http://www.apache.org/licenses/LICENSE-2.0
9#
10# Unless required by applicable law or agreed to in writing, software
11# distributed under the License is distributed on an "AS IS" BASIS,
12# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13# See the License for the specific language governing permissions and
14# limitations under the License.
15#
16
17import os
18import sys
19
20def should_insert_before(line, lib_name):
21    prefix = "- title: \"androidx."
22    return line.startswith(prefix) and line[len(prefix):] > lib_name
23
24def exec(path):
25    lib_name = path.split("/")[-2]
26    toc_file = "reference/androidx/_toc.yaml"
27    with open(toc_file, "r") as file:
28        lines = file.readlines()
29    idx = next(
30        (idx for idx, line in enumerate(lines) if should_insert_before(line, lib_name)),
31        len(lines)
32    )
33    lines.insert(idx, f"- include: /{path}\n")
34    with open(toc_file, "w") as file:
35        file.write("".join(lines))
36
37exec(sys.argv[1])