#!/usr/bin/env python3
import pathlib
import sys

import bpy


ARMATURE_NAME = "WebGapArmatureCollectionShowAllArmature"
OBJECT_NAME = "WebGapArmatureCollectionShowAllObject"
FIRST_COLLECTION = "WebGapArmatureCollectionShowAllFirst"
HIDDEN_COLLECTION = "WebGapArmatureCollectionShowAllHidden"
LAST_COLLECTION = "WebGapArmatureCollectionShowAllLast"
FIRST_BONE = "WebGapArmatureCollectionShowAllFirstBone"
HIDDEN_BONE = "WebGapArmatureCollectionShowAllHiddenBone"
LAST_BONE = "WebGapArmatureCollectionShowAllLastBone"


def main():
    arguments = sys.argv[sys.argv.index("--") + 1 :]
    if len(arguments) != 1:
        raise SystemExit("usage: blender -b --factory-startup --python M16-GAP-00229.py -- OUTPUT")
    bpy.ops.wm.read_factory_settings(use_empty=True)
    armature = bpy.data.armatures.new(ARMATURE_NAME)
    obj = bpy.data.objects.new(OBJECT_NAME, armature)
    bpy.context.scene.collection.objects.link(obj)
    first = armature.collections.new(FIRST_COLLECTION)
    hidden = armature.collections.new(HIDDEN_COLLECTION)
    last = armature.collections.new(LAST_COLLECTION)
    bpy.context.view_layer.objects.active = obj
    obj.select_set(True)
    bpy.ops.object.mode_set(mode="EDIT")
    edit_bones = {}
    for name, x in ((FIRST_BONE, 0.0), (HIDDEN_BONE, 1.0), (LAST_BONE, 2.0)):
        bone = armature.edit_bones.new(name)
        bone.head = (x, 0.0, 0.0)
        bone.tail = (x, 1.0, 0.0)
        edit_bones[name] = bone
    bpy.ops.object.mode_set(mode="OBJECT")
    first.assign(armature.bones[FIRST_BONE])
    hidden.assign(armature.bones[HIDDEN_BONE])
    last.assign(armature.bones[LAST_BONE])
    hidden.is_visible = False
    bpy.ops.wm.save_as_mainfile(
        filepath=str(pathlib.Path(arguments[0]).resolve()), check_existing=False, compress=True
    )


if __name__ == "__main__":
    main()