Редирект в NGINX с get параметром

Нужно сделать такой редирект

https://site.ru/?page=12               -> https://site.ru/12
https://site.ru/?arg=21&page=12&arg2=3 -> https://site.ru/12?arg=21&arg2=3
https://site.ru/category/?page=12      -> https://site.ru/category/12

Предполагаю что нужно как то так

location (что тут д.б.?) {
      if ( $arg_page = (\d+) ) {
      rewrite ^(.*)$ /$uri/$1 permanent;
    }
  }

Ничего не вышло. Нужно учесть еще другие аргументы.