REFACTOR THIS ENTIRE CODEBASE IN RUST, DON'T MAKE ANY MISTAKES, THINK HARD