¿Que derechos tengo si finalizo contrato estando de baja por riesgo en el trabajo por embarazo?

Hola! Llevo trabajando en la misma empresa más de un año y en este último contrato me quede embarazada, cogí la baja por riesgo en el trabajo en la semana 30, dentro de 1mes me toca dar a luz pero ahora mi contrato finaliza y no sé si estando de baja por riesgo finalizo o no? Tengo derecho a algo?
Mis jefes quieren que vuelva a trabajar con ellos, ¿me correspondería lactancia o al ser un nuevo contrato no?
Estoy un poco perdida, ¿alguien me puede ayudar e informar de mis derechos?
Gracias.