[SOLUCIONADO] Revision de codigo button change and hide inputs on click

[Ferdinand1945]

Hola Amigos, acabo de hacer una funcion en jquery para que un boton cambie o mas bien para que se oculte un boton "save" cuando se hace click sobre el y deshabilite todos los imputs de un formulario, ademas muestra un boton "edit" (que este habilitaria todos los imputs de nuevo) cuando "save" esta oculto. Todo funciona de maravilla, pero siento que mi codigo es horrible. Alguien me podria decir si hay una manera mas "limpia" o mas "pro" de hacerlo?

Esta es mi funcion

Código:

 buttonSave: function() {
            	$(':input').prop('readonly', true);
            	$('#save').addClass("hide"); 
            	$('#edit').removeClass('hide');
            	
            	if($('#save').hasClass('hide') && $('#edit').is(":visible") === true){
            	 $('#edit').on('click', function(){
         			$(':input').prop('readonly', false);
         			$('#edit').addClass('hide');
         			$('#save').removeClass('hide');
         		 });
            	} else if ($('#edit').is(':hidden')){
                    $(':input').prop('readonly', true);
                } 
            }

Mychas gracias a todos de antemano! (reaclaro que todo funciona de maravilla y no me tira ningun error, pero el siento que mi codigo es muy pobre y redundante)

[Ferdinand1945]

creo que no hay una manera "mas correcta" que esta para hacerlo a no ser que cambie la arquitectura del codigo... gracias a todos los que hecharon un vistazo!