#!/usr/bin/env python3

'''

Copyright (C) 2012-2013 Povilas Kanapickas <povilas@radix.lt>

This file is part of cppreference-doc

This program is free software: you can redistribute it and/or modify

it under the terms of the GNU General Public License as published by

the Free Software Foundation, either version 3 of the License, or

(at your option) any later version.

This program is distributed in the hope that it will be useful,

but WITHOUT ANY WARRANTY; without even the implied warranty of

MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

GNU General Public License for more details.

You should have received a copy of the GNU General Public License

along with this program. If not, see http://www.gnu.org/licenses/.

'''

import sys

import lxml.etree as e

from link_map import LinkMap

if len(sys.argv) != 3:

print('''Please provide the following 2 argument:

* the file name of the source file

* the file name of the destination file

''')

in_fn = sys.argv[1]

out_fn = sys.argv[2]

mapping = LinkMap()

mapping.read('output/link-map.xml')

root = e.parse(in_fn)

el_mod = root.xpath('//*[@link]')

for el in el_mod:

link = el.get('link')

target = mapping.get_dest(link)

if target is None:

print('Could not find ' + link + ' in mapping')

target = '404'

el.set('link', target)

out_f = open(out_fn, 'wb')

out_f.write(e.tostring(root, encoding='utf-8', pretty_print=True,

xml_declaration=True))

out_f.close()