How to make mesh show texture by datatype use shader?

[zzhgithub]

I am making a voxel game just like MC;
I gen a mesh and insert vertex attribute into。I plan to make a custom Material with 'wgsl';
but I do not know what next?
Is there any demo like the plan?
If I use demo https://github.com/bevyengine/bevy/blob/main/examples/shader/array_texture.rs and https://github.com/bevyengine/bevy/blob/main/assets/shaders/array_texture.wgsl.
What should I do to rewrite a @rertex function?

[zzhgithub]

Here is my demo But is wrong. Why?

https://github.com/zzhgithub/bevy_array_shader_demo

[zzhgithub]

https://github.com/zzhgithub/bevy_array_shader_demo it's fixed!