/*

 Copyright 2010 James Pether Sörling Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0 Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License. 

         $Id

 */

 package org.directdemocracyportal.democracy.service.governmentloader;

 import java.net.MalformedURLException;

 import java.util.ArrayList;

 import java.util.List;

 import org.apache.commons.logging.Log;

 import org.apache.commons.logging.LogFactory;

 import org.directdemocracyportal.democracy.model.core.Role;

 import org.directdemocracyportal.democracy.model.world.Organisation;

 import org.directdemocracyportal.democracy.model.world.OrganisationType;

 import org.directdemocracyportal.democracy.model.world.Person;

 import org.directdemocracyportal.democracy.model.world.Organisation.OrganisationRoles;

 import org.directdemocracyportal.democracy.service.OrganisationAlreadyExistException;

 import org.directdemocracyportal.democracy.service.PortalService;

 import com.gargoylesoftware.htmlunit.WebClient;

 import com.gargoylesoftware.htmlunit.html.HtmlPage;

 import com.gargoylesoftware.htmlunit.html.HtmlTable;

 import com.gargoylesoftware.htmlunit.html.HtmlTableCell;

 import com.gargoylesoftware.htmlunit.html.HtmlTableRow;

 /**

  * The Class SwedishParliamentOrganisationStructureImporterImpl.

  */

 public class SwedishParliamentOrganisationStructureImporterImpl implements

         GovernmentImporter

 {

     /** The log. */

     private static Log log = LogFactory

             .getLog(SwedishParliamentOrganisationStructureImporterImpl.class);

     /** The web client. */

     private final WebClient webClient;

     /** The portal service. */

     private final PortalService portalService;

     /**

      * Instantiates a new swedish parliament organisation structure importer impl.

      *

      * @param webClient the web client

      * @param portalService the portal service

      * @throws MalformedURLException the malformed url exception

      */

     public SwedishParliamentOrganisationStructureImporterImpl(

             WebClient webClient,PortalService portalService) throws MalformedURLException {

         this.webClient = webClient;

         this.portalService = portalService;

     }

     /*

      * (non-Javadoc)

      *

      * @see org.directdemocracyportal.democracy.service.governmentloader.GovernmentImporter#doImport()

      */

     @SuppressWarnings("unchecked")

     public void doImport() {

         try {

             Long orgId = portalService.findOrgByName(SwedishGovernmentImporterImpl.SVERIGES_RIKSDAG).getId();

             createOrganisationWithMembers("Arbetsmarknadsutskottet", "AU",

                     "http://www.riksdagen.se/webbnav/index.aspx?nid=5304", orgId);

             createOrganisationWithMembers("Civilutskottet", "CU",

                     "http://www.riksdagen.se/webbnav/index.aspx?nid=6104", orgId);

             createOrganisationWithMembers("Finansutskottet", "FiU",

                     "http://www.riksdagen.se/webbnav/index.aspx?nid=5404", orgId);

             createOrganisationWithMembers("Försvarsutskottet", "FöU",

                     "http://www.riksdagen.se/webbnav/index.aspx?nid=5454", orgId);

             createOrganisationWithMembers("Justitieutskottet", "JuU",

                     "http://www.riksdagen.se/webbnav/index.aspx?nid=5504", orgId);

             createOrganisationWithMembers("Konstitutionsutskottet", "KU",

                     "http://www.riksdagen.se/webbnav/index.aspx?nid=5554", orgId);

             createOrganisationWithMembers("Kulturutskottet", "KrU",

                     "http://www.riksdagen.se/webbnav/index.aspx?nid=5604", orgId);

             createOrganisationWithMembers("Miljö- och jordbruksutskottet", "MJU",

                     "http://www.riksdagen.se/webbnav/index.aspx?nid=5704", orgId);

             createOrganisationWithMembers("Näringsutskottet", "NU",

                     "http://www.riksdagen.se/webbnav/index.aspx?nid=5754", orgId);

             createOrganisationWithMembers("Skatteutskottet", "SkU",

                     "http://www.riksdagen.se/webbnav/index.aspx?nid=5804", orgId);

             createOrganisationWithMembers("Socialförsäkringsutskottet", "SfU",

                     "http://www.riksdagen.se/webbnav/index.aspx?nid=5854", orgId);

             createOrganisationWithMembers("Socialutskottet", "SoU",

                     "http://www.riksdagen.se/webbnav/index.aspx?nid=5904", orgId);

             createOrganisationWithMembers("Trafikutskottet", "TU",

                     "http://www.riksdagen.se/webbnav/index.aspx?nid=5954", orgId);

             createOrganisationWithMembers("Utbildningsutskottet", "UbU",

                     "http://www.riksdagen.se/webbnav/index.aspx?nid=6004", orgId);

             createOrganisationWithMembers("Utrikesutskottet", "UU",

                     "http://www.riksdagen.se/webbnav/index.aspx?nid=6054", orgId);

             createOrganisationWithMembers("Sammansatta konstitutions- och utrikesutskottet",

                     "KUU",

                     "http://www.riksdagen.se/webbnav/index.aspx?nid=16640", orgId);

             createOrganisationWithMembers("Sammansatta utrikes- och försvarsutskottet", "UFöU",

                     "http://www.riksdagen.se/webbnav/index.aspx?nid=16110", orgId);

             createOrganisationWithMembers("EU-nämnden", "",

                     "http://www.riksdagen.se/Webbnav/index.aspx?nid=6320", orgId);

         } catch (Exception e) {

             e.printStackTrace();

         }

     }

     /**

      * Creates the organisation with members.

      *

      * @param orgName the org name

      * @param shortName the short name

      * @param url the url

      * @param orgId the org id

      * @throws Exception the exception

      */

     private void createOrganisationWithMembers(String orgName,

             String shortName, String url, Long orgId) throws Exception  {

         HtmlPage page = (HtmlPage) webClient.getPage(url);

         OrgStructureAnswerPage answerPage = new OrgStructureAnswerPage(

                 (HtmlTable) page.getDocumentElement()

                         .getHtmlElementsByTagName("table").iterator().next(),

                 orgName, shortName);

         try {

             Organisation orgUnit = portalService.createOrgUnit(orgName, shortName, OrganisationType.Internal, orgId);

             Person president = portalService.findMemberByNameInOrg(orgId,answerPage.presidentName);

             if (president != null && president.getId() != null) {

                 portalService.addMemberById(orgUnit.getId(), president.getId());

                 Role presidentRole = portalService.createRoleInOrg(OrganisationRoles.President.toString(), orgUnit);

                 portalService.addRolePlayedById(presidentRole.getId(), president.getId());

             } else {

                 System.out.println("Missing " + answerPage.presidentName);

             }

             Person vicePresident = portalService.findMemberByNameInOrg(orgId,answerPage.vicePresidentName);

             if (vicePresident != null && vicePresident.getId() != null) {

                 portalService.addMemberById(orgUnit.getId(), vicePresident.getId());

                 Role vicePresidentRole = portalService.createRoleInOrg(OrganisationRoles.Vice_President.toString(), orgUnit);

                 portalService.addRolePlayedById(vicePresidentRole.getId(), vicePresident.getId());

             } else {

                 System.out.println("Missing " + answerPage.vicePresidentName);

             }

             for (String name : answerPage.memberNames) {

                 Person member = portalService.findMemberByNameInOrg(orgId,name);

                 if (member != null && member.getId() != null) {

                     portalService.addMemberById(orgUnit.getId(), member.getId());

                 } else {

                     System.out.println("Missing " + name);

                 }

             }

             Role substituteRole = portalService.createRoleInOrg(OrganisationRoles.Substitute.toString(), orgUnit);

             for (String name : answerPage.extrasNames) {

                 Person substitute = portalService.findMemberByNameInOrg(orgId,name);

                 if (substitute != null && substitute.getId() != null) {

                     portalService.addRolePlayedById(substituteRole.getId(), substitute.getId());

                 } else {

                     System.out.println("Missing " + name);

                 }

             }

         } catch (OrganisationAlreadyExistException e) {

             // TODO Auto-generated catch block

             e.printStackTrace();

         }

     }

     /**

      * The Class OrgStructureAnswerPage.

      */

     class OrgStructureAnswerPage

     {

         /** The org name. */

         private String orgName;

         /** The short name. */

         private String shortName;

         /** The president name. */

         private String presidentName;

         /** The vice president name. */

         private String vicePresidentName;

         /** The member names. */

         private List<String> memberNames = new ArrayList<String>();

         /** The extras names. */

         private List<String> extrasNames = new ArrayList<String>();

         /**

          * Instantiates a new org structure answer page.

          *

          * @param table the table

          * @param OrgName the org name

          * @param shortName the short name

          */

         public OrgStructureAnswerPage(HtmlTable table, String OrgName,

                 String shortName) {

             orgName = OrgName;

             this.shortName = shortName;

             List<HtmlTableRow> rows = new ArrayList<HtmlTableRow>(table

                     .getRows());

             if (rows.remove(0).asText().equals("Ordförande")) {

                 presidentName = parseName(rows.remove(0).asText());

             }

             if (rows.remove(0).asText().equals("Vice ordförande")) {

                 vicePresidentName = parseName(rows.remove(0).asText());

             }

             if (rows.remove(0).asText().equals("Ledamot")) {

                 HtmlTableRow row = rows.remove(0);

                 while (!row.asText().equals("Suppleant")) {

                     HtmlTableCell cell1 = row.getCell(0);

                     memberNames.add(parseName(cell1.asText()));

                     if (row.getCells().size() > 1) {

                         HtmlTableCell cell2 = row.getCell(1);

                         memberNames.add(parseName(cell2.asText()));

                     }

                     row = rows.remove(0);

                 }

                 for (HtmlTableRow extras : rows) {

                     HtmlTableCell cell1 = extras.getCell(0);

                     extrasNames.add(parseName(cell1.asText()));

                     if (extras.getCells().size() > 1) {

                         HtmlTableCell cell2 = extras.getCell(1);

                         extrasNames.add(parseName(cell2.asText()));

                     }

                 }

             }

         }

         /**

          * Parses the name.

          *

          * @param str the str

          * @return the string

          */

         private String parseName(String str) {

             return str.substring(0, str.indexOf("(") - 1).trim();

         }

         /**

          * Gets the org name.

          *

          * @return the org name

          */

         public String getOrgName() {

             return orgName;

         }

         /**

          * Gets the short name.

          *

          * @return the short name

          */

         public String getShortName() {

             return shortName;

         }

         /**

          * Gets the president name.

          *

          * @return the president name

          */

         public String getPresidentName() {

             return presidentName;

         }

         /**

          * Gets the vice president name.

          *

          * @return the vice president name

          */

         public String getVicePresidentName() {

             return vicePresidentName;

         }

         /**

          * Gets the member names.

          *

          * @return the member names

          */

         public List<String> getMemberNames() {

             return memberNames;

         }

         /**

          * Gets the extras names.

          *

          * @return the extras names

          */

         public List<String> getExtrasNames() {

             return extrasNames;

         }

     }

 }